Only handle clientSendsOneMessage

jd-opensource · Dec 30, 2024 · df7f322 · df7f322
1 parent cf50a99
commit df7f322
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...src/main/java/com/jd/live/agent/plugin/router/gprc/interceptor/GrpcServerInterceptor.java b/...src/main/java/com/jd/live/agent/plugin/router/gprc/interceptor/GrpcServerInterceptor.java
@@ -72,9 +72,10 @@ public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
                 ServerCall<ReqT, RespT> call,
                 Metadata headers,
                 ServerCallHandler<ReqT, RespT> next) {
-
-            Callable<Object> callable = () -> new LiveServerCallListener<ReqT>(next.startCall(call, headers));
-
+            if (!call.getMethodDescriptor().getType().clientSendsOneMessage()) {
+                return next.startCall(call, headers);
+            }
+            Callable<Object> callable = () -> new LiveServerCallListener<>(next.startCall(call, headers));
             GrpcInboundRequest request = new GrpcInboundRequest(call, headers);
             try {
                 Object result = !request.isSystem()