Polishing.

Rename tests, replace space indents with tabs.

Closes #707

Author: christophstrobl

README.adoc

@@ -8,6 +8,7 @@ We have separate folders for the samples of individual modules:
 
 == Spring Data for Apache Cassandra
 
+* `aot-optimization` - Use Ahead-Of-Time Repositories with Spring Data for Apache Cassandra.
 * `example` - Shows core Spring Data support for Apache Cassandra.
 * `kotlin` - Example for using Cassandra with Kotlin.
 * `reactive` - Example project to show reactive template and repository support.
@@ -23,6 +24,7 @@ Local Elasticsearch instance must be running to run the tests.
 
 == Spring Data JDBC
 
+* `aot-optimization` - Use Ahead-Of-Time Repositories with Spring Data JDBC.
 * `basic` - Basic usage of Spring Data JDBC.
 * `graalvm-native` - This example compiles a basic Spring Data JDBC application into a GraalVM native image.
 * `howto` - A collection of projects to go with the https://spring.io/blog/2021/09/09/spring-data-jdbc-how-to-use-custom-id-generation[Spring Data JDBC - How to blog posts].
@@ -35,6 +37,7 @@ Local Elasticsearch instance must be running to run the tests.
 
 == Spring Data JPA
 
+* `aot-optimization` - Use Ahead-Of-Time Repositories with Spring Data JPA.
 * `eclipselink` - Sample project to show how to use Spring Data JPA with Spring Boot and https://www.eclipse.org/eclipselink/[Eclipselink].
 * `example` - Probably the project you want to have a look at first.
 Contains a variety of sample packages, showcasing the different levels at which you can use Spring Data JPA.
@@ -55,6 +58,7 @@ Contains also examples running on Virtual Threads.
 
 == Spring Data MongoDB
 
+* `aot-optimization` - Use Ahead-Of-Time Repositories with Spring Data MongoDB.
 * `aggregation` - Example project to showcase the MongoDB aggregation framework support.
 * `example` - Example project for general repository functionality (including geo-spatial functionality), Querydsl integration and advanced topics.
 * `fluent-api` - Example project to show the new fluent API (`MongoTemplate`-alternative) to interact with MongoDB.

cassandra/vector-search/README.md

@@ -34,4 +34,4 @@ public interface CommentRepository extends Repository<Comment, String> {
 }
 ```
 
-This example contains a test class to illustrate vector search with a Repository in `VectorAppTest`.
+This example contains a test class to illustrate vector search with a Repository in `CassandraVectorSearchTest`.

cassandra/vector-search/pom.xml

@@ -3,66 +3,41 @@
          xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
 	<modelVersion>4.0.0</modelVersion>
-
 	<parent>
 		<groupId>org.springframework.data.examples</groupId>
 		<artifactId>spring-data-cassandra-examples</artifactId>
 		<version>4.0.0-SNAPSHOT</version>
-    </parent>
+	</parent>
 
-    <artifactId>spring-data-cassandra-vector-search</artifactId>
+	<artifactId>spring-data-cassandra-vector-search</artifactId>
+	<name>Spring Data Cassandra - Vector Search</name>
 
 	<dependencies>
 
-        <dependency>
-            <groupId>org.jspecify</groupId>
-            <artifactId>jspecify</artifactId>
-            <version>1.0.0</version>
-        </dependency>
-
-        <dependency>
-            <groupId>org.springframework.boot</groupId>
-            <artifactId>spring-boot-starter-test</artifactId>
-            <scope>test</scope>
-        </dependency>
-
-        <dependency>
-            <groupId>org.springframework.boot</groupId>
-            <artifactId>spring-boot-testcontainers</artifactId>
-            <scope>test</scope>
-        </dependency>
-
-        <dependency>
-            <groupId>org.testcontainers</groupId>
-            <artifactId>testcontainers-cassandra</artifactId>
-            <exclusions>
-                <exclusion>
-                    <groupId>com.datastax.cassandra</groupId>
-                    <artifactId>cassandra-driver-core</artifactId>
-                </exclusion>
-            </exclusions>
-        </dependency>
-
-        <dependency>
-            <groupId>org.springframework.data.examples</groupId>
-            <artifactId>spring-data-cassandra-example-utils</artifactId>
-            <version>${project.version}</version>
-            <scope>test</scope>
-        </dependency>
-
-    </dependencies>
-
-    <build>
-        <plugins>
-            <plugin>
-                <groupId>org.apache.maven.plugins</groupId>
-                <artifactId>maven-compiler-plugin</artifactId>
-            </plugin>
-            <plugin>
-                <groupId>org.springframework.boot</groupId>
-                <artifactId>spring-boot-maven-plugin</artifactId>
-            </plugin>
-        </plugins>
-    </build>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-testcontainers</artifactId>
+			<scope>test</scope>
+		</dependency>
+
+		<dependency>
+			<groupId>org.testcontainers</groupId>
+			<artifactId>testcontainers-cassandra</artifactId>
+			<exclusions>
+				<exclusion>
+					<groupId>com.datastax.cassandra</groupId>
+					<artifactId>cassandra-driver-core</artifactId>
+				</exclusion>
+			</exclusions>
+		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.data.examples</groupId>
+			<artifactId>spring-data-cassandra-example-utils</artifactId>
+			<version>${project.version}</version>
+			<scope>test</scope>
+		</dependency>
+
+	</dependencies>
 
 </project>

cassandra/vector-search/src/main/java/example/springdata/vector/Comment.java

@@ -5,7 +5,7 @@
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
- *      https://www.apache.org/licenses/LICENSE-2.0
+ *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
@@ -29,48 +29,45 @@
 @Table
 public class Comment {
 
-    @Id
-    private String id;
+	private @Id String id;
 
-    private String country;
-    private String description;
+	private String country;
+	private String description;
 
-    @SaiIndexed
-    @VectorType(dimensions = 5)
-    private Vector embedding;
+	@SaiIndexed
+	@VectorType(dimensions = 5) private Vector embedding;
 
-    public Comment() {
-    }
+	public Comment() {}
 
-    public Comment(String country, String description, Vector embedding) {
-        this.id = UUID.randomUUID().toString();
-        this.country = country;
-        this.description = description;
-        this.embedding = embedding;
-    }
+	public Comment(String country, String description, Vector embedding) {
+		this.id = UUID.randomUUID().toString();
+		this.country = country;
+		this.description = description;
+		this.embedding = embedding;
+	}
 
-    public static Comment of(Comment source) {
-        return new Comment(source.getCountry(), source.getDescription(), source.getEmbedding());
-    }
+	public static Comment of(Comment source) {
+		return new Comment(source.getCountry(), source.getDescription(), source.getEmbedding());
+	}
 
-    public String getId() {
-        return id;
-    }
+	public String getId() {
+		return id;
+	}
 
-    public String getCountry() {
-        return country;
-    }
+	public String getCountry() {
+		return country;
+	}
 
-    public String getDescription() {
-        return description;
-    }
+	public String getDescription() {
+		return description;
+	}
 
-    public Vector getEmbedding() {
-        return embedding;
-    }
+	public Vector getEmbedding() {
+		return embedding;
+	}
 
-    @Override
-    public String toString() {
-        return "%s (%s)".formatted(getDescription(), getCountry());
-    }
+	@Override
+	public String toString() {
+		return "%s (%s)".formatted(getDescription(), getCountry());
+	}
 }

cassandra/vector-search/src/main/java/example/springdata/vector/CommentRepository.java

@@ -5,7 +5,7 @@
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
- *      https://www.apache.org/licenses/LICENSE-2.0
+ *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
@@ -25,8 +25,8 @@
 
 public interface CommentRepository extends CrudRepository<Comment, String> {
 
-    SearchResults<Comment> searchTop10ByEmbeddingNear(Vector embedding, ScoringFunction function);
+	SearchResults<Comment> searchTop10ByEmbeddingNear(Vector embedding, ScoringFunction function);
 
-    @Query("SELECT id, description, country, similarity_cosine(embedding,:embedding) AS score FROM comment ORDER BY embedding ANN OF :embedding LIMIT :limit")
-    SearchResults<Comment> searchAnnotated(Vector embedding, Score distance, Limit limit);
+	@Query("SELECT id, description, country, similarity_cosine(embedding,:embedding) AS score FROM comment ORDER BY embedding ANN OF :embedding LIMIT :limit")
+	SearchResults<Comment> searchAnnotated(Vector embedding, Score distance, Limit limit);
 }

cassandra/vector-search/src/main/java/example/springdata/vector/VectorApp.java

@@ -5,7 +5,7 @@
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
- *      https://www.apache.org/licenses/LICENSE-2.0
+ *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
@@ -25,28 +25,28 @@
 @SpringBootApplication
 public class VectorApp {
 
-    public static void main(String[] args) {
-        SpringApplication.run(VectorApp.class, args);
-    }
+	public static void main(String[] args) {
+		SpringApplication.run(VectorApp.class, args);
+	}
 
-    @Component
-    static class DbInitializer implements CommandLineRunner {
+	@Component
+	static class DbInitializer implements CommandLineRunner {
 
-        private final CassandraTemplate template;
+		private final CassandraTemplate template;
 
-        DbInitializer(CassandraTemplate template) {
-            this.template = template;
-        }
+		DbInitializer(CassandraTemplate template) {
+			this.template = template;
+		}
 
-        @Override
-        public void run(String... args) {
+		@Override
+		public void run(String... args) {
 
-            template.truncate(Comment.class);
+			template.truncate(Comment.class);
 
-            template.insert(new Comment("de", "comment 'one'", Vector.of(0.1001f, 0.22345f, 0.33456f, 0.44567f, 0.55678f)));
-            template.insert(new Comment("de", "comment 'two'", Vector.of(0.2001f, 0.32345f, 0.43456f, 0.54567f, 0.65678f)));
-            template.insert(new Comment("en", "comment 'three'", Vector.of(0.9001f, 0.82345f, 0.73456f, 0.64567f, 0.55678f)));
-            template.insert(new Comment("de", "comment 'four'", Vector.of(0.9001f, 0.92345f, 0.93456f, 0.94567f, 0.95678f)));
-        }
-    }
+			template.insert(new Comment("de", "comment 'one'", Vector.of(0.1001f, 0.22345f, 0.33456f, 0.44567f, 0.55678f)));
+			template.insert(new Comment("de", "comment 'two'", Vector.of(0.2001f, 0.32345f, 0.43456f, 0.54567f, 0.65678f)));
+			template.insert(new Comment("en", "comment 'three'", Vector.of(0.9001f, 0.82345f, 0.73456f, 0.64567f, 0.55678f)));
+			template.insert(new Comment("de", "comment 'four'", Vector.of(0.9001f, 0.92345f, 0.93456f, 0.94567f, 0.95678f)));
+		}
+	}
 }

cassandra/vector-search/src/main/resources/application.properties

@@ -2,5 +2,4 @@ logging.level.org=WARN
 logging.level.com.datastax=WARN
 
 spring.cassandra.schema-action=recreate
-spring.cassandra.
 spring.cassandra.keyspace-name=vector_search_keyspace

cassandra/vector-search/src/test/java/example/springdata/vector/CassandraDBConfiguration.java

@@ -5,7 +5,7 @@
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
- *      https://www.apache.org/licenses/LICENSE-2.0
+ *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
@@ -17,27 +17,42 @@
 
 import java.net.InetSocketAddress;
 
-import com.datastax.oss.driver.api.core.CqlSession;
-import org.springframework.boot.autoconfigure.cassandra.CassandraProperties;
-import org.springframework.boot.autoconfigure.cassandra.CqlSessionBuilderCustomizer;
+import org.springframework.boot.cassandra.autoconfigure.CassandraConnectionDetails;
+import org.springframework.boot.cassandra.autoconfigure.CassandraProperties;
+import org.springframework.boot.cassandra.autoconfigure.CqlSessionBuilderCustomizer;
+import org.springframework.boot.testcontainers.service.connection.ServiceConnection;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
+import org.testcontainers.cassandra.CassandraContainer;
+import org.testcontainers.utility.DockerImageName;
+
+import com.datastax.oss.driver.api.core.CqlSession;
+
 @Configuration
 public class CassandraDBConfiguration {
 
-    @Bean
-    CqlSessionBuilderCustomizer sessionBuilderCustomizer(CassandraProperties properties) {
-        return sessionBuilder -> {
+	@Bean
+	@ServiceConnection
+	CassandraContainer pgVectorContainer() {
+		return new CassandraContainer(DockerImageName.parse("cassandra:5")).withReuse(true);
+	}
+
+	@Bean
+	CqlSessionBuilderCustomizer sessionBuilderCustomizer(CassandraConnectionDetails connectionDetails,
+			CassandraProperties properties) {
+
+		return sessionBuilder -> {
 
-            InetSocketAddress contactPoint = new InetSocketAddress(properties.getContactPoints().iterator().next(), properties.getPort());
+			CqlSession session = CqlSession.builder()
+					.addContactPoints(connectionDetails.getContactPoints().stream()
+							.map(it -> new InetSocketAddress(it.host(), it.port())).toList())
+					.withLocalDatacenter(connectionDetails.getLocalDatacenter()).build();
 
-            CqlSession session = CqlSession.builder().addContactPoint(contactPoint)
-                .withLocalDatacenter(properties.getLocalDatacenter()).build();
+			session.execute("CREATE KEYSPACE IF NOT EXISTS " + properties.getKeyspaceName() + " WITH replication = \n"
+					+ "{'class':'SimpleStrategy','replication_factor':'1'};");
+			session.close();
+		};
+	}
 
-            session.execute("CREATE KEYSPACE IF NOT EXISTS " + properties.getKeyspaceName() + " WITH replication = \n"
-                + "{'class':'SimpleStrategy','replication_factor':'1'};");
-            session.close();
-        };
-    }
 }