Use synchronized latch to keep senderOnComplete called once

apache · Apr 13, 2024 · 8e812cd · 8e812cd
1 parent 878009a
commit 8e812cd
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/...sh-sdk-java/src/main/java/org/apache/eventmesh/client/grpc/consumer/SubStreamHandler.java b/...sh-sdk-java/src/main/java/org/apache/eventmesh/client/grpc/consumer/SubStreamHandler.java
@@ -130,12 +130,15 @@ public void run() {
     }
 
     public void close() {
-        if (this.sender != null) {
-            senderOnComplete();
+        synchronized (sender) {
+            if (latch.getCount() == 0) {
+                return;
+            }
+            if (this.sender != null) {
+                senderOnComplete();
+            }
+            latch.countDown();
         }
-
-        latch.countDown();
-
         log.info("SubStreamHandler closed.");
     }