don't block request futures in client event loop

rfresh2 · Jan 30, 2025 · 272345f · 272345f
1 parent bceaa84
commit 272345f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/main/java/com/zenith/util/RequestFuture.java b/src/main/java/com/zenith/util/RequestFuture.java
@@ -1,5 +1,6 @@
 package com.zenith.util;
 
+import com.zenith.Proxy;
 import lombok.SneakyThrows;
 import org.jetbrains.annotations.NotNull;
 
@@ -47,14 +48,27 @@ public boolean isDone() {
     @SneakyThrows
     @Override
     public Boolean get() {
+        var client = Proxy.getInstance().getClient();
+        if (client != null && client.getClientEventLoop().inEventLoop()) {
+            throw new IllegalStateException("Cannot block on RequestFuture in client event loop");
+        }
         Wait.waitUntil(() -> completed, 1, 1L, TimeUnit.SECONDS);
         return accepted;
     }
 
     @SneakyThrows
     @Override
     public Boolean get(final long timeout, @NotNull final TimeUnit unit) {
+        var client = Proxy.getInstance().getClient();
+        if (client != null && client.getClientEventLoop().inEventLoop()) {
+            throw new IllegalStateException("Cannot block on RequestFuture in client event loop");
+        }
         Wait.waitUntil(() -> completed, 1, timeout, unit);
         return accepted;
     }
+
+    public boolean getNow() {
+        if (!completed) return false;
+        return accepted;
+    }
 }