增加本地三个配置到MetaData里

discovery-plugin-framework-consul/src/main/java/com/nepxion/discovery/plugin/framework/context/ConsulApplicationContextInitializer.java

@@ -9,10 +9,13 @@ package com.nepxion.discovery.plugin.framework.context;
  * @version 1.0
  */
 
+import java.util.List;
+
 import org.springframework.beans.BeansException;
 import org.springframework.cloud.consul.discovery.ConsulDiscoveryProperties;
 import org.springframework.cloud.consul.serviceregistry.ConsulServiceRegistry;
 import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.core.env.ConfigurableEnvironment;
 
 import com.nepxion.discovery.plugin.framework.constant.ConsulConstant;
 import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
@@ -26,9 +29,16 @@ public class ConsulApplicationContextInitializer extends PluginApplicationContex
 
             return new ConsulServiceRegistryDecorator(consulServiceRegistry, applicationContext);
         } else if (bean instanceof ConsulDiscoveryProperties) {
+            ConfigurableEnvironment environment = applicationContext.getEnvironment();
+
             ConsulDiscoveryProperties consulDiscoveryProperties = (ConsulDiscoveryProperties) bean;
             consulDiscoveryProperties.setPreferIpAddress(true);
-            consulDiscoveryProperties.getTags().add(PluginConstant.DISCOVERY_PLUGIN + "=" + ConsulConstant.DISCOVERY_PLUGIN);
+
+            List<String> tags = consulDiscoveryProperties.getTags();
+            tags.add(PluginConstant.DISCOVERY_PLUGIN + "=" + ConsulConstant.DISCOVERY_PLUGIN);
+            tags.add(PluginConstant.SPRING_APPLICATION_REGISTER_CONTROL_ENABLED + "=" + PluginContextAware.isRegisterControlEnabled(environment));
+            tags.add(PluginConstant.SPRING_APPLICATION_DISCOVERY_CONTROL_ENABLED + "=" + PluginContextAware.isDiscoveryControlEnabled(environment));
+            tags.add(PluginConstant.SPRING_APPLICATION_CONFIG_REST_CONTROL_ENABLED + "=" + PluginContextAware.isConfigRestControlEnabled(environment));
 
             return bean;
         } else {

discovery-plugin-framework-eureka/src/main/java/com/nepxion/discovery/plugin/framework/context/EurekaApplicationContextInitializer.java

@@ -9,10 +9,13 @@ package com.nepxion.discovery.plugin.framework.context;
  * @version 1.0
  */
 
+import java.util.Map;
+
 import org.springframework.beans.BeansException;
 import org.springframework.cloud.netflix.eureka.EurekaInstanceConfigBean;
 import org.springframework.cloud.netflix.eureka.serviceregistry.EurekaServiceRegistry;
 import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.core.env.ConfigurableEnvironment;
 
 import com.nepxion.discovery.plugin.framework.constant.EurekaConstant;
 import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
@@ -26,9 +29,16 @@ public class EurekaApplicationContextInitializer extends PluginApplicationContex
 
             return new EurekaServiceRegistryDecorator(eurekaServiceRegistry, applicationContext);
         } else if (bean instanceof EurekaInstanceConfigBean) {
+            ConfigurableEnvironment environment = applicationContext.getEnvironment();
+
             EurekaInstanceConfigBean eurekaInstanceConfig = (EurekaInstanceConfigBean) bean;
             eurekaInstanceConfig.setPreferIpAddress(true);
-            eurekaInstanceConfig.getMetadataMap().put(PluginConstant.DISCOVERY_PLUGIN, EurekaConstant.DISCOVERY_PLUGIN);
+
+            Map<String, String> metadataMap = eurekaInstanceConfig.getMetadataMap();
+            metadataMap.put(PluginConstant.DISCOVERY_PLUGIN, EurekaConstant.DISCOVERY_PLUGIN);
+            metadataMap.put(PluginConstant.SPRING_APPLICATION_REGISTER_CONTROL_ENABLED, PluginContextAware.isRegisterControlEnabled(environment).toString());
+            metadataMap.put(PluginConstant.SPRING_APPLICATION_DISCOVERY_CONTROL_ENABLED, PluginContextAware.isDiscoveryControlEnabled(environment).toString());
+            metadataMap.put(PluginConstant.SPRING_APPLICATION_CONFIG_REST_CONTROL_ENABLED, PluginContextAware.isConfigRestControlEnabled(environment).toString());
 
             return bean;
         } else {

discovery-plugin-framework-zookeeper/src/main/java/com/nepxion/discovery/plugin/framework/context/ZookeeperApplicationContextInitializer.java

@@ -9,10 +9,13 @@ package com.nepxion.discovery.plugin.framework.context;
  * @version 1.0
  */
 
+import java.util.Map;
+
 import org.springframework.beans.BeansException;
 import org.springframework.cloud.zookeeper.discovery.ZookeeperDiscoveryProperties;
 import org.springframework.cloud.zookeeper.serviceregistry.ZookeeperServiceRegistry;
 import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.core.env.ConfigurableEnvironment;
 
 import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
 import com.nepxion.discovery.plugin.framework.constant.ZookeeperConstant;
@@ -26,9 +29,16 @@ public class ZookeeperApplicationContextInitializer extends PluginApplicationCon
 
             return new ZookeeperServiceRegistryDecorator(zookeeperServiceRegistry, applicationContext);
         } else if (bean instanceof ZookeeperDiscoveryProperties) {
+            ConfigurableEnvironment environment = applicationContext.getEnvironment();
+
             ZookeeperDiscoveryProperties zookeeperDiscoveryProperties = (ZookeeperDiscoveryProperties) bean;
             zookeeperDiscoveryProperties.setPreferIpAddress(true);
-            zookeeperDiscoveryProperties.getMetadata().put(PluginConstant.DISCOVERY_PLUGIN, ZookeeperConstant.DISCOVERY_PLUGIN);
+
+            Map<String, String> metadata = zookeeperDiscoveryProperties.getMetadata();
+            metadata.put(PluginConstant.DISCOVERY_PLUGIN, ZookeeperConstant.DISCOVERY_PLUGIN);
+            metadata.put(PluginConstant.SPRING_APPLICATION_REGISTER_CONTROL_ENABLED, PluginContextAware.isRegisterControlEnabled(environment).toString());
+            metadata.put(PluginConstant.SPRING_APPLICATION_DISCOVERY_CONTROL_ENABLED, PluginContextAware.isDiscoveryControlEnabled(environment).toString());
+            metadata.put(PluginConstant.SPRING_APPLICATION_CONFIG_REST_CONTROL_ENABLED, PluginContextAware.isConfigRestControlEnabled(environment).toString());
 
             return bean;
         } else {