Skip to content

D
discovery
Commit 9d4d1042 by Nepxion
Options

增加本地三个配置到MetaData里

parent bec49db6
Showing with 33 additions and 3 deletions
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 {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment