Fix ut

Signed-off-by: crazyhzm <[email protected]>
CrazyHZM · Feb 4, 2024 · 404c5ed · 404c5ed
1 parent 806aba4
commit 404c5ed
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/...ig/dubbo-config-spring6/src/main/java/org/apache/dubbo/config/spring6/utils/AotUtils.java b/...ig/dubbo-config-spring6/src/main/java/org/apache/dubbo/config/spring6/utils/AotUtils.java
@@ -29,20 +29,21 @@
 
 public class AotUtils {
 
-    private static final Set<Class<?>> serializationTypeCache = new LinkedHashSet<>();
-
     private AotUtils() {}
 
     public static void registerSerializationForService(Class<?> serviceType, RuntimeHints hints) {
+        Set<Class<?>> serializationTypeCache = new LinkedHashSet<>();
         Arrays.stream(serviceType.getMethods()).forEach((method) -> {
             Arrays.stream(method.getParameterTypes())
-                    .forEach((parameterType) -> registerSerializationType(parameterType, hints));
+                    .forEach(
+                            (parameterType) -> registerSerializationType(parameterType, hints, serializationTypeCache));
 
-            registerSerializationType(method.getReturnType(), hints);
+            registerSerializationType(method.getReturnType(), hints, serializationTypeCache);
         });
     }
 
-    private static void registerSerializationType(Class<?> registerType, RuntimeHints hints) {
+    private static void registerSerializationType(
+            Class<?> registerType, RuntimeHints hints, Set<Class<?>> serializationTypeCache) {
         if (isPrimitive(registerType)) {
             hints.serialization().registerType(TypeReference.of(ClassUtils.getBoxedClass(registerType)));
             serializationTypeCache.add(registerType);
@@ -53,12 +54,12 @@ private static void registerSerializationType(Class<?> registerType, RuntimeHint
 
                 Arrays.stream(registerType.getDeclaredFields()).forEach((field -> {
                     if (!serializationTypeCache.contains(field.getType())) {
-                        registerSerializationType(field.getType(), hints);
+                        registerSerializationType(field.getType(), hints, serializationTypeCache);
                         serializationTypeCache.add(field.getType());
                     }
                 }));
 
-                registerSerializationType(registerType.getSuperclass(), hints);
+                registerSerializationType(registerType.getSuperclass(), hints, serializationTypeCache);
             }
         }
     }