(iotdb) branch master updated: Pipe: Fix PipeDataRegionEventCounter not work in PipeRealtimeDataRegionExtractor (#12747)

[rong]

This is an automated email from the ASF dual-hosted git repository.

rong pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git


The following commit(s) were added to refs/heads/master by this push:
     new e6e819ad1fb Pipe: Fix PipeDataRegionEventCounter not work in 
PipeRealtimeDataRegionExtractor (#12747)
e6e819ad1fb is described below

commit e6e819ad1fbebccefd7d7f05eda95ce192f79f0f
Author: Caideyipi <87789683+caidey...@users.noreply.github.com>
AuthorDate: Mon Jun 17 18:14:17 2024 +0800

    Pipe: Fix PipeDataRegionEventCounter not work in 
PipeRealtimeDataRegionExtractor (#12747)
---
 .../apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java    | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git 
a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java
 
b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java
index 0e1aa557ec0..0aed61801ef 100644
--- 
a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java
+++ 
b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/pipe/metric/PipeDataRegionEventCounter.java
@@ -21,6 +21,7 @@ package org.apache.iotdb.db.pipe.metric;
 
 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 class PipeDataRegionEventCounter extends 
PipeEventCounter {
 
   @Override
   public void increaseEventCount(Event event) {
+    if (event instanceof PipeRealtimeEvent) {
+      event = ((PipeRealtimeEvent) event).getEvent();
+    }
     if (Objects.isNull(event)) {
       return;
     }
@@ -65,6 +69,9 @@ public class PipeDataRegionEventCounter extends 
PipeEventCounter {
 
   @Override
   public void decreaseEventCount(Event event) {
+    if (event instanceof PipeRealtimeEvent) {
+      event = ((PipeRealtimeEvent) event).getEvent();
+    }
     if (Objects.isNull(event)) {
       return;
     }