feat: 新增关闭监听器及超时管理器控制方法

Author: magicliang

trpc-proto/trpc-proto-http/src/main/java/com/tencent/trpc/proto/http/client/AbstractConsumerInvoker.java

@@ -35,6 +35,8 @@
 import com.tencent.trpc.core.rpc.TimeoutManager;
 import com.tencent.trpc.core.rpc.def.DefTimeoutManager;
 import com.tencent.trpc.core.utils.JsonUtils;
+import com.tencent.trpc.core.common.ConfigManager;
+import com.tencent.trpc.core.common.ShutdownListener;
 import com.tencent.trpc.core.utils.ProtoJsonConverter;
 import com.tencent.trpc.core.utils.RpcUtils;
 import com.tencent.trpc.core.utils.StringUtils;
@@ -64,7 +66,12 @@ public abstract class AbstractConsumerInvoker<T> implements ConsumerInvoker<T> {
     /**
      * The request timeout manager, used to handle the result of client requests in timeout scenarios.
      */
-    private static final TimeoutManager TIMEOUT_MANAGER = new DefTimeoutManager(10);
+    private static TimeoutManager TIMEOUT_MANAGER = new DefTimeoutManager(10);
+    
+    /**
+     * Internal shutdown listener that handles the shutdown of the timeout manager
+     */
+    private final ShutdownListener shutdownListener = new InternalShutdownListener();
 
     /**
      * Http client for the request
@@ -96,6 +103,9 @@ public AbstractConsumerInvoker(AbstractRpcClient client,
         if (extMap.containsKey(KEYSTORE_PATH) && extMap.containsKey(KEYSTORE_PASS)) {
             scheme = HTTPS_SCHEME;
         }
+        
+        // register shutdown listener
+        ConfigManager.getInstance().registerShutdownListener(shutdownListener);
     }
 
     /**
@@ -285,4 +295,38 @@ public ProtocolConfig getProtocolConfig() {
     public Class<T> getInterface() {
         return config.getServiceInterface();
     }
+
+    /**
+     * Stop the timeout manager
+     */
+    public static void stop() {
+        TIMEOUT_MANAGER.close();
+    }
+
+    /**
+     * Called when the container is reset.
+     */
+    public static void reset() {
+        TIMEOUT_MANAGER = new DefTimeoutManager(10);
+    }
+
+    /**
+     * Get the internal shutdown listener for testing purposes
+     *
+     * @return the internal shutdown listener
+     */
+    public ShutdownListener getShutdownListener() {
+        return shutdownListener;
+    }
+
+    /**
+     * Internal shutdown listener implementation
+     */
+    private class InternalShutdownListener implements ShutdownListener {
+        @Override
+        public void onShutdown() {
+            logger.info("AbstractConsumerInvoker received shutdown notification");
+            AbstractConsumerInvoker.stop();
+        }
+    }
 }

trpc-proto/trpc-proto-http/src/test/java/com/tencent/trpc/proto/http/HttpRpcClientTest.java

@@ -15,6 +15,7 @@
 import static com.tencent.trpc.transport.http.common.Constants.HTTP_SCHEME;
 
 import com.tencent.trpc.core.common.ConfigManager;
+import com.tencent.trpc.core.common.ShutdownListener;
 import com.tencent.trpc.core.common.config.BackendConfig;
 import com.tencent.trpc.core.common.config.ConsumerConfig;
 import com.tencent.trpc.core.common.config.GlobalConfig;
@@ -49,6 +50,7 @@
 import tests.service.impl1.GreeterJavaBeanServiceImpl;
 import tests.service.impl1.GreeterJsonServiceImpl1;
 import tests.service.impl1.GreeterServiceImpl1;
+import com.tencent.trpc.proto.http.client.AbstractConsumerInvoker;
 
 public class HttpRpcClientTest {
 
@@ -488,4 +490,37 @@ public void testHttpRpcAttachmentWithJavaBean() {
             backendConfig.stop();
         }
     }
+
+    @Test
+    public void testAbstractConsumerInvokerShutdownListener() {
+        BackendConfig backendConfig = new BackendConfig();
+        backendConfig.setName("serviceId");
+        backendConfig.setRequestTimeout(REQUEST_TIMEOUT);
+        backendConfig.setMaxConns(MAX_CONNECTIONS);
+        backendConfig.setNamingUrl("ip://127.0.0.1:18088");
+        backendConfig.setKeepAlive(false);
+        backendConfig.setConnsPerAddr(4);
+        backendConfig.setProtocol("http");
+
+        ConsumerConfig<GreeterService> consumerConfig = new ConsumerConfig<>();
+        consumerConfig.setServiceInterface(GreeterService.class);
+        consumerConfig.setBackendConfig(backendConfig);
+
+        try {
+            GreeterService proxy = consumerConfig.getProxy();
+            
+            // Test that the shutdown listener is properly registered
+            // We can't directly access the AbstractConsumerInvoker from the proxy,
+            // but we can test the static methods
+            AbstractConsumerInvoker.reset();
+            
+            // Test shutdown functionality
+            AbstractConsumerInvoker.stop();
+            
+            logger.info("AbstractConsumerInvoker shutdown listener test completed successfully");
+            
+        } finally {
+            backendConfig.stop();
+        }
+    }
 }