Skip to content

Commit ccd1824

Browse files
author
Eugen
committed
Merge pull request eugenp#291 from sameira/master
Spring Data Cassandra
2 parents cdcc25e + 68c7e86 commit ccd1824

8 files changed

Lines changed: 368 additions & 0 deletions

File tree

‎spring-data-cassandra/pom.xml‎

Lines changed: 110 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,110 @@
1+
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
2+
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
3+
<modelVersion>4.0.0</modelVersion>
4+
5+
<groupId>org.baeldung</groupId>
6+
<artifactId>spring-data-cassandra</artifactId>
7+
<version>0.0.1-SNAPSHOT</version>
8+
<packaging>jar</packaging>
9+
10+
<name>spring-data-cassandra</name>
11+
12+
<properties>
13+
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
14+
<org.springframework.data.version>1.3.1.RELEASE</org.springframework.data.version>
15+
<org.springframework.version>4.2.2.RELEASE</org.springframework.version>
16+
<junit.version>4.11</junit.version>
17+
<org.slf4j.version>1.7.12</org.slf4j.version>
18+
<logback.version>1.1.3</logback.version>
19+
<cassandra-driver-core.version>2.1.5</cassandra-driver-core.version>
20+
<cassandra-unit-spring.version>2.1.9.2</cassandra-unit-spring.version>
21+
<cassandra-unit-shaded>2.1.9.2</cassandra-unit-shaded>
22+
<hector-core.version>2.0-0</hector-core.version>
23+
</properties>
24+
25+
<dependencies>
26+
<dependency>
27+
<groupId>org.springframework.data</groupId>
28+
<artifactId>spring-data-cassandra</artifactId>
29+
<version>${org.springframework.data.version}</version>
30+
</dependency>
31+
<dependency>
32+
<groupId>org.springframework</groupId>
33+
<artifactId>spring-core</artifactId>
34+
<version>${org.springframework.version}</version>
35+
</dependency>
36+
<dependency>
37+
<groupId>junit</groupId>
38+
<artifactId>junit-dep</artifactId>
39+
<version>${junit.version}</version>
40+
<scope>test</scope>
41+
</dependency>
42+
<dependency>
43+
<groupId>org.springframework</groupId>
44+
<artifactId>spring-test</artifactId>
45+
<version>${org.springframework.version}</version>
46+
<scope>test</scope>
47+
</dependency>
48+
<dependency>
49+
<groupId>org.cassandraunit</groupId>
50+
<artifactId>cassandra-unit-spring</artifactId>
51+
<version>${cassandra-unit-spring.version}</version>
52+
<scope>test</scope>
53+
<exclusions>
54+
<exclusion>
55+
<groupId>org.cassandraunit</groupId>
56+
<artifactId>cassandra-unit</artifactId>
57+
</exclusion>
58+
</exclusions>
59+
</dependency>
60+
<dependency>
61+
<groupId>org.cassandraunit</groupId>
62+
<artifactId>cassandra-unit-shaded</artifactId>
63+
<version>${cassandra-unit-shaded}</version>
64+
<scope>test</scope>
65+
</dependency>
66+
<dependency>
67+
<groupId>org.hectorclient</groupId>
68+
<artifactId>hector-core</artifactId>
69+
<version>${hector-core.version}</version>
70+
</dependency>
71+
<dependency>
72+
<groupId>com.datastax.cassandra</groupId>
73+
<artifactId>cassandra-driver-core</artifactId>
74+
<version>${cassandra-driver-core.version}</version>
75+
<optional>true</optional>
76+
</dependency>
77+
<dependency>
78+
<groupId>org.slf4j</groupId>
79+
<artifactId>slf4j-api</artifactId>
80+
<version>${org.slf4j.version}</version>
81+
</dependency>
82+
<dependency>
83+
<groupId>ch.qos.logback</groupId>
84+
<artifactId>logback-classic</artifactId>
85+
<version>${logback.version}</version>
86+
</dependency>
87+
<dependency>
88+
<groupId>org.slf4j</groupId>
89+
<artifactId>jcl-over-slf4j</artifactId>
90+
<version>${org.slf4j.version}</version>
91+
</dependency>
92+
<dependency>
93+
<groupId>org.slf4j</groupId>
94+
<artifactId>log4j-over-slf4j</artifactId>
95+
<version>${org.slf4j.version}</version>
96+
</dependency>
97+
</dependencies>
98+
<build>
99+
<plugins>
100+
<plugin>
101+
<artifactId>maven-compiler-plugin</artifactId>
102+
<version>2.3.2</version>
103+
<configuration>
104+
<source>1.7</source>
105+
<target>1.7</target>
106+
</configuration>
107+
</plugin>
108+
</plugins>
109+
</build>
110+
</project>

‎spring-data-cassandra/src/main/java/org/baeldung/spring/data/cassandra/config/CassandraConfig.java‎

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
package org.baeldung.spring.data.cassandra.config;
2+
3+
import org.apache.commons.logging.Log;
4+
import org.apache.commons.logging.LogFactory;
5+
import org.springframework.beans.factory.annotation.Autowired;
6+
import org.springframework.context.annotation.Bean;
7+
import org.springframework.context.annotation.Configuration;
8+
import org.springframework.context.annotation.PropertySource;
9+
import org.springframework.core.env.Environment;
10+
import org.springframework.data.cassandra.config.CassandraClusterFactoryBean;
11+
import org.springframework.data.cassandra.config.java.AbstractCassandraConfiguration;
12+
import org.springframework.data.cassandra.mapping.BasicCassandraMappingContext;
13+
import org.springframework.data.cassandra.mapping.CassandraMappingContext;
14+
import org.springframework.data.cassandra.repository.config.EnableCassandraRepositories;
15+
16+
@Configuration
17+
@PropertySource(value = { "classpath:cassandra.properties" })
18+
@EnableCassandraRepositories(basePackages = "org.baeldung.spring.data.cassandra.repository")
19+
public class CassandraConfig extends AbstractCassandraConfiguration {
20+
21+
private static final Log LOGGER = LogFactory.getLog(CassandraConfig.class);
22+
@Autowired
23+
private Environment environment;
24+
25+
@Override
26+
protected String getKeyspaceName() {
27+
return environment.getProperty("cassandra.keyspace");
28+
}
29+
30+
@Bean
31+
public CassandraClusterFactoryBean cluster() {
32+
CassandraClusterFactoryBean cluster = new CassandraClusterFactoryBean();
33+
cluster.setContactPoints(environment.getProperty("cassandra.contactpoints"));
34+
cluster.setPort(Integer.parseInt(environment.getProperty("cassandra.port")));
35+
LOGGER.info("Cluster created with contact points [" + environment.getProperty("cassandra.contactpoints") + "] " + "& port [" + Integer.parseInt(environment.getProperty("cassandra.port")) + "].");
36+
return cluster;
37+
}
38+
39+
@Bean
40+
public CassandraMappingContext cassandraMapping() throws ClassNotFoundException {
41+
return new BasicCassandraMappingContext();
42+
}
43+
}

‎spring-data-cassandra/src/main/java/org/baeldung/spring/data/cassandra/model/Book.java‎

Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
package org.baeldung.spring.data.cassandra.model;
2+
3+
import org.springframework.cassandra.core.Ordering;
4+
import org.springframework.cassandra.core.PrimaryKeyType;
5+
import org.springframework.data.cassandra.mapping.Column;
6+
import org.springframework.data.cassandra.mapping.PrimaryKeyColumn;
7+
import org.springframework.data.cassandra.mapping.Table;
8+
9+
import java.util.HashSet;
10+
import java.util.Set;
11+
import java.util.UUID;
12+
13+
@Table
14+
public class Book {
15+
@PrimaryKeyColumn(name = "isbn", ordinal = 2, type = PrimaryKeyType.CLUSTERED, ordering = Ordering.DESCENDING)
16+
private UUID id;
17+
@PrimaryKeyColumn(name = "title", ordinal = 0, type = PrimaryKeyType.PARTITIONED)
18+
private String title;
19+
20+
@PrimaryKeyColumn(name = "publisher", ordinal = 1, type = PrimaryKeyType.PARTITIONED)
21+
private String publisher;
22+
@Column
23+
private Set<String> tags = new HashSet<>();
24+
25+
public Book(UUID id, String title, String publisher, Set<String> tags) {
26+
this.id = id;
27+
this.title = title;
28+
this.publisher = publisher;
29+
this.tags.addAll(tags);
30+
}
31+
32+
public UUID getId() {
33+
return id;
34+
}
35+
36+
public String getTitle() {
37+
return title;
38+
}
39+
40+
public String getPublisher() {
41+
return publisher;
42+
}
43+
44+
public Set getTags() {
45+
return tags;
46+
}
47+
48+
public void setId(UUID id) {
49+
this.id = id;
50+
}
51+
52+
public void setTitle(String title) {
53+
this.title = title;
54+
}
55+
56+
public void setPublisher(String publisher) {
57+
this.publisher = publisher;
58+
}
59+
60+
public void setTags(Set<String> tags) {
61+
this.tags = tags;
62+
}
63+
}

