support sofa registry kubernetes

sofastack · Jan 26, 2024 · de944ef · de944ef
1 parent e31f08c
commit de944ef
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/...-kubernetes/src/main/java/com/alipay/sofa/rpc/registry/kubernetes/KubernetesRegistry.java b/...-kubernetes/src/main/java/com/alipay/sofa/rpc/registry/kubernetes/KubernetesRegistry.java
@@ -238,12 +238,14 @@ public void batchUnSubscribe(List<ConsumerConfig> configs) {
 
     @Override
     public void destroy() {
-        for (ProviderConfig providerConfig : providerInstances.keySet()) {
-            unRegister(providerConfig);
-        }
-        for (ConsumerConfig consumerConfig : consumerListeners.keySet()) {
-            unSubscribe(consumerConfig);
-        }
+        // unRegister provider
+        providerInstances.forEach((k, v) -> unRegister(k));
+        providerInstances.clear();
+
+        // unRegister consumer
+        consumerListeners.forEach((k, v) -> unSubscribe(k));
+        consumerListeners.clear();
+
         // close kubernetes client
         kubernetesClient.close();
     }