Fix nacos reconnection failed issue because PayloadRegistry without i…

…nitializing due to incorrect classloader when creating nacos client. Signed-off-by: rztao <[email protected]>
sermant-io · Nov 17, 2024 · 904b8a1 · 904b8a1
1 parent f6f66b0
commit 904b8a1
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/...bo-registry-service/src/main/java/io/sermant/dubbo/registry/utils/NamingServiceUtils.java b/...bo-registry-service/src/main/java/io/sermant/dubbo/registry/utils/NamingServiceUtils.java
@@ -57,12 +57,16 @@ private NamingServiceUtils() {
     public static NamingService buildNamingService(Map<String, String> parameters, NacosRegisterConfig registerConfig,
         RegisterServiceCommonConfig commonConfig) {
         Properties nacosProperties = buildNacosProperties(parameters, registerConfig, commonConfig);
+        ClassLoader tempClassLoader = Thread.currentThread().getContextClassLoader();
+        Thread.currentThread().setContextClassLoader(NamingServiceUtils.class.getClassLoader());
         try {
             return NacosFactory.createNamingService(nacosProperties);
         } catch (NacosException e) {
             LOGGER.log(Level.SEVERE, String.format(Locale.ENGLISH, "create namingService failed: {%s}",
                     e.getErrMsg()), e);
             throw new IllegalStateException(e);
+        } finally {
+            Thread.currentThread().setContextClassLoader(tempClassLoader);
         }
     }
 

diff --git a/...istry-service/src/main/java/io/sermant/registry/service/register/NacosServiceManager.java b/...istry-service/src/main/java/io/sermant/registry/service/register/NacosServiceManager.java
@@ -104,11 +104,23 @@ private void buildNamingService(Properties properties) throws NacosException {
     }
 
     private NamingService createNewNamingService(Properties properties) throws NacosException {
-        return new NacosNamingService(properties);
+        ClassLoader tempClassLoader = Thread.currentThread().getContextClassLoader();
+        Thread.currentThread().setContextClassLoader(this.getClass().getClassLoader());
+        try {
+            return new NacosNamingService(properties);
+        } finally {
+            Thread.currentThread().setContextClassLoader(tempClassLoader);
+        }
     }
 
     private NamingMaintainService createNamingMaintainService(Properties properties) throws NacosException {
-        return new NacosNamingMaintainService(properties);
+        ClassLoader tempClassLoader = Thread.currentThread().getContextClassLoader();
+        Thread.currentThread().setContextClassLoader(this.getClass().getClassLoader());
+        try {
+            return new NacosNamingMaintainService(properties);
+        } finally {
+            Thread.currentThread().setContextClassLoader(tempClassLoader);
+        }
     }
 
     /**