Move the unary call if-condition in TimeoutServerInterceptor

api/src/main/java/io/grpc/TimeoutServerInterceptor.java

@@ -18,8 +18,9 @@
 
 /**
  * An optional ServerInterceptor that can interrupt server calls that are running for too long time.
+ * In this way, it prevents problematic code from using up all threads.
  *
- * <p>You can add it to your server using the ServerBuilder#intercept(ServerInterceptor) method.
+ * <p>How to use: you can add it to your server using the ServerBuilder#intercept(ServerInterceptor) method.
  *
  * <p>Limitation: it only applies the timeout to unary calls (streaming calls will run without timeout).
  */
@@ -36,8 +37,13 @@ public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
       ServerCall<ReqT, RespT> serverCall,
       Metadata metadata,
       ServerCallHandler<ReqT, RespT> serverCallHandler) {
-    return new TimeoutServerCallListener<>(
-        serverCallHandler.startCall(serverCall, metadata), serverCall, serverTimeoutManager);
+    // Only intercepts unary calls because the timeout is inapplicable to streaming calls.
+    if (serverCall.getMethodDescriptor().getType().clientSendsOneMessage()) {
+      return new TimeoutServerCallListener<>(
+              serverCallHandler.startCall(serverCall, metadata), serverCall, serverTimeoutManager);
+    } else {
+      return serverCallHandler.startCall(serverCall, metadata);
+    }
   }
 
   /** A listener that intercepts the RPC method invocation for timeout control. */
@@ -57,17 +63,12 @@ private TimeoutServerCallListener(
     }
 
     /**
-     * Only intercepts unary calls because the timeout is inapplicable to streaming calls.
      * Intercepts onHalfClose() because the RPC method is called in it. See
      * io.grpc.stub.ServerCalls.UnaryServerCallHandler.UnaryServerCallListener
      */
     @Override
     public void onHalfClose() {
-      if (serverCall.getMethodDescriptor().getType().clientSendsOneMessage()) {
-        serverTimeoutManager.intercept(super::onHalfClose);
-      } else {
-        super.onHalfClose();
-      }
+      serverTimeoutManager.intercept(super::onHalfClose);
     }
   }
 }