‎spring-data-cassandra/src/main/java/org/baeldung/spring/data/cassandra/repository/BookRepository.java‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
package org.baeldung.spring.data.cassandra.repository;
2+
3+
import org.baeldung.spring.data.cassandra.model.Book;
4+
import org.springframework.data.cassandra.repository.CassandraRepository;
5+
import org.springframework.data.cassandra.repository.Query;
6+
import org.springframework.stereotype.Repository;
7+
8+
@Repository
9+
public interface BookRepository extends CassandraRepository<Book> {
10+
@Query("select * from book where title = ?0 and publisher=?1")
11+
Iterable<Book> findByTitleAndPublisher(String title, String publisher);
12+
}

‎spring-data-cassandra/src/main/resources/cassandra.properties‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
cassandra.contactpoints=127.0.0.1
2+
cassandra.port=9142
3+
cassandra.keyspace=testKeySpace

‎spring-data-cassandra/src/main/resources/logback.xml‎

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
<configuration>
2+
3+
<appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
4+
<encoder>
5+
<pattern>web - %date [%thread] %-5level %logger{36} - %message%n
6+
</pattern>
7+
</encoder>
8+
</appender>
9+
10+
<logger name="org.springframework" level="WARN" />
11+
<logger name="org.springframework.transaction" level="WARN" />
12+
13+
<!-- in order to debug some marshalling issues, this needs to be TRACE -->
14+
<logger name="org.springframework.web.servlet.mvc" level="WARN" />
15+
16+
<root level="INFO">
17+
<appender-ref ref="STDOUT" />
18+
</root>
19+
20+
</configuration>

‎spring-data-cassandra/src/main/resources/test.png‎

855 Bytes
Loading

‎spring-data-cassandra/src/test/java/org/baeldung/spring/data/cassandra/repository/BookRepositoryIntegrationTest.java‎

