feat(s3stream): optimize eventLoop

AutoMQ · Aug 15, 2024 · 8cc63bd · 8cc63bd
1 parent c8e5240
commit 8cc63bd
Showing 1 changed file with 5 additions and 0 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
@@ -87,6 +87,11 @@ public CompletableFuture<Void> submit(Runnable task) {
     public void execute(Runnable task) {
         check();
         tasks.add(task);
+        if (shutdown.get()) {
+            if (tasks.remove(task)) {
+                throw new IllegalStateException("EventLoop is shutdown");
+            }
+        }
     }
 
     public CompletableFuture<Void> shutdownGracefully() {