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

simplify switch

Browse Source
This commit is contained in:
Jonathan Shook 2021-06-23 11:25:15 -05:00
parent a80e6d599b
commit 1e8ad0cc91
2 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
driver-cql-shaded/src/main/java/io/nosqlbench/activitytype/cql/core/CQLBindHelper.java
View File

@ -1,7 +1,7 @@
package io.nosqlbench.activitytype.cql.core;
import com.datastax.driver.core.*;
import io.nosqlbench.engine.api.activityconfig.ParsedStmt;
import io.nosqlbench.engine.api.activityconfig.ParsedStmtOp;
import io.nosqlbench.engine.api.activityconfig.yaml.OpTemplate;
import java.math.BigDecimal;
@ -79,13 +79,10 @@ public class CQLBindHelper {
public static BoundStatement bindStatement(Statement statement, String name, Object value, DataType.Name typeName) {
switch (typeName) {
case ASCII: // ASCII(1, String.class)
return ((BoundStatement) statement).bind().setString(name, (String) value);
case VARCHAR: // VARCHAR(13, String.class)
return ((BoundStatement) statement).bind().setString(name, (String) value);
case TEXT: // TEXT(10, String.class)
return ((BoundStatement) statement).bind().setString(name, (String) value);
case BIGINT: // BIGINT(2, Long.class)
return ((BoundStatement) statement).bind().setLong(name, (long) value);
case COUNTER: // COUNTER(5, Long.class)
return ((BoundStatement) statement).bind().setLong(name, (long) value);
case BLOB: // BLOB(3, ByteBuffer.class)
@ -102,24 +99,20 @@ public class CQLBindHelper {
return ((BoundStatement) statement).bind().setFloat(name, (float) value);
case INET: // INET(16, InetAddress.class)
return ((BoundStatement) statement).bind().setInet(name, (InetAddress) value);
case VARINT: // VARINT(14, BigInteger.class)
case SMALLINT:
case TINYINT:
case INT: // INT(9, Integer.class)
return ((BoundStatement) statement).bind().setInt(name, (int) value);
case TIMESTAMP: // TIMESTAMP(11, Date.class)
return ((BoundStatement) statement).bind().setTimestamp(name, (Date) value);
case UUID: // UUID(12, UUID.class)
return ((BoundStatement) statement).bind().setUUID(name, (UUID) value);
case TIMEUUID: // TIMEUUID(15, UUID.class)
return ((BoundStatement) statement).bind().setUUID(name, (UUID) value);
case VARINT: // VARINT(14, BigInteger.class)
return ((BoundStatement) statement).bind().setInt(name, (int) value);
case UDT: // UDT(48, UDTValue.class)
return ((BoundStatement) statement).bind().setUDTValue(name, (UDTValue) value);
case TUPLE: // TUPLE(49, TupleValue.class
return ((BoundStatement) statement).bind().setTupleValue(name, (TupleValue) value);
case SMALLINT:
return ((BoundStatement) statement).bind().setInt(name, (int) value);
case TINYINT:
return ((BoundStatement) statement).bind().setInt(name, (int) value);
case DATE:
return ((BoundStatement) statement).bind().setDate(name, (LocalDate) value);
case TIME:
@ -129,8 +122,8 @@ public class CQLBindHelper {
}
}
public static Map<String, String> parseAndGetSpecificBindings(OpTemplate opDef, ParsedStmt parsed) {
String statement = opDef.getStmt();
public static Map<String, String> parseAndGetSpecificBindings(OpTemplate opDef, ParsedStmtOp parsed) {
String statement = opDef.getStmt().orElseThrow();
Set<String> extraBindings = new HashSet<>(opDef.getBindings().keySet());
Map<String, String> specificBindings = new LinkedHashMap<>();

8
driver-cqld3-shaded/src/main/java/io/nosqlbench/activitytype/cql/core/CQLBindHelper.java
View File

@ -1,8 +1,8 @@
package io.nosqlbench.activitytype.cql.core;
import com.datastax.driver.core.*;
import io.nosqlbench.engine.api.activityconfig.ParsedStmt;
import io.nosqlbench.engine.api.activityconfig.yaml.OpDef;
import io.nosqlbench.engine.api.activityconfig.ParsedStmtOp;
import io.nosqlbench.engine.api.activityconfig.yaml.OpTemplate;
import java.math.BigDecimal;
import java.net.InetAddress;
@ -122,10 +122,10 @@ public class CQLBindHelper {
}
}
public static Map<String, String> parseAndGetSpecificBindings(OpDef opDef, ParsedStmt parsed) {
public static Map<String, String> parseAndGetSpecificBindings(OpTemplate opDef, ParsedStmtOp parsed) {
List<String> spans = new ArrayList<>();
String statement = opDef.getStmt();
String statement = opDef.getStmt().orElseThrow();
Set<String> extraBindings = new HashSet<>();
extraBindings.addAll(opDef.getBindings().keySet());