feat: sync ask with timeout

zfoo-project · Oct 13, 2024 · 92fc58f · 92fc58f
1 parent 37a30d9
commit 92fc58f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/net/src/main/java/com/zfoo/net/router/IRouter.java b/net/src/main/java/com/zfoo/net/router/IRouter.java
@@ -56,6 +56,8 @@ public interface IRouter {
      */
     <T> SyncAnswer<T> syncAsk(Session session, Object packet, @Nullable Class<T> answerClass, @Nullable Object argument) throws Exception;
 
+    <T> SyncAnswer<T> syncAsk(Session session, Object packet, @Nullable Class<T> answerClass, @Nullable Object argument, long timeoutMillis) throws Exception;
+
     <T> AsyncAnswer<T> asyncAsk(Session session, Object packet, @Nullable Class<T> answerClass, @Nullable Object argument);
 
     <T> AsyncAnswer<T> asyncAsk(Session session, Object packet, @Nullable Class<T> answerClass, @Nullable Object argument, long timeoutMillis);

diff --git a/net/src/main/java/com/zfoo/net/router/Router.java b/net/src/main/java/com/zfoo/net/router/Router.java
@@ -223,6 +223,11 @@ public void send(Session session, Object packet) {
 
     @Override
     public <T> SyncAnswer<T> syncAsk(Session session, Object packet, @Nullable Class<T> answerClass, @Nullable Object argument) throws Exception {
+        return syncAsk(session, packet, answerClass, argument, DEFAULT_TIMEOUT);
+    }
+
+    @Override
+    public <T> SyncAnswer<T> syncAsk(Session session, Object packet, Class<T> answerClass, Object argument, long timeoutMillis) throws Exception {
         var clientSignalAttachment = new SignalAttachment();
         if (argument == null) {
             clientSignalAttachment.setClient(SignalAttachment.SIGNAL_NATIVE_NO_ARGUMENT_CLIENT);
@@ -237,7 +242,7 @@ public <T> SyncAnswer<T> syncAsk(Session session, Object packet, @Nullable Class
             // 里面调用的依然是：send方法发送消息
             send(session, packet, clientSignalAttachment);
 
-            Object responsePacket = clientSignalAttachment.getResponseFuture().get(DEFAULT_TIMEOUT, TimeUnit.MILLISECONDS);
+            Object responsePacket = clientSignalAttachment.getResponseFuture().get(timeoutMillis, TimeUnit.MILLISECONDS);
 
             if (responsePacket.getClass() == Error.class) {
                 throw new ErrorResponseException((Error) responsePacket);
@@ -254,7 +259,6 @@ public <T> SyncAnswer<T> syncAsk(Session session, Object packet, @Nullable Class
         } finally {
             SignalBridge.removeSignalAttachment(clientSignalAttachment);
         }
-
     }
 
     /**
@@ -384,15 +388,15 @@ public void atReceiver(PacketReceiverTask packetReceiverTask) {
         }
     }
 
-    protected void exceptionHandler(Exception e, PacketReceiverTask packetReceiverTask){
+    protected void exceptionHandler(Exception e, PacketReceiverTask packetReceiverTask) {
         var session = packetReceiverTask.getSession();
         var packet = packetReceiverTask.getPacket();
         var attachment = packetReceiverTask.getAttachment();
         EventBus.post(ServerExceptionEvent.valueOf(session, packet, attachment, e));
         logger.error("at{} e[uid:{}][sid:{}] invoke exception", StringUtils.capitalize(packet.getClass().getSimpleName()), session.getUid(), session.getSid(), e);
     }
 
-    protected void throwableHandler(Throwable t, PacketReceiverTask packetReceiverTask){
+    protected void throwableHandler(Throwable t, PacketReceiverTask packetReceiverTask) {
         var session = packetReceiverTask.getSession();
         var packet = packetReceiverTask.getPacket();
         logger.error("at{} e[uid:{}][sid:{}] invoke error", StringUtils.capitalize(packet.getClass().getSimpleName()), session.getUid(), session.getSid(), t);