fix:shutdown grpc connection gracefully.

Author: SkyeBeFreeman

polaris-plugins/polaris-plugins-connector/connector-polaris-grpc/src/main/java/com/tencent/polaris/plugins/connector/grpc/Connection.java

@@ -128,14 +128,31 @@ public void closeConnection() {
             if (ref.get() <= 0 && !closed) {
                 LOG.info("connection {}: closed", connID);
                 closed = true;
-                ManagedChannel shutdownChan = channel.shutdownNow();
-                if (null == shutdownChan) {
-                    return;
+                // Gracefully shutdown the gRPC managed-channel.
+                if (channel != null && !channel.isShutdown()) {
+                    try {
+                        channel.shutdown();
+                        if (!channel.awaitTermination(1, TimeUnit.SECONDS)) {
+                            LOG.warn("Timed out gracefully shutting down connection: {}. ", connID);
+                        }
+                    } catch (Exception e) {
+                        LOG.error("Unexpected exception while waiting for channel {} gracefully termination", connID, e);
+                    }
                 }
-                try {
-                    shutdownChan.awaitTermination(100, TimeUnit.MILLISECONDS);
-                } catch (InterruptedException e) {
-                    LOG.error(String.format("interrupted while closing connection %s", connID), e);
+
+                // Forcefully shutdown if still not terminated.
+                if (channel != null && !channel.isTerminated()) {
+                    try {
+                        channel.shutdownNow();
+                        if (!channel.awaitTermination(100, TimeUnit.MILLISECONDS)) {
+                            LOG.warn("Timed out forcefully shutting down connection: {}. ", connID);
+                        }
+                        LOG.debug("Success to forcefully shutdown connection: {}. ", connID);
+                    } catch (Exception e) {
+                        LOG.error("Unexpected exception while waiting for channel {} forcefully termination", connID, e);
+                    }
+                } else {
+                    LOG.debug("Success to gracefully shutdown connection: {}. ", connID);
                 }
             }
         }

polaris-plugins/polaris-plugins-connector/connector-polaris-grpc/src/main/java/com/tencent/polaris/plugins/connector/grpc/SpecStreamClient.java

@@ -264,9 +264,9 @@ public void onNext(ResponseProto.DiscoverResponse response) {
             return;
         }
         if (updateTask.getTaskType() == Type.FIRST) {
-            LOG.info("[ServerConnector]receive response for {}", serviceEventKey);
+            LOG.info("[ServerConnector]request(id={}) receive response for {}", getReqId(), serviceEventKey);
         } else {
-            LOG.debug("[ServerConnector]receive response for {}", serviceEventKey);
+            LOG.debug("[ServerConnector]request(id={}) receive response for {}", getReqId(), serviceEventKey);
         }
         boolean svcDeleted = updateTask.notifyServerEvent(new ServerEvent(serviceEventKey, response, null));
         if (!svcDeleted) {

pom.xml

@@ -64,7 +64,7 @@
 
     <properties>
         <!-- Project revision -->
-        <revision>1.12.4</revision>
+        <revision>1.12.5</revision>
 
         <timestamp>${maven.build.timestamp}</timestamp>
         <skip.maven.deploy>false</skip.maven.deploy>