fix:优雅关闭；单例序列化器

version5/krpc-common/src/main/java/common/serializer/myserializer/Serializer.java

@@ -20,16 +20,19 @@ public interface Serializer {
 
     int getType();
 
+    // 定义静态常量 serializerMap
+    static final Map<Integer, Serializer> serializerMap = new HashMap<>();
+
     // 使用 Map 存储序列化器
     static Serializer getSerializerByCode(int code) {
         // 静态映射，保证只初始化一次
-        Map<Integer, Serializer> serializerMap = new HashMap<>();
+        if(serializerMap.isEmpty()) {
             serializerMap.put(0, new ObjectSerializer());
             serializerMap.put(1, new JsonSerializer());
             serializerMap.put(2, new KryoSerializer());
             serializerMap.put(3, new HessianSerializer());
             serializerMap.put(4, new ProtostuffSerializer());
-
+        }
         return serializerMap.get(code); // 如果不存在，则返回 null
     }
 }

version5/krpc-consumer/src/main/java/com/kama/consumer/ConsumerTest.java

@@ -61,6 +61,7 @@ public class ConsumerTest {
 
         // Gracefully shutdown the executor service
         executorService.shutdown();
+        clientProxy.close();
     }
 
 }

version5/krpc-core/src/main/java/com/kama/client/proxy/ClientProxy.java

@@ -109,4 +109,11 @@ public class ClientProxy implements InvocationHandler {
         }
         return sb.toString();
     }
+
+    //关闭创建的资源
+    //注：如果在需要C-S保持长连接的场景下无需调用close方法
+    public void close(){
+        rpcClient.close();
+        serviceCenter.close();
+    }
 }

version5/krpc-core/src/main/java/com/kama/client/rpcclient/RpcClient.java

@@ -14,4 +14,5 @@ import common.message.RpcResponse;
 
 public interface RpcClient {
     RpcResponse sendRequest(RpcRequest request);
+    void close();
 }

version5/krpc-core/src/main/java/com/kama/client/rpcclient/impl/NettyRpcClient.java

@@ -68,7 +68,6 @@ public class NettyRpcClient implements RpcClient {
             // 其它场景也可以选择添加监听器的方式来异步获取结果 channelFuture.addListener...
             AttributeKey<RpcResponse> key = AttributeKey.valueOf("RPCResponse");
             RpcResponse response = channel.attr(key).get();
-
             if (response == null) {
                 log.error("服务响应为空，可能是请求失败或超时");
                 return RpcResponse.fail("服务响应为空");
@@ -82,14 +81,13 @@ public class NettyRpcClient implements RpcClient {
         } catch (Exception e) {
             log.error("发送请求时发生异常: {}", e.getMessage(), e);
         } finally {
-            // 连接断开后，优雅地关闭 Netty 资源
-            shutdown();
+            //
         }
         return RpcResponse.fail("请求失败");
     }
 
     // 优雅关闭 Netty 资源
-    private void shutdown() {
+    public void close() {
         try {
             if (eventLoopGroup != null) {
                 eventLoopGroup.shutdownGracefully().sync();

version5/krpc-core/src/main/java/com/kama/client/rpcclient/impl/SimpleSocketRpcClient.java

@@ -54,4 +54,9 @@ public class SimpleSocketRpcClient implements RpcClient {
 
         return response;
     }
+
+    @Override
+    public void close() {
+
+    }
 }

version5/krpc-core/src/main/java/com/kama/client/servicecenter/ServiceCenter.java

@@ -19,4 +19,7 @@ public interface ServiceCenter {
 
     //判断是否可重试
     boolean checkRetry(InetSocketAddress serviceAddress, String methodSignature);
+
+    //关闭客户端
+    void close();
 }

version5/krpc-core/src/main/java/com/kama/client/servicecenter/ZKServiceCenter.java

@@ -104,6 +104,11 @@ public class ZKServiceCenter implements ServiceCenter {
         return retryServiceCache.contains(methodSignature);
     }
 
+    @Override
+    public void close() {
+        client.close();
+    }
+
     // 将InetSocketAddress解析为格式为ip:port的字符串
     private String getServiceAddress(InetSocketAddress serverAddress){
         return serverAddress.getHostName() + ":" + serverAddress.getPort();