check for null when a host dies

patricioe · Apr 14, 2011 · 7a53f8d · 7a53f8d
1 parent ba650e1
commit 7a53f8d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/com/riptano/cassandra/stress/InsertCommand.java b/src/main/java/com/riptano/cassandra/stress/InsertCommand.java
@@ -52,7 +52,8 @@ public Void call() throws Exception {
             try {
                 MutationResult mr = mutator.execute();
                 LatencyTracker writeCount = commandRunner.latencies.get(mr.getHostUsed());
-                writeCount.addMicro(mr.getExecutionTimeMicro());
+                if ( writeCount != null )
+                  writeCount.addMicro(mr.getExecutionTimeMicro());
                 mutator.discardPendingMutations();
 
                 log.info("executed batch of {}. {} of {} complete", new Object[]{commandArgs.batchSize, rows, commandArgs.getKeysPerThread()});