Auto module export with reentrant protection

Signed-off-by: JermaineHua <[email protected]>
sofastack · Sep 23, 2024 · e95642b · e95642b
1 parent 6d844ab
commit e95642b
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/sofa-boot-project/sofa-boot/src/main/java/com/alipay/sofa/boot/util/ModuleUtil.java b/sofa-boot-project/sofa-boot/src/main/java/com/alipay/sofa/boot/util/ModuleUtil.java
@@ -23,22 +23,29 @@
 import java.lang.reflect.Field;
 import java.util.Map;
 import java.util.Optional;
+import java.util.concurrent.atomic.AtomicBoolean;
 
 /**
  * @author huazhongming
  * @since 4.4.0
  */
 public class ModuleUtil {
 
-    private static final Logger              LOGGER = SofaBootLoggerFactory
-                                                        .getLogger(ModuleUtil.class);
+    private static final Logger              LOGGER     = SofaBootLoggerFactory
+                                                            .getLogger(ModuleUtil.class);
 
     private static final MethodHandle        implAddOpensToAllUnnamed;
+
     private static final MethodHandle        implAddOpens;
+
     private static final MethodHandle        implAddExportsToAllUnnamed;
+
     private static final MethodHandle        implAddExports;
+
     private static final Map<String, Module> nameToModules;
 
+    private static final AtomicBoolean       isExported = new AtomicBoolean(false);
+
     static {
         implAddOpensToAllUnnamed = createModuleMethodHandle("implAddOpensToAllUnnamed",
             String.class);
@@ -54,13 +61,14 @@ public class ModuleUtil {
      */
     public static void exportAllJDKModulePackageToAll() {
         try {
-            if (nameToModules != null) {
+            if (isExported.compareAndSet(false,true) && nameToModules != null) {
                 nameToModules.forEach((name, module) -> module.getPackages().forEach(pkgName -> {
                     if (isJDKModulePackage(pkgName)) {
                         addOpensToAll(module, pkgName);
                         addExportsToAll(module, pkgName);
                     }
                 }));
+                isExported.compareAndExchange(false,true);
             }
         } catch (Throwable t) {
             LOGGER.error("Failed to export all JDK module package to all", t);
@@ -76,8 +84,7 @@ private static boolean isJDKModulePackage(String modulePackageName) {
      */
     public static void exportAllModulePackageToAll() {
         try {
-            Map<String, Module> nameToModules = getNameToModule();
-            if (nameToModules != null) {
+            if (isExported.compareAndSet(false,true) && nameToModules != null) {
                 nameToModules.forEach((name, module) -> module.getPackages().forEach(pkgName -> {
                     addOpensToAll(module, pkgName);
                     addExportsToAll(module, pkgName);