Merge pull request #1772 from grainier/latency-tracker-fix

Fix latency tracker markOuts
siddhi-io · Mar 18, 2022 · 7ab327e · 7ab327e
2 parents 6124e06 + 8330c51
commit 7ab327e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/modules/siddhi-core/src/main/java/io/siddhi/core/table/Table.java b/modules/siddhi-core/src/main/java/io/siddhi/core/table/Table.java
@@ -222,6 +222,10 @@ public void addEvents(ComplexEventChunk<StreamEvent> addingEventChunk, int noOfE
         }
         addingEventChunk.reset();
         add(addingEventChunk);
+        if (latencyTrackerInsert != null &&
+                Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
+            latencyTrackerInsert.markOut();
+        }
         if (throughputTrackerInsert != null &&
                 Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
             throughputTrackerInsert.eventsIn(noOfEvents);
@@ -336,6 +340,10 @@ public void deleteEvents(ComplexEventChunk<StateEvent> deletingEventChunk, Compi
             latencyTrackerDelete.markIn();
         }
         delete(deletingEventChunk, compiledCondition);
+        if (latencyTrackerDelete != null &&
+                Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
+            latencyTrackerDelete.markOut();
+        }
         if (throughputTrackerDelete != null &&
                 Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
             throughputTrackerDelete.eventsIn(noOfEvents);
@@ -415,6 +423,10 @@ public void updateEvents(ComplexEventChunk<StateEvent> updatingEventChunk,
             latencyTrackerUpdate.markIn();
         }
         update(updatingEventChunk, compiledCondition, compiledUpdateSet);
+        if (latencyTrackerUpdate != null &&
+                Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
+            latencyTrackerUpdate.markOut();
+        }
         if (throughputTrackerUpdate != null &&
                 Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
             throughputTrackerUpdate.eventsIn(noOfEvents);
@@ -500,6 +512,10 @@ public void updateOrAddEvents(ComplexEventChunk<StateEvent> updateOrAddingEventC
         }
         updateOrAdd(updateOrAddingEventChunk, compiledCondition, compiledUpdateSet,
                 addingStreamEventExtractor);
+        if (latencyTrackerUpdateOrInsert != null &&
+                Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
+            latencyTrackerUpdateOrInsert.markOut();
+        }
         if (throughputTrackerUpdateOrInsert != null &&
                 Level.BASIC.compareTo(siddhiAppContext.getRootMetricsLevel()) <= 0) {
             throughputTrackerUpdateOrInsert.eventsIn(noOfEvents);