feat(eventLoop): optimize eventLoop

AutoMQ · Aug 14, 2024 · c8e5240 · c8e5240
1 parent b1445e4
commit c8e5240
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/s3stream/src/main/java/com/automq/stream/utils/threads/EventLoop.java b/s3stream/src/main/java/com/automq/stream/utils/threads/EventLoop.java
@@ -90,9 +90,11 @@ public void execute(Runnable task) {
     }
 
     public CompletableFuture<Void> shutdownGracefully() {
-        if (shutdown.compareAndSet(false, true)) {
-            if (!shutdownCf.isDone()) {
-                tasks.add(WAKEUP_TASK);
+        while (!shutdown.get()) {
+            if (shutdown.compareAndSet(false, true)) {
+                if (!shutdownCf.isDone()) {
+                    tasks.add(WAKEUP_TASK);
+                }
             }
         }
         return shutdownCf;