BAEL-4982 Batching In Cassandra (#11574)

* Implemented cassandra batch query * Added netty version param * Reformatted correctly * Reformatted correctly * Reformatted correctly * Formatting fix resolved * Formatting fix resolved * Removed unused method * Refactored method for better readability * tab spaces corrected Co-authored-by: saikat chakraborty <saikat.chakraborty@tesco.com>
2022-01-20 19:10:00 +05:30
parent 5a3494c262
commit 49319e7eb0
7 changed files with 525 additions and 1 deletions
@@ -0,0 +1,133 @@
+package com.baeldung.cassandra.batch.epository;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import org.apache.cassandra.exceptions.ConfigurationException;
+import org.apache.thrift.transport.TTransportException;
+import org.cassandraunit.utils.EmbeddedCassandraServerHelper;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.FixMethodOrder;
+import org.junit.Test;
+import org.junit.runners.MethodSorters;
+
+import com.baeldung.cassandra.batch.CassandraConnector;
+import com.baeldung.cassandra.batch.domain.Product;
+import com.baeldung.cassandra.batch.repository.KeyspaceRepository;
+import com.baeldung.cassandra.batch.repository.ProductRepository;
+import com.datastax.oss.driver.api.core.CqlIdentifier;
+import com.datastax.oss.driver.api.core.CqlSession;
+import com.datastax.oss.driver.api.core.cql.ColumnDefinition;
+import com.datastax.oss.driver.api.core.cql.ResultSet;
+
+@FixMethodOrder(MethodSorters.NAME_ASCENDING)
+public class ProductRepositoryIntegrationTest {
+
+    private KeyspaceRepository schemaRepository;
+
+    private ProductRepository productRepository;
+
+    private CqlSession session;
+
+    private final String KEYSPACE_NAME = "testBaeldungKeyspace";
+    private final String PRODUCT = "product";
+
+    @BeforeClass
+    public static void init() throws ConfigurationException, TTransportException, IOException, InterruptedException {
+        // Start an embedded Cassandra Server
+        EmbeddedCassandraServerHelper.startEmbeddedCassandra(20000L);
+    }
+
+    @Before
+    public void connect() {
+        CassandraConnector client = new CassandraConnector();
+        client.connect("127.0.0.1", 9142,"datacenter1");
+        session = client.getSession();
+        schemaRepository = new KeyspaceRepository(client.getSession());
+        schemaRepository.createKeyspace(KEYSPACE_NAME, 1);
+        schemaRepository.useKeyspace(KEYSPACE_NAME);
+        productRepository = new ProductRepository(client.getSession());
+    }
+    
+    @Test
+    public void whenCreatingAProductTable_thenCreatedCorrectly() {
+        productRepository.deleteTable(KEYSPACE_NAME);
+        productRepository.createProductTable(KEYSPACE_NAME);
+
+        ResultSet result = session.execute("SELECT * FROM " + KEYSPACE_NAME + "." + PRODUCT + ";");
+
+        List<ColumnDefinition> colDef = new ArrayList<>();
+        
+        result.getColumnDefinitions().forEach(columnDef -> colDef.add(columnDef));
+        List<String> columnNames = colDef.stream().map(ColumnDefinition::getName).map(CqlIdentifier::toString).collect(Collectors.toList());
+        assertEquals(columnNames.size(), 5);
+        assertTrue(columnNames.contains("product_id"));
+        assertTrue(columnNames.contains("variant_id"));
+        assertTrue(columnNames.contains("product_name"));
+        assertTrue(columnNames.contains("description"));
+        assertTrue(columnNames.contains("price"));
+    }
+
+    @Test
+    public void whenCreatingRelatedProductBatch_thenCreatedCorrectly() {
+        productRepository.deleteTable(KEYSPACE_NAME);
+        productRepository.createProductTableByName(KEYSPACE_NAME);
+        productRepository.createProductByIdTable(KEYSPACE_NAME);
+    	
+        Product product = getTestProduct();
+        productRepository.insertProductBatch(product);
+        List<Product> productByIdList = productRepository.selectAllProductById(KEYSPACE_NAME);
+        List<Product> productByNameList = productRepository.selectAllProductByName(KEYSPACE_NAME);
+        
+        assertEquals(productByIdList.size(), 1);
+        assertEquals(productByNameList.size(), 1);
+        assertEquals(productByIdList.get(0).getProductName(), "Banana");
+        assertEquals(productByNameList.get(0).getProductName(), "Banana");
+        assertEquals(productByIdList.get(0).getDescription(), "Banana");
+        assertEquals(productByNameList.get(0).getDescription(), "Banana");
+        assertEquals(productByIdList.get(0).getPrice(), 12f,0f);
+        assertEquals(productByNameList.get(0).getPrice(), 12f,0f);      
+    }
+    
+    @Test
+    public void whenCreatingMultiVariantProductBatch_thenCreatedCorrectly() {
+        productRepository.deleteTable(KEYSPACE_NAME);
+        productRepository.createProductTable(KEYSPACE_NAME);
+    	
+        Product productV1 = getTestProduct();
+        Product productV2 = getTestProduct();
+        productRepository.insertProductVariantBatch(productV1, productV2);
+        List<Product> productList = productRepository.selectAllProduct(KEYSPACE_NAME);
+        
+        assertEquals(productList.size(), 2);
+        assertEquals(productList.get(0).getProductName(), "Banana");
+        assertEquals(productList.get(1).getProductName(), "Banana");   
+        assertEquals(productList.get(0).getDescription(), "Banana");
+        assertEquals(productList.get(1).getDescription(), "Banana");
+        assertEquals(productList.get(0).getPrice(), 12f,0f);
+        assertEquals(productList.get(1).getPrice(), 12f,0f);
+    }
+    
+
+
+    @AfterClass
+    public static void cleanup() {
+        EmbeddedCassandraServerHelper.cleanEmbeddedCassandra();
+    }
+    
+    private Product getTestProduct() {
+        Product product = new Product();
+        product.setProductName("Banana");
+        product.setDescription("Banana");
+        product.setPrice(12f);
+		
+        return product;
+    }
+}