CASSANDRA-21126: Vector search support in cassandra-easy-stress

build.gradle

@@ -87,6 +87,9 @@ dependencies {
     implementation libs.hadoop.common
     implementation libs.hadoop.mapreduce.client.common
 
+    // HDF5 support for Vector Search
+    implementation libs.jhdf
+
     // MCP Server dependencies
     implementation (libs.mcp.sdk) {
         exclude group: 'org.slf4j', module: 'slf4j-api'

gradle/libs.versions.toml

@@ -35,6 +35,7 @@ coroutines-test = "1.9.0"
 mcp-sdk = "0.7.2"
 ktor = "3.3.0"
 testcontainers = "1.19.8"
+jhdf = "0.7.0"
 
 [libraries]
 jcommander = { module = "com.beust:jcommander", version.ref = "jcommander" }
@@ -71,6 +72,8 @@ parquet-hadoop = { module = "org.apache.parquet:parquet-hadoop", version.ref = "
 hadoop-common = { module = "org.apache.hadoop:hadoop-common", version.ref = "hadoop" }
 hadoop-mapreduce-client-common = { module = "org.apache.hadoop:hadoop-mapreduce-client-common", version.ref = "hadoop" }
 
+jhdf = { module = "io.jhdf:jhdf", version.ref = "jhdf" }
+
 junit-jupiter-engine = { module = "org.junit.jupiter:junit-jupiter-engine", version.ref = "junit" }
 junit-jupiter-params = { module = "org.junit.jupiter:junit-jupiter-params", version.ref = "junit-params" }
 assertj-core = { module = "org.assertj:assertj-core", version.ref = "assertj" }

src/main/kotlin/org/apache/cassandra/easystress/OperationCallback.kt

@@ -65,15 +65,17 @@ class OperationCallback(
         // TODO (visibility): include details about paging?
         context.collect(op, Either.Left(result!!), startNanos, endNanos)
 
-        // do the callback for mutations
-        // might extend this to select, but I can't see a reason for it now
+        // do the callback for mutations and selects
         when (op) {
             is Operation.Mutation -> {
                 runner.onSuccess(op, result)
             }
             is Operation.DDL -> {
                 runner.onSuccess(op, result)
             }
+            is Operation.SelectStatement -> {
+                runner.onSuccess(op, result)
+            }
             is Operation.Stop -> {
                 throw OperationStopException()
             }

src/main/kotlin/org/apache/cassandra/easystress/workloads/IStressWorkload.kt

@@ -57,6 +57,11 @@ interface IStressRunner {
         op: Operation.DDL,
         result: AsyncResultSet?,
     ) { }
+
+    fun onSuccess(
+        op: Operation.SelectStatement,
+        result: AsyncResultSet?,
+    ) { }
 }
 
 /**
@@ -131,6 +136,7 @@ sealed class Operation(
 
     class SelectStatement(
         bound: BoundStatement,
+        val callbackPayload: Any? = null,
     ) : Operation(bound)
 
     class Deletion(