IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

Include cleaner handling for cql vector types

Browse Source
This commit is contained in:
jeffbanks 2023-06-05 18:42:18 -05:00
parent 08989143ed
commit 7bbe41705c
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
adapter-cqld4/src/main/java/io/nosqlbench/adapter/cqld4/opdispensers/CQLD4PreparedStmtDiagnostics.java
View File

@ -25,6 +25,7 @@ import com.datastax.oss.driver.api.core.data.CqlDuration;
import com.datastax.oss.driver.api.core.data.CqlVector; import com.datastax.oss.driver.api.core.data.CqlVector;
import com.datastax.oss.driver.api.core.data.TupleValue; import com.datastax.oss.driver.api.core.data.TupleValue;
import com.datastax.oss.driver.api.core.data.UdtValue; import com.datastax.oss.driver.api.core.data.UdtValue;
import com.datastax.oss.driver.api.core.type.CqlVectorType;
import com.datastax.oss.driver.api.core.type.DataType; import com.datastax.oss.driver.api.core.type.DataType;
import io.nosqlbench.adapter.cqld4.optypes.Cqld4CqlOp; import io.nosqlbench.adapter.cqld4.optypes.Cqld4CqlOp;
import io.nosqlbench.api.errors.OpConfigError; import io.nosqlbench.api.errors.OpConfigError;
@ -57,8 +58,14 @@ public class CQLD4PreparedStmtDiagnostics {
Object colval, DataType coltype) { Object colval, DataType coltype) {
return switch (coltype.getProtocolCode()) { return switch (coltype.getProtocolCode()) {
// TODO - need to handle 'custom' more specifically, interim change for vector search. case CUSTOM -> {
case CUSTOM -> bound.setCqlVector(colname, (CqlVector<?>) colval); if (coltype instanceof CqlVectorType) {
yield bound.setCqlVector(colname, (CqlVector<?>) colval);
}
throw new RuntimeException("Unhandled CUSTOM type for diagnostic: "
+ coltype.getClass().getSimpleName());
}
case ASCII, VARCHAR -> bound.setString(colname, (String) colval); case ASCII, VARCHAR -> bound.setString(colname, (String) colval);
case BIGINT, COUNTER -> bound.setLong(colname, (long) colval); case BIGINT, COUNTER -> bound.setLong(colname, (long) colval);
case BLOB -> bound.setByteBuffer(colname, (ByteBuffer) colval); case BLOB -> bound.setByteBuffer(colname, (ByteBuffer) colval);