@Test(groups = "short") public void should_apply_statement_timestamp_only_to_batched_queries_without_timestamp() { BatchStatement batch = new BatchStatement(); batch.add(new SimpleStatement("INSERT INTO foo (k, v) VALUES (1, 1)")); batch.add(new SimpleStatement("INSERT INTO foo (k, v) VALUES (2, 1) USING TIMESTAMP 20")); batch.setDefaultTimestamp(10); session().execute(batch); long writeTime1 = session().execute("SELECT writeTime(v) FROM foo WHERE k = 1").one().getLong(0); long writeTime2 = session().execute("SELECT writeTime(v) FROM foo WHERE k = 2").one().getLong(0); assertEquals(writeTime1, 10); assertEquals(writeTime2, 20); }
@Override public String writeSingle(String key) throws Exception { BatchStatement batch = new BatchStatement(); for (int i = 0; i < config.getBatchSize(); i++) { BoundStatement bStmt; if(config.getUseMultiPartition()) { if(randomObj.nextBoolean()) { bStmt = getBStmtTable1(key); } else { bStmt = getBStmtTable2(key); } } else { bStmt = getBStmtTable1(key); } bStmt.setConsistencyLevel(ConsistencyLevel.valueOf(config.getWriteConsistencyLevel())); batch.add(bStmt); } if(config.getUseTimestamp()) { batch.setDefaultTimestamp(Instant.now().toEpochMilli()*1000); } session.execute(batch); batch.clear(); return ResultOK; }
@Override public String writeSingle(String key) throws Exception { BatchStatement batch = new BatchStatement(); for (int i = 0; i < config.getBatchSize(); i++) { BoundStatement bStmt; if(config.getUseMultiPartition()) { if(randomObj.nextBoolean()) { bStmt = getBStmtTable1(key); } else { bStmt = getBStmtTable2(key); } } else { bStmt = getBStmtTable1(key); } bStmt.setConsistencyLevel(ConsistencyLevel.valueOf(config.getWriteConsistencyLevel())); batch.add(bStmt); } if(config.getUseTimestamp()) { batch.setDefaultTimestamp(Instant.now().toEpochMilli()*1000); } session.execute(batch); batch.clear(); return ResultOK; }
@Test(groups = "short") public void should_apply_statement_timestamp_only_to_batched_queries_without_timestamp() { BatchStatement batch = new BatchStatement(); batch.add(new SimpleStatement("INSERT INTO foo (k, v) VALUES (1, 1)")); batch.add(new SimpleStatement("INSERT INTO foo (k, v) VALUES (2, 1) USING TIMESTAMP 20")); batch.setDefaultTimestamp(10); session().execute(batch); long writeTime1 = session().execute("SELECT writeTime(v) FROM foo WHERE k = 1").one().getLong(0); long writeTime2 = session().execute("SELECT writeTime(v) FROM foo WHERE k = 2").one().getLong(0); assertEquals(writeTime1, 10); assertEquals(writeTime2, 20); }