fix

XiaoMi · Nov 17, 2020 · d9a1c46 · d9a1c46
1 parent 4a2a4b9
commit d9a1c46
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/main/java/com/xiaomi/infra/pegasus/rpc/interceptor/ReplicaSessionInterceptor.java b/src/main/java/com/xiaomi/infra/pegasus/rpc/interceptor/ReplicaSessionInterceptor.java
@@ -25,5 +25,6 @@ public interface ReplicaSessionInterceptor {
   void onConnected(ReplicaSession session);
 
   // The behavior when rpc session is sending a message.
+  // @returns false if this message shouldn't be sent.
   boolean onSendMessage(ReplicaSession session, final ReplicaSession.RequestEntry entry);
 }
diff --git a/src/main/java/com/xiaomi/infra/pegasus/rpc/interceptor/ReplicaSessionInterceptorManager.java b/src/main/java/com/xiaomi/infra/pegasus/rpc/interceptor/ReplicaSessionInterceptorManager.java
@@ -41,10 +41,11 @@ public void onConnected(ReplicaSession session) {
   }
 
   public boolean onSendMessage(ReplicaSession session, final ReplicaSession.RequestEntry entry) {
-    boolean ret = true;
     for (ReplicaSessionInterceptor interceptor : interceptors) {
-      ret &= interceptor.onSendMessage(session, entry);
+      if (!interceptor.onSendMessage(session, entry)) {
+        return false;
+      }
     }
-    return ret;
+    return true;
   }
 }