Fix beans information lost parentId (#1320)

Signed-off-by: JermaineHua <crazyhzm@apache.org>
sofastack · May 14, 2024 · 6318683 · 6318683
1 parent d65c12b
commit 6318683
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/...fa-boot-actuator/src/main/java/com/alipay/sofa/boot/actuator/beans/IsleBeansEndpoint.java b/...fa-boot-actuator/src/main/java/com/alipay/sofa/boot/actuator/beans/IsleBeansEndpoint.java
@@ -57,19 +57,33 @@ public ApplicationBeans beans() {
         ApplicationBeans applicationBeans = super.beans();
         ApplicationRuntimeModel applicationRuntimeModel = context.getBean(
             SofaBootConstants.APPLICATION, ApplicationRuntimeModel.class);
-        Map<String, ContextBeans> moduleApplicationContexts = getModuleApplicationContexts(applicationRuntimeModel);
+
+        String springParentId = null;
+        for (Map.Entry<String, ContextBeans> entry : applicationBeans.getContexts().entrySet()) {
+            if (entry.getValue().getParentId() == null) {
+                springParentId = entry.getKey();
+                break;
+            }
+        }
+
+        Map<String, ContextBeans> moduleApplicationContexts = getModuleApplicationContexts(
+            applicationRuntimeModel, springParentId);
         applicationBeans.getContexts().putAll(moduleApplicationContexts);
         return applicationBeans;
     }
 
-    private Map<String, ContextBeans> getModuleApplicationContexts(ApplicationRuntimeModel applicationRuntimeModel) {
+    private Map<String, ContextBeans> getModuleApplicationContexts(ApplicationRuntimeModel applicationRuntimeModel,String springParentId) {
         Map<String, ContextBeans> contexts = new HashMap<>();
         List<DeploymentDescriptor> installedModules = applicationRuntimeModel.getInstalled();
         installedModules.forEach(descriptor -> {
             ApplicationContext applicationContext = descriptor.getApplicationContext();
+            String parentId = descriptor.getSpringParent();
+            if (parentId == null){
+                parentId = springParentId;
+            }
             if (applicationContext instanceof ConfigurableApplicationContext) {
                 ContextBeans contextBeans = describing((ConfigurableApplicationContext) applicationContext,
-                        descriptor.getSpringParent());
+                        parentId);
                 if (contextBeans != null) {
                     contexts.put(descriptor.getModuleName(), contextBeans);
                 }