iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bael 5016: IN Query Pattern in Cassandra (#14707) 

* Add Cassandra IN Query and update pom file

* Add Cassandra IN Query and update pom file

* add assertion on tests

* format code and move package

* format code

---------

Co-authored-by: Saikat <“saikatcse03@gmail.com”>
This commit is contained in:
Saikat Chakraborty 2023-09-15 18:28:09 +05:30 committed by GitHub
parent 3d4080b6bc
commit 07059835d1
5 changed files with 211 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
persistence-modules/spring-data-cassandra-2/pom.xml
View File

@ -65,6 +65,18 @@
<artifactId>java-driver-mapper-runtime</artifactId>
<version>4.15.0</version>
</dependency>
<dependency>
<groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter-engine</artifactId>
<version>${junit.jupiter.version}</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter-api</artifactId>
<version>${junit.jupiter.version}</version>
<scope>test</scope>
</dependency>
</dependencies>
<build>
@ -88,10 +100,10 @@
<properties>
<java.version>11</java.version>
<org.springframework.data.version>3.1.11</org.springframework.data.version>
<testcontainers.version>1.15.3</testcontainers.version>
<org.springframework.data.version>3.4.15</org.springframework.data.version>
<testcontainers.version>1.19.0</testcontainers.version>
<system.stubs.version>1.1.0</system.stubs.version>
<junit.jupiter.version>5.6.2</junit.jupiter.version>
<junit.jupiter.version>5.9.3</junit.jupiter.version>
</properties>
</project>

12
persistence-modules/spring-data-cassandra-2/src/main/java/org/baeldung/cassandra/inquery/CassandraApplication.java Normal file
View File

@ -0,0 +1,12 @@
package org.baeldung.cassandra.inquery;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
@SpringBootApplication
public class CassandraApplication {
public static void main(String[] args) {
SpringApplication.run(CassandraApplication.class, args);
}
}

49
persistence-modules/spring-data-cassandra-2/src/main/java/org/baeldung/cassandra/inquery/model/Product.java Normal file
View File

@ -0,0 +1,49 @@
package org.baeldung.cassandra.inquery.model;
import org.springframework.data.cassandra.core.cql.PrimaryKeyType;
import org.springframework.data.cassandra.core.mapping.Column;
import org.springframework.data.cassandra.core.mapping.PrimaryKeyColumn;
import org.springframework.data.cassandra.core.mapping.Table;
import java.util.Objects;
import java.util.UUID;
@Table
public class Product {
@PrimaryKeyColumn(name = "product_id", ordinal = 0, type = PrimaryKeyType.PARTITIONED)
private UUID productId;
@PrimaryKeyColumn(name = "product_name", ordinal = 1, type = PrimaryKeyType.CLUSTERED)
private String productName;
@Column("description")
private String description;
@Column("price")
private double price;
public Product(UUID productId, String productName, String description, double price) {
this.productId = productId;
this.productName = productName;
this.description = description;
this.price = price;
}
public String getProductName() {
return productName;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Product product = (Product) o;
return productId.equals(product.productId) && productName.equals(product.productName);
}
@Override
public int hashCode() {
return Objects.hash(productId, productName);
}
}

20
persistence-modules/spring-data-cassandra-2/src/main/java/org/baeldung/cassandra/inquery/repository/ProductRepository.java Normal file
View File

@ -0,0 +1,20 @@
package org.baeldung.cassandra.inquery.repository;
import org.baeldung.cassandra.inquery.model.Product;
import org.springframework.data.cassandra.repository.CassandraRepository;
import org.springframework.data.cassandra.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;
import java.util.List;
import java.util.UUID;
@Repository
public interface ProductRepository extends CassandraRepository<Product, UUID> {
@Query("select * from product where product_id in :productIds")
List<Product> findByProductIds(@Param("productIds") List<UUID> productIds);
@Query("select * from product where product_id = :productId and product_name in :productNames")
List<Product> findByProductIdAndNames(@Param("productId") UUID productId, @Param("productNames") List<String> productNames);
}

115
persistence-modules/spring-data-cassandra-2/src/test/java/org/baeldung/cassandra/inquery/ProductRepositoryNestedLiveTest.java Normal file
View File

@ -0,0 +1,115 @@
package org.baeldung.cassandra.inquery;
import com.datastax.driver.core.Cluster;
import com.datastax.driver.core.Session;
import com.datastax.driver.core.utils.UUIDs;
import org.baeldung.cassandra.inquery.model.Product;
import org.baeldung.cassandra.inquery.repository.ProductRepository;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.testcontainers.containers.CassandraContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;
import java.util.List;
import java.util.UUID;
import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.*;
@Testcontainers
@SpringBootTest
class ProductRepositoryIntegrationTest {
private static final String KEYSPACE_NAME = "mynamespace";
@Container
private static final CassandraContainer cassandra = (CassandraContainer) new CassandraContainer("cassandra:3.11.2")
.withExposedPorts(9042);
@BeforeAll
static void setupCassandraConnectionProperties() {
System.setProperty("spring.data.cassandra.keyspace-name", KEYSPACE_NAME);
System.setProperty("spring.data.cassandra.contact-points", cassandra.getHost());
System.setProperty("spring.data.cassandra.port", String.valueOf(cassandra.getMappedPort(9042)));
createKeyspace(cassandra.getCluster());
}
static void createKeyspace(Cluster cluster) {
try (Session session = cluster.connect()) {
session.execute("CREATE KEYSPACE IF NOT EXISTS " + KEYSPACE_NAME + " WITH replication = \n" +
"{'class':'SimpleStrategy','replication_factor':'1'};");
}
}
@Nested
class ApplicationContextLiveTest {
@Test
void givenCassandraContainer_whenSpringContextIsBootstrapped_thenContainerIsRunningWithNoExceptions() {
assertThat(cassandra.isRunning()).isTrue();
}
}
@Nested
class ProductRepositoryLiveTest {
@Autowired
private ProductRepository productRepository;
@Test
void givenValidProductsIsFetched_whenFindByProductIdsIsCalled_thenProductIsReturned() {
UUID productId1 = UUIDs.timeBased();
UUID productId2 = UUIDs.timeBased();
UUID productId3 = UUIDs.timeBased();
Product product1 = new Product(productId1, "Apple", "Apple v1", 12.5);
Product product2 = new Product(productId2, "Apple v2", "Apple v2", 15.5);
Product product3 = new Product(productId3, "Banana", "Banana v1", 5.5);
Product product4 = new Product(productId3, "Banana v2", "Banana v2", 15.5);
productRepository.saveAll(List.of(product1, product2, product3, product4));
List<Product> existingProducts = productRepository.findByProductIds(List.of(productId1, productId2));
assertEquals(2, existingProducts.size());
assertTrue(existingProducts.contains(product1));
assertTrue(existingProducts.contains(product2));
}
@Test
void givenExistingProducts_whenFindByIdAndNamesIsCalled_thenProductIsReturned() {
UUID productId1 = UUIDs.timeBased();
UUID productId2 = UUIDs.timeBased();
Product product1 = new Product(productId1, "Apple", "Apple v1", 12.5);
Product product2 = new Product(productId1, "Apple v2", "Apple v2", 15.5);
Product product3 = new Product(productId2, "Banana", "Banana v1", 5.5);
Product product4 = new Product(productId2, "Banana v2", "Banana v2", 15.5);
productRepository.saveAll(List.of(product1, product2, product3, product4));
List<Product> existingProducts = productRepository.findByProductIdAndNames(productId1,
List.of(product1.getProductName(), product2.getProductName()));
assertEquals(2, existingProducts.size());
assertTrue(existingProducts.contains(product1));
assertTrue(existingProducts.contains(product2));
}
@Test
void givenNonExistingProductName_whenFindByIdAndNamesIsCalled_thenProductIsReturned() {
UUID productId1 = UUIDs.timeBased();
UUID productId2 = UUIDs.timeBased();
Product product1 = new Product(productId1, "Apple", "Apple v1", 12.5);
Product product2 = new Product(productId1, "Apple v2", "Apple v2", 15.5);
Product product3 = new Product(productId2, "Banana", "Banana v1", 5.5);
Product product4 = new Product(productId2, "Banana v2", "Banana v2", 15.5);
productRepository.saveAll(List.of(product1, product2, product4));
List<Product> existingProducts = productRepository.findByProductIdAndNames(productId1,
List.of(product3.getProductName()));
assertEquals(0, existingProducts.size());
}
}
}