Lines changed: 117 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,117 @@
1+
package org.baeldung.spring.data.cassandra.repository;
2+
3+
import com.datastax.driver.core.Cluster;
4+
import com.datastax.driver.core.Session;
5+
import com.datastax.driver.core.utils.UUIDs;
6+
import com.google.common.collect.ImmutableSet;
7+
import org.apache.cassandra.exceptions.ConfigurationException;
8+
import org.apache.commons.logging.Log;
9+
import org.apache.commons.logging.LogFactory;
10+
import org.apache.thrift.transport.TTransportException;
11+
import org.baeldung.spring.data.cassandra.config.CassandraConfig;
12+
import org.baeldung.spring.data.cassandra.model.Book;
13+
import org.cassandraunit.utils.EmbeddedCassandraServerHelper;
14+
import org.junit.*;
15+
import org.junit.runner.RunWith;
16+
import org.springframework.beans.factory.annotation.Autowired;
17+
import org.springframework.cassandra.core.cql.CqlIdentifier;
18+
import org.springframework.data.cassandra.core.CassandraAdminOperations;
19+
import org.springframework.test.context.ContextConfiguration;
20+
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
21+
22+
import java.io.IOException;
23+
import java.util.HashMap;
24+
25+
import static org.junit.Assert.assertEquals;
26+
import static org.junit.Assert.assertNotEquals;
27+
28+
@RunWith(SpringJUnit4ClassRunner.class)
29+
@ContextConfiguration(classes = CassandraConfig.class)
30+
public class BookRepositoryIntegrationTest {
31+
32+
private static final Log LOGGER = LogFactory.getLog(BookRepositoryIntegrationTest.class);
33+
34+
public static final String KEYSPACE_CREATION_QUERY = "CREATE KEYSPACE IF NOT EXISTS testKeySpace WITH replication = { 'class': 'SimpleStrategy', 'replication_factor': '3' };";
35+
36+
public static final String KEYSPACE_ACTIVATE_QUERY = "USE testKeySpace;";
37+
38+
public static final String DATA_TABLE_NAME = "book";
39+
40+
@Autowired
41+
private BookRepository bookRepository;
42+
43+
@Autowired
44+
private CassandraAdminOperations adminTemplate;
45+
46+
@BeforeClass
47+
public static void startCassandraEmbedded() throws InterruptedException, TTransportException, ConfigurationException, IOException {
48+
EmbeddedCassandraServerHelper.startEmbeddedCassandra();
49+
Cluster cluster = Cluster.builder().addContactPoints("127.0.0.1")
50+
.withPort(9142).build();
51+
LOGGER.info("Server Started at 127.0.0.1:9142... ");
52+
Session session = cluster.connect();
53+
session.execute(KEYSPACE_CREATION_QUERY);
54+
session.execute(KEYSPACE_ACTIVATE_QUERY);
55+
LOGGER.info(session.execute("Select * from Book").all().toArray());
56+
Thread.sleep(5000);
57+
LOGGER.info("KeySpace created and activated.");
58+
}
59+
60+
@Before
61+
public void createTable() throws InterruptedException, TTransportException, ConfigurationException, IOException {
62+
adminTemplate.createTable(true, CqlIdentifier.cqlId(DATA_TABLE_NAME), Book.class, new HashMap<String, Object>());
63+
}
64+
65+
@Test
66+
public void whenSavingBook_thenAvailableOnRetrieval() {
67+
Book javaBook = new Book(UUIDs.timeBased(), "Head First Java",
68+
"O'Reilly Media", ImmutableSet.of("Computer", "Software"));
69+
bookRepository.save(ImmutableSet.of(javaBook));
70+
Iterable<Book> books = bookRepository.findByTitleAndPublisher("Head First Java", "O'Reilly Media");
71+
assertEquals(javaBook.getId(), books.iterator().next().getId());
72+
}
73+
74+
@Test
75+
public void whenUpdatingBooks_thenAvailableOnRetrieval() {
76+
Book javaBook = new Book(UUIDs.timeBased(), "Head First Java", "O'Reilly Media", ImmutableSet.of("Computer", "Software"));
77+
bookRepository.save(ImmutableSet.of(javaBook));
78+
Iterable<Book> books = bookRepository.findByTitleAndPublisher("Head First Java", "O'Reilly Media");
79+
javaBook.setTitle("Head First Java Second Edition");
80+
bookRepository.save(ImmutableSet.of(javaBook));
81+
Iterable<Book> updateBooks = bookRepository.findByTitleAndPublisher("Head First Java Second Edition", "O'Reilly Media");
82+
assertEquals(javaBook.getTitle(), updateBooks.iterator().next().getTitle());
83+
}
84+
85+
@Test(expected = java.util.NoSuchElementException.class)
86+
public void whenDeletingExistingBooks_thenNotAvailableOnRetrieval() {
87+
Book javaBook = new Book(UUIDs.timeBased(), "Head First Java", "O'Reilly Media", ImmutableSet.of("Computer", "Software"));
88+
bookRepository.save(ImmutableSet.of(javaBook));
89+
bookRepository.delete(javaBook);
90+
Iterable<Book> books = bookRepository.findByTitleAndPublisher("Head First Java", "O'Reilly Media");
91+
assertNotEquals(javaBook.getId(), books.iterator().next().getId());
92+
}
93+
94+
@Test
95+
public void whenSavingBooks_thenAllShouldAvailableOnRetrieval() {
96+
Book javaBook = new Book(UUIDs.timeBased(), "Head First Java",
97+
"O'Reilly Media", ImmutableSet.of("Computer", "Software"));
98+
Book dPatternBook = new Book(UUIDs.timeBased(), "Head Design Patterns",
99+
"O'Reilly Media", ImmutableSet.of("Computer", "Software"));
100+
bookRepository.save(ImmutableSet.of(javaBook));
101+
bookRepository.save(ImmutableSet.of(dPatternBook));
102+
Iterable<Book> books = bookRepository.findAll();
103+
int bookCount = 0;
104+
for (Book book : books) bookCount++;
105+
assertEquals(bookCount, 2);
106+
}
107+
108+
@After
109+
public void dropTable() {
110+
adminTemplate.dropTable(CqlIdentifier.cqlId(DATA_TABLE_NAME));
111+
}
112+
113+
@AfterClass
114+
public static void stopCassandraEmbedded() {
115+
EmbeddedCassandraServerHelper.cleanEmbeddedCassandra();
116+
}
117+
}

0 commit comments

Comments
 (0)