diff --git a/scenarios/turbine-heat-sensor/src/main/java/io/pravega/turbineheatsensor/TurbineHeatSensor.java b/scenarios/turbine-heat-sensor/src/main/java/io/pravega/turbineheatsensor/TurbineHeatSensor.java index 573811db..a7bbc20e 100644 --- a/scenarios/turbine-heat-sensor/src/main/java/io/pravega/turbineheatsensor/TurbineHeatSensor.java +++ b/scenarios/turbine-heat-sensor/src/main/java/io/pravega/turbineheatsensor/TurbineHeatSensor.java @@ -438,13 +438,16 @@ public void run() { do { try { final EventRead result = reader.readNextEvent(0); - produceStats.runAndRecordTime(() -> { - return CompletableFuture.completedFuture(null); - }, Long.parseLong(result.getEvent().split(",")[0]), 100); + if (result.getEvent() != null) { + consumeStats.runAndRecordTime(() -> { + return CompletableFuture.completedFuture(null); + }, Long.parseLong(result.getEvent().split(",")[0]), 100); + totalEvents--; + } } catch (ReinitializationRequiredException e) { e.printStackTrace(); } - } while ( totalEvents-- > 0 ); + } while (totalEvents > 0); } finally { reader.close();