Pipe: Fix PipeDataRegionEventCounter not work in PipeRealtimeDataRegi…

…onExtractor (#12747)
apache · Jun 17, 2024 · e6e819a · e6e819a
1 parent 89af73d
commit e6e819a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...re/datanode/src/main/java/org/apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java b/...re/datanode/src/main/java/org/apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java
@@ -21,6 +21,7 @@
 
 import org.apache.iotdb.commons.pipe.metric.PipeEventCounter;
 import org.apache.iotdb.db.pipe.event.common.heartbeat.PipeHeartbeatEvent;
+import org.apache.iotdb.db.pipe.event.realtime.PipeRealtimeEvent;
 import org.apache.iotdb.pipe.api.event.Event;
 import org.apache.iotdb.pipe.api.event.dml.insertion.TabletInsertionEvent;
 import org.apache.iotdb.pipe.api.event.dml.insertion.TsFileInsertionEvent;
@@ -51,6 +52,9 @@ public Integer getPipeHeartbeatEventCount() {
 
   @Override
   public void increaseEventCount(Event event) {
+    if (event instanceof PipeRealtimeEvent) {
+      event = ((PipeRealtimeEvent) event).getEvent();
+    }
     if (Objects.isNull(event)) {
       return;
     }
@@ -65,6 +69,9 @@ public void increaseEventCount(Event event) {
 
   @Override
   public void decreaseEventCount(Event event) {
+    if (event instanceof PipeRealtimeEvent) {
+      event = ((PipeRealtimeEvent) event).getEvent();
+    }
     if (Objects.isNull(event)) {
       return;
     }