Merge pull request #31 from jblang/master

Fix batch consistency level bug
Netflix · Jun 12, 2017 · c9964ca · c9964ca
2 parents 7b92b7e + b5b18bb
commit c9964ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...nch-cass-plugins/src/main/java/com/netflix/ndbench/plugin/cass/CassJavaDriverGeneric.java b/...nch-cass-plugins/src/main/java/com/netflix/ndbench/plugin/cass/CassJavaDriverGeneric.java
@@ -52,15 +52,15 @@ public String readSingle(String key) throws Exception {
 
     @Override
     public String writeSingle(String key) throws Exception {
-        BatchStatement batch = new BatchStatement();
+        BatchStatement batch = new BatchStatement(BatchStatement.Type.UNLOGGED);
         for (int i = 0; i < this.MaxColCount; i++) {
             BoundStatement bStmt = writePstmt.bind();
             bStmt.setString("key", key);
             bStmt.setInt("column1", i);
             bStmt.setString("value", this.dataGenerator.getRandomValue());
-            bStmt.setConsistencyLevel(this.WriteConsistencyLevel);
             batch.add(bStmt);
         }
+	batch.setConsistencyLevel(this.WriteConsistencyLevel);
         session.execute(batch);
         batch.clear();
         return ResultOK;