Optimize registry type

jd-opensource · Jun 6, 2024 · 820e65d · 820e65d
1 parent e61d25d
commit 820e65d
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/.../jd/live/agent/plugin/registry/dubbo/v2_7/interceptor/InitServiceMetadataInterceptor.java b/.../jd/live/agent/plugin/registry/dubbo/v2_7/interceptor/InitServiceMetadataInterceptor.java
@@ -19,11 +19,9 @@
 import com.jd.live.agent.core.plugin.definition.InterceptorAdaptor;
 import com.jd.live.agent.governance.policy.PolicySupplier;
 import com.jd.live.agent.governance.policy.PolicyType;
-import org.apache.dubbo.config.AbstractInterfaceConfig;
-import org.apache.dubbo.config.AbstractReferenceConfig;
-import org.apache.dubbo.config.AbstractServiceConfig;
-import org.apache.dubbo.config.RegistryConfig;
+import org.apache.dubbo.config.*;
 
+import java.util.HashMap;
 import java.util.Map;
 
 import static org.apache.dubbo.common.constants.RegistryConstants.REGISTRY_TYPE_KEY;
@@ -60,6 +58,16 @@ public void onEnter(ExecutableContext ctx) {
             type = getRegistryType((AbstractReferenceConfig) config);
         }
         if ((type & REGISTRY_TYPE_SERVICE) > 0) {
+            if ((type & REGISTRY_TYPE_INTERFACE) == 0) {
+                // Only register with service mode
+                ApplicationConfig applicationConfig = config.getApplication();
+                Map<String, String> map = applicationConfig.getParameters();
+                if (map == null) {
+                    map = new HashMap<>();
+                    applicationConfig.setParameters(map);
+                }
+                map.put(REGISTRY_TYPE_KEY, SERVICE_REGISTRY_TYPE);
+            }
             policySupplier.subscribe(config.getApplication().getName(), PolicyType.SERVICE_POLICY);
         }
         if ((type & REGISTRY_TYPE_INTERFACE) > 0) {