Merge pull request #261 from Netflix/fix-bound-statement

fix bound statement
Netflix · Jun 9, 2021 · e22f93c · e22f93c
2 parents 0b88b07 + 39aea68
commit e22f93c
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/...ass-plugins/src/main/java/com/netflix/ndbench/plugin/datastax4/CassJavaDriverGeneric.java b/...ass-plugins/src/main/java/com/netflix/ndbench/plugin/datastax4/CassJavaDriverGeneric.java
@@ -57,9 +57,9 @@ protected CassJavaDriverGeneric(
     @Override public String readSingle(String key) throws Exception {
         int nRows = 0;
 
-        BoundStatement bStmt = readPstmt.bind();
-        bStmt.setString("key", key);
-        bStmt.setConsistencyLevel(consistencyLevel(config.getReadConsistencyLevel()));
+        BoundStatement bStmt = readPstmt.bind()
+                .setString("key", key)
+                .setConsistencyLevel(consistencyLevel(config.getReadConsistencyLevel()));
         ResultSet rs = session.execute(bStmt);
         List<Row> result=rs.all();
 
@@ -130,12 +130,10 @@ private ConsistencyLevel consistencyLevel(String consistencyLevelConfig) {
 
     private BoundStatement getWriteBStmt(String key, int col)
     {
-        BoundStatement bStmt = writePstmt.bind();
-        bStmt.setString("key", key);
-        bStmt.setInt("column1", col);
+        BoundStatement bStmt = writePstmt.bind().setString("key", key).setInt("column1", col);
         for (int i = 0; i < config.getColsPerRow(); i++)
         {
-            bStmt.setString(getValueColumnName(i), this.dataGenerator.getRandomValue());
+            bStmt = bStmt.setString(getValueColumnName(i), this.dataGenerator.getRandomValue());
         }
         return bStmt;
     }