Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.util.concurrent.locks.ReentrantLock; + +import javax.annotation.PostConstruct; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; + +import com.nepxion.discovery.plugin.entity.DiscoveryEntity; +import com.nepxion.discovery.plugin.strategy.FilterStrategy; +import com.nepxion.discovery.plugin.strategy.VersionStrategy; + +@Configuration +public class DiscoveryPluginConfig { + @Autowired + private DiscoveryPluginConfigLoader discoveryPluginConfigLoader; + + @Bean + public DiscoveryPluginConfigParser discoveryPluginConfigParser() { + return new DiscoveryPluginConfigParser(); + } + + @Bean + public DiscoveryPluginConfigLoader discoveryPluginConfigLoader() { + return new DiscoveryPluginConfigLoader(); + } + + @Bean + public DiscoveryEntity discoveryEntity() { + return new DiscoveryEntity(); + } + + @Bean + public ReentrantLock reentrantLock() { + return new ReentrantLock(); + } + + @Bean + public FilterStrategy filterStrategy() { + return new FilterStrategy(); + } + + @Bean + public VersionStrategy versionStrategy() { + return new VersionStrategy(); + } + + @PostConstruct + public void initialize() { + discoveryPluginConfigLoader.initialize(); + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/config/DiscoveryPluginConfigLoader.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/config/DiscoveryPluginConfigLoader.java new file mode 100644 index 0000000000..704b520115 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/config/DiscoveryPluginConfigLoader.java @@ -0,0 +1,93 @@ +package com.nepxion.discovery.plugin.config; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.IOException; +import java.io.InputStream; + +import org.apache.commons.io.IOUtils; +import org.dom4j.DocumentException; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; + +import com.google.common.eventbus.Subscribe; +import com.nepxion.discovery.plugin.constant.DiscoveryPluginConstant; +import com.nepxion.discovery.plugin.exception.DiscoveryPluginException; +import com.nepxion.discovery.plugin.loader.FileLoader; +import com.nepxion.eventbus.annotation.EventBus; +import com.nepxion.eventbus.core.Event; + +@EventBus +public class DiscoveryPluginConfigLoader { + private static final Logger LOG = LoggerFactory.getLogger(DiscoveryPluginConfigLoader.class); + + @Value("${" + DiscoveryPluginConstant.SPRING_APPLICATION_DISCOVERY_REMOTE_CONFIG_ENABLED + ":false}") + private Boolean remoteConfigEnabled; + + @Autowired + private FileLoader fileLoader; + + @Autowired + private DiscoveryPluginConfigParser discoveryPluginConfigParser; + + public void initialize() throws DiscoveryPluginException { + LOG.info("********** {} config starts to initialize **********", remoteConfigEnabled ? "Remote" : "Local"); + + InputStream inputStream = null; + try { + if (remoteConfigEnabled) { + inputStream = fileLoader.getRemoteInputStream(); + } else { + inputStream = fileLoader.getLocalInputStream(); + } + parse(inputStream); + } catch (IOException e) { + throw new DiscoveryPluginException(e); + } catch (DocumentException e) { + throw new DiscoveryPluginException(e); + } finally { + if (inputStream != null) { + IOUtils.closeQuietly(inputStream); + } + } + } + + @Subscribe + public void refresh(Event event) { + if (!remoteConfigEnabled) { + return; + } + + LOG.info("********** Remote config change has been retrieved **********"); + + InputStream inputStream = (InputStream) event.getSource(); + try { + parse(inputStream); + } catch (IOException e) { + throw new DiscoveryPluginException(e); + } catch (DocumentException e) { + throw new DiscoveryPluginException(e); + } finally { + if (inputStream != null) { + IOUtils.closeQuietly(inputStream); + } + } + } + + private void parse(InputStream inputStream) throws DocumentException, IOException { + if (inputStream == null) { + throw new DiscoveryPluginException("Failed to load " + (remoteConfigEnabled ? "remote" : "local") + " config, no input stream returns"); + } + + discoveryPluginConfigParser.parse(inputStream); + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/config/DiscoveryPluginConfigParser.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/config/DiscoveryPluginConfigParser.java new file mode 100644 index 0000000000..45932343b0 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/config/DiscoveryPluginConfigParser.java @@ -0,0 +1,139 @@ +package com.nepxion.discovery.plugin.config; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.util.Iterator; +import java.util.List; +import java.util.Map; +import java.util.concurrent.locks.ReentrantLock; + +import org.apache.commons.lang3.StringUtils; +import org.dom4j.Element; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; + +import com.nepxion.discovery.plugin.constant.DiscoveryPluginConstant; +import com.nepxion.discovery.plugin.entity.ConsumerEntity; +import com.nepxion.discovery.plugin.entity.DiscoveryEntity; +import com.nepxion.discovery.plugin.entity.FilterEntity; +import com.nepxion.discovery.plugin.entity.FilterType; +import com.nepxion.discovery.plugin.entity.VersionEntity; +import com.nepxion.discovery.plugin.exception.DiscoveryPluginException; +import com.nepxion.discovery.plugin.xml.Dom4JParser; + +public class DiscoveryPluginConfigParser extends Dom4JParser { + private static final Logger LOG = LoggerFactory.getLogger(DiscoveryPluginConfigParser.class); + + @Autowired + private DiscoveryEntity discoveryEntity; + + @Autowired + private ReentrantLock reentrantLock; + + @SuppressWarnings("rawtypes") + @Override + protected void parseRoot(Element element) { + LOG.info("Start to parse discovery.xml..."); + + int filterElementCount = element.elements(DiscoveryPluginConstant.FILTER_ELEMENT_NAME).size(); + if (filterElementCount > 1) { + throw new DiscoveryPluginException("The count of element [" + DiscoveryPluginConstant.FILTER_ELEMENT_NAME + "] can't be more than 1"); + } + + int versionElementCount = element.elements(DiscoveryPluginConstant.VERSION_ELEMENT_NAME).size(); + if (versionElementCount > 1) { + throw new DiscoveryPluginException("The count of element [" + DiscoveryPluginConstant.VERSION_ELEMENT_NAME + "] can't be more than 1"); + } + + FilterEntity filterEntity = new FilterEntity(); + VersionEntity versionEntity = new VersionEntity(); + for (Iterator elementIterator = element.elementIterator(); elementIterator.hasNext();) { + Object childElementObject = elementIterator.next(); + if (childElementObject instanceof Element) { + Element childElement = (Element) childElementObject; + + if (StringUtils.equals(childElement.getName(), DiscoveryPluginConstant.FILTER_ELEMENT_NAME)) { + parseFilter(childElement, filterEntity); + } else if (StringUtils.equals(childElement.getName(), DiscoveryPluginConstant.VERSION_ELEMENT_NAME)) { + parseVersion(childElement, versionEntity); + } + } + } + + try { + reentrantLock.lock(); + + discoveryEntity.setFilterEntity(filterEntity); + discoveryEntity.setVersionEntity(versionEntity); + } finally { + reentrantLock.unlock(); + } + + LOG.info("Discovery entity is {}", discoveryEntity); + } + + @SuppressWarnings("rawtypes") + private void parseFilter(Element element, FilterEntity filterEntity) { + String filterType = element.attribute(DiscoveryPluginConstant.FILTER_TYPE_ATTRIBUTE_NAME).getData().toString().trim(); + String globalFilterValue = element.attribute(DiscoveryPluginConstant.FILTER_VALUE_ATTRIBUTE_NAME).getData().toString().trim(); + filterEntity.setFilterType(FilterType.fromString(filterType)); + filterEntity.setFilterValue(globalFilterValue); + + MapTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +public class DiscoveryPluginConstant { + public static final String SPRING_APPLICATION_DISCOVERY_FILTER_ENABLED = "spring.application.discovery.filter.enabled"; + public static final String SPRING_APPLICATION_DISCOVERY_VERSION_ENABLED = "spring.application.discovery.version.enabled"; + public static final String SPRING_APPLICATION_DISCOVERY_REMOTE_CONFIG_ENABLED = "spring.application.discovery.remote.config.enabled"; + + public static final String SPRING_APPLICATION_NAME = "spring.application.name"; + public static final String EUREKA_METADATA_VERSION = "version"; + + public static final String DISCOVERY_ELEMENT_NAME = "discovery"; + public static final String FILTER_ELEMENT_NAME = "filter"; + public static final String SERVICE_ELEMENT_NAME = "service"; + public static final String FILTER_TYPE_ATTRIBUTE_NAME = "filter-type"; + public static final String FILTER_VALUE_ATTRIBUTE_NAME = "filter-value"; + public static final String SERVICE_NAME_ATTRIBUTE_NAME = "service-name"; + public static final String VERSION_ELEMENT_NAME = "version"; + public static final String CONSUMER_ELEMENT_NAME = "consumer"; + public static final String PROVIDER_ELEMENT_NAME = "provider"; + public static final String VERSION_VALUE_NAME_ATTRIBUTE_NAME = "version-value"; + + public static final String SEPARATE = ";"; +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/context/DiscoveryApplicationContextInitializer.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/context/DiscoveryApplicationContextInitializer.java new file mode 100644 index 0000000000..abc39af762 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/context/DiscoveryApplicationContextInitializer.java @@ -0,0 +1,45 @@ +package com.nepxion.discovery.plugin.context; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import org.springframework.beans.BeansException; +import org.springframework.beans.factory.config.InstantiationAwareBeanPostProcessorAdapter; +import org.springframework.cloud.client.discovery.DiscoveryClient; +import org.springframework.cloud.netflix.eureka.EurekaInstanceConfigBean; +import org.springframework.cloud.netflix.eureka.serviceregistry.EurekaServiceRegistry; +import org.springframework.context.ApplicationContextInitializer; +import org.springframework.context.ConfigurableApplicationContext; + +import com.nepxion.discovery.plugin.decorator.DiscoveryClientDecorator; +import com.nepxion.discovery.plugin.decorator.EurekaServiceRegistryDecorator; + +public class DiscoveryApplicationContextInitializer implements ApplicationContextInitializerTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.util.List; + +import org.springframework.cloud.client.ServiceInstance; +import org.springframework.cloud.client.discovery.DiscoveryClient; +import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.core.env.ConfigurableEnvironment; + +import com.nepxion.discovery.plugin.constant.DiscoveryPluginConstant; +import com.nepxion.discovery.plugin.strategy.VersionStrategy; + +public class DiscoveryClientDecorator implements DiscoveryClient { + private DiscoveryClient discoveryClient; + private ConfigurableApplicationContext applicationContext; + private ConfigurableEnvironment environment; + + public DiscoveryClientDecorator(DiscoveryClient discoveryClient, ConfigurableApplicationContext applicationContext) { + this.discoveryClient = discoveryClient; + this.applicationContext = applicationContext; + this.environment = applicationContext.getEnvironment(); + } + + @Override + public String description() { + return discoveryClient.description(); + } + + @Deprecated + @Override + public ServiceInstance getLocalServiceInstance() { + return discoveryClient.getLocalServiceInstance(); + } + + @Override + public ListTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import org.springframework.cloud.client.serviceregistry.ServiceRegistry; +import org.springframework.cloud.netflix.eureka.serviceregistry.EurekaRegistration; +import org.springframework.cloud.netflix.eureka.serviceregistry.EurekaServiceRegistry; +import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.core.env.ConfigurableEnvironment; + +import com.nepxion.discovery.plugin.constant.DiscoveryPluginConstant; +import com.nepxion.discovery.plugin.strategy.FilterStrategy; + +public class EurekaServiceRegistryDecorator extends EurekaServiceRegistry { + private ServiceRegistryTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.Serializable; +import java.util.LinkedHashMap; +import java.util.Map; + +import org.apache.commons.lang3.builder.EqualsBuilder; +import org.apache.commons.lang3.builder.HashCodeBuilder; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +public class ConsumerEntity implements Serializable { + private static final long serialVersionUID = -2976565258172148792L; + + private String serviceName; + private MapTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.Serializable; + +import org.apache.commons.lang3.builder.EqualsBuilder; +import org.apache.commons.lang3.builder.HashCodeBuilder; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +public class DiscoveryEntity implements Serializable { + private static final long serialVersionUID = 7079024435084528751L; + + private FilterEntity filterEntity; + private VersionEntity versionEntity; + + public FilterEntity getFilterEntity() { + return filterEntity; + } + + public void setFilterEntity(FilterEntity filterEntity) { + this.filterEntity = filterEntity; + } + + public VersionEntity getVersionEntity() { + return versionEntity; + } + + public void setVersionEntity(VersionEntity versionEntity) { + this.versionEntity = versionEntity; + } + + @Override + public int hashCode() { + return HashCodeBuilder.reflectionHashCode(this); + } + + @Override + public boolean equals(Object object) { + return EqualsBuilder.reflectionEquals(this, object); + } + + @Override + public String toString() { + return ToStringBuilder.reflectionToString(this, ToStringStyle.MULTI_LINE_STYLE); + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/entity/FilterEntity.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/entity/FilterEntity.java new file mode 100644 index 0000000000..b819c6d862 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/entity/FilterEntity.java @@ -0,0 +1,70 @@ +package com.nepxion.discovery.plugin.entity; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.Serializable; +import java.util.LinkedHashMap; +import java.util.Map; + +import org.apache.commons.lang3.builder.EqualsBuilder; +import org.apache.commons.lang3.builder.HashCodeBuilder; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +public class FilterEntity implements Serializable { + private static final long serialVersionUID = -2097322826969006191L; + + private FilterType filterType; + private String filterValue; + private MapTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +public enum FilterType { + BLACKLIST("BLACKLIST", "黑名单"), + WHITELIST("WHITELIST", "白名单"); + + private String value; + private String description; + + private FilterType(String value, String description) { + this.value = value; + this.description = description; + } + + public String getValue() { + return value; + } + + public String getDescription() { + return description; + } + + public static FilterType fromString(String value) { + for (FilterType type : FilterType.values()) { + if (type.getValue().equalsIgnoreCase(value)) { + return type; + } + } + + throw new IllegalArgumentException("No matched type with value=" + value); + } + + @Override + public String toString() { + return value; + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/entity/VersionEntity.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/entity/VersionEntity.java new file mode 100644 index 0000000000..07bde344dd --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/entity/VersionEntity.java @@ -0,0 +1,52 @@ +package com.nepxion.discovery.plugin.entity; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.Serializable; +import java.util.ArrayList; +import java.util.List; + +import org.apache.commons.lang3.builder.EqualsBuilder; +import org.apache.commons.lang3.builder.HashCodeBuilder; +import org.apache.commons.lang3.builder.ToStringBuilder; +import org.apache.commons.lang3.builder.ToStringStyle; + +public class VersionEntity implements Serializable { + private static final long serialVersionUID = 6281838121286637807L; + + private ListTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +public class DiscoveryPluginException extends RuntimeException { + private static final long serialVersionUID = 7975167663357170655L; + + public DiscoveryPluginException() { + super(); + } + + public DiscoveryPluginException(String message) { + super(message); + } + + public DiscoveryPluginException(String message, Throwable cause) { + super(message, cause); + } + + public DiscoveryPluginException(Throwable cause) { + super(cause); + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/loader/AbstractFileLoader.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/loader/AbstractFileLoader.java new file mode 100644 index 0000000000..0c53d17849 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/loader/AbstractFileLoader.java @@ -0,0 +1,36 @@ +package com.nepxion.discovery.plugin.loader; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.IOException; +import java.io.InputStream; + +import org.apache.commons.lang3.StringUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.ApplicationContext; + +public abstract class AbstractFileLoader implements FileLoader { + @Autowired + private ApplicationContext applicationContext; + + @Override + public InputStream getLocalInputStream() throws IOException { + String localContextPath = getLocalContextPath(); + if (StringUtils.isEmpty(localContextPath)) { + return null; + } + + String localFilePath = applicationContext.getEnvironment().resolvePlaceholders(localContextPath); + + return applicationContext.getResource(localFilePath).getInputStream(); + } + + protected abstract String getLocalContextPath(); +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/loader/FileLoader.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/loader/FileLoader.java new file mode 100644 index 0000000000..43d9319cb2 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/loader/FileLoader.java @@ -0,0 +1,19 @@ +package com.nepxion.discovery.plugin.loader; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.IOException; +import java.io.InputStream; + +public interface FileLoader { + InputStream getLocalInputStream() throws IOException; + + InputStream getRemoteInputStream() throws IOException; +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/strategy/FilterStrategy.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/strategy/FilterStrategy.java new file mode 100644 index 0000000000..9f152801dd --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/strategy/FilterStrategy.java @@ -0,0 +1,44 @@ +package com.nepxion.discovery.plugin.strategy; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.util.Map; + +import org.apache.commons.lang3.StringUtils; +import org.springframework.beans.factory.annotation.Autowired; + +import com.nepxion.discovery.plugin.constant.DiscoveryPluginConstant; +import com.nepxion.discovery.plugin.entity.DiscoveryEntity; +import com.nepxion.discovery.plugin.entity.FilterEntity; +import com.nepxion.discovery.plugin.exception.DiscoveryPluginException; + +public class FilterStrategy { + @Autowired + private DiscoveryEntity discoveryEntity; + + public void apply(String serviceId, String ipAddress) { + FilterEntity filterEntity = discoveryEntity.getFilterEntity(); + String globalFilterValue = filterEntity.getFilterValue(); + validate(globalFilterValue, ipAddress); + + MapTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.util.Arrays; +import java.util.Iterator; +import java.util.List; +import java.util.Map; +import java.util.concurrent.locks.ReentrantLock; + +import org.apache.commons.lang.StringUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.cloud.client.ServiceInstance; + +import com.nepxion.discovery.plugin.constant.DiscoveryPluginConstant; +import com.nepxion.discovery.plugin.entity.ConsumerEntity; +import com.nepxion.discovery.plugin.entity.DiscoveryEntity; +import com.nepxion.discovery.plugin.entity.VersionEntity; + +public class VersionStrategy { + @Autowired + private DiscoveryEntity discoveryEntity; + + @Autowired + private ReentrantLock reentrantLock; + + public void apply(String consumerServiceId, String providerServiceId, ListTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import org.springframework.beans.factory.support.AbstractBeanDefinition; +import org.springframework.beans.factory.support.BeanDefinitionBuilder; +import org.springframework.beans.factory.support.BeanDefinitionRegistry; +import org.springframework.context.ConfigurableApplicationContext; + +public class BeanRegisterUtil { + public static BeanDefinitionRegistry getRegistry(ConfigurableApplicationContext applicationContext) { + return (BeanDefinitionRegistry) applicationContext.getBeanFactory(); + } + + public static void registerBean(BeanDefinitionRegistry beanDefinitionRegistry, String beanName, String beanClassName) { + BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(beanClassName); + AbstractBeanDefinition beanDefinition = beanDefinitionBuilder.getBeanDefinition(); + beanDefinition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_TYPE); + beanDefinitionRegistry.registerBeanDefinition(beanName, beanDefinition); + } + + public static void registerBean(BeanDefinitionRegistry beanDefinitionRegistry, String beanName, Class beanClass) { + BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(beanClass); + AbstractBeanDefinition beanDefinition = beanDefinitionBuilder.getBeanDefinition(); + beanDefinition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_TYPE); + beanDefinitionRegistry.registerBeanDefinition(beanName, beanDefinition); + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JConstant.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JConstant.java new file mode 100644 index 0000000000..6c1526d2c7 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JConstant.java @@ -0,0 +1,15 @@ +package com.nepxion.discovery.plugin.xml; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +public class Dom4JConstant { + public static final String ENCODING_UTF_8 = "UTF-8"; + public static final String ENCODING_ISO_8859_1 = "ISO-8859-1"; +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JParser.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JParser.java new file mode 100644 index 0000000000..de6652937b --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JParser.java @@ -0,0 +1,65 @@ +package com.nepxion.discovery.plugin.xml; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.File; +import java.io.IOException; +import java.io.InputStream; +import java.io.UnsupportedEncodingException; + +import org.dom4j.Document; +import org.dom4j.DocumentException; +import org.dom4j.Element; + +public abstract class Dom4JParser { + public void parse(String text) throws DocumentException { + Document document = Dom4JReader.getDocument(text); + + parse(document); + } + + public void parseFormat(String text) throws DocumentException, UnsupportedEncodingException { + Document document = Dom4JReader.getFormatDocument(text); + + parse(document); + } + + public void parse(File file) throws DocumentException, IOException, UnsupportedEncodingException { + Document document = Dom4JReader.getDocument(file); + + parse(document); + } + + public void parseFormat(File file) throws DocumentException, IOException, UnsupportedEncodingException { + Document document = Dom4JReader.getFormatDocument(file); + + parse(document); + } + + public void parse(InputStream inputStream) throws DocumentException, IOException { + Document document = Dom4JReader.getDocument(inputStream); + + parse(document); + } + + public void parseFormat(InputStream inputStream) throws DocumentException, IOException, UnsupportedEncodingException { + Document document = Dom4JReader.getFormatDocument(inputStream); + + parse(document); + } + + public void parse(Document document) { + Element rootElement = document.getRootElement(); + + parseRoot(rootElement); + } + + protected abstract void parseRoot(Element element); +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JReader.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JReader.java new file mode 100644 index 0000000000..df254eca4e --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JReader.java @@ -0,0 +1,123 @@ +package com.nepxion.discovery.plugin.xml; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.File; +import java.io.FileInputStream; +import java.io.IOException; +import java.io.InputStream; +import java.io.InputStreamReader; +import java.io.Reader; +import java.io.UnsupportedEncodingException; +import java.net.URL; + +import org.dom4j.Document; +import org.dom4j.DocumentException; +import org.dom4j.DocumentHelper; +import org.dom4j.io.SAXReader; +import org.xml.sax.InputSource; + +public class Dom4JReader { + public static Document getDocument(String text) throws DocumentException { + return DocumentHelper.parseText(text); + } + + public static Document getFormatDocument(String text) throws DocumentException, UnsupportedEncodingException { + return getFormatDocument(text, Dom4JConstant.ENCODING_UTF_8); + } + + public static Document getFormatDocument(String text, String charset) throws DocumentException, UnsupportedEncodingException { + String formatText = new String(text.getBytes(Dom4JConstant.ENCODING_ISO_8859_1), Dom4JConstant.ENCODING_UTF_8); + + return getDocument(formatText); + } + + public static Document getDocument(File file) throws DocumentException, IOException { + InputStream inputStream = new FileInputStream(file); + + return getDocument(inputStream); + } + + public static Document getFormatDocument(File file) throws DocumentException, IOException, UnsupportedEncodingException { + return getFormatDocument(file, Dom4JConstant.ENCODING_UTF_8); + } + + public static Document getFormatDocument(File file, String charset) throws DocumentException, IOException, UnsupportedEncodingException { + InputStream inputStream = new FileInputStream(file); + + return getFormatDocument(inputStream, charset); + } + + public static Document getDocument(InputSource inputSource) throws DocumentException { + SAXReader saxReader = new SAXReader(); + + return saxReader.read(inputSource); + } + + public static Document getFormatDocument(InputSource inputSource) throws DocumentException { + return getFormatDocument(inputSource, Dom4JConstant.ENCODING_UTF_8); + } + + public static Document getFormatDocument(InputSource inputSource, String charset) throws DocumentException { + inputSource.setEncoding(charset); + + return getDocument(inputSource); + } + + public static Document getDocument(InputStream inputStream) throws DocumentException, IOException { + SAXReader saxReader = new SAXReader(); + + Document document = null; + try { + document = saxReader.read(inputStream); + } catch (DocumentException e) { + throw e; + } finally { + if (inputStream != null) { + inputStream.close(); + } + } + + return document; + } + + public static Document getFormatDocument(InputStream inputStream) throws DocumentException, IOException, UnsupportedEncodingException { + return getFormatDocument(inputStream, Dom4JConstant.ENCODING_UTF_8); + } + + public static Document getFormatDocument(InputStream inputStream, String charset) throws DocumentException, IOException, UnsupportedEncodingException { + Reader inputStreamReader = new InputStreamReader(inputStream, charset); + + return getDocument(inputStreamReader); + } + + public static Document getDocument(Reader reader) throws DocumentException, IOException { + SAXReader saxReader = new SAXReader(); + + Document document = null; + try { + document = saxReader.read(reader); + } catch (DocumentException e) { + throw e; + } finally { + if (reader != null) { + reader.close(); + } + } + + return document; + } + + public static Document getDocument(URL url) throws DocumentException { + SAXReader saxReader = new SAXReader(); + + return saxReader.read(url); + } +} \ No newline at end of file diff --git a/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JWriter.java b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JWriter.java new file mode 100644 index 0000000000..c200271441 --- /dev/null +++ b/discovery-plugin/src/main/java/com/nepxion/discovery/plugin/xml/Dom4JWriter.java @@ -0,0 +1,51 @@ +package com.nepxion.discovery.plugin.xml; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.io.UnsupportedEncodingException; + +import org.dom4j.Document; +import org.dom4j.DocumentHelper; +import org.dom4j.io.OutputFormat; +import org.dom4j.io.XMLWriter; + +public class Dom4JWriter { + public static Document createDocument() { + return DocumentHelper.createDocument(); + } + + public static String getText(Document document) throws IOException, UnsupportedEncodingException { + return getText(document, Dom4JConstant.ENCODING_UTF_8); + } + + public static String getText(Document document, String charset) throws IOException, UnsupportedEncodingException { + ByteArrayOutputStream baos = new ByteArrayOutputStream(); + OutputFormat outputFormat = new OutputFormat(" ", true, charset); + + try { + XMLWriter writer = new XMLWriter(baos, outputFormat); + writer.write(document); + + baos.flush(); + } catch (UnsupportedEncodingException e) { + throw e; + } catch (IOException e) { + throw e; + } finally { + if (baos != null) { + baos.close(); + } + } + + return baos.toString(charset); + } +} \ No newline at end of file diff --git a/discovery-springcloud-example-a/pom.xml b/discovery-springcloud-example-a/pom.xml new file mode 100644 index 0000000000..bce62b1e16 --- /dev/null +++ b/discovery-springcloud-example-a/pom.xml @@ -0,0 +1,68 @@ + +Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import org.springframework.boot.autoconfigure.SpringBootApplication; +import org.springframework.boot.builder.SpringApplicationBuilder; +import org.springframework.cloud.client.discovery.EnableDiscoveryClient; + +@SpringBootApplication +@EnableDiscoveryClient +public class DiscoveryPluginApplication { + public static void main(String[] args) { + new SpringApplicationBuilder(DiscoveryPluginApplication.class).web(true).run(args); + } +} \ No newline at end of file diff --git a/discovery-springcloud-example-a/src/main/java/com/nepxion/discovery/plugin/example/DiscoveryPluginController.java b/discovery-springcloud-example-a/src/main/java/com/nepxion/discovery/plugin/example/DiscoveryPluginController.java new file mode 100644 index 0000000000..4af96a2844 --- /dev/null +++ b/discovery-springcloud-example-a/src/main/java/com/nepxion/discovery/plugin/example/DiscoveryPluginController.java @@ -0,0 +1,30 @@ +package com.nepxion.discovery.plugin.example; + +/** + *Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.util.List; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.cloud.client.ServiceInstance; +import org.springframework.cloud.client.discovery.DiscoveryClient; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestMethod; +import org.springframework.web.bind.annotation.RestController; + +@RestController +public class DiscoveryPluginController { + @Autowired + private DiscoveryClient discoveryClient; + + @RequestMapping(value = "/instances", method = RequestMethod.GET) + public ListTitle: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import java.io.File; +import java.io.FileInputStream; +import java.io.FileNotFoundException; +import java.io.IOException; +import java.io.InputStream; +import java.util.Timer; +import java.util.TimerTask; +import java.util.concurrent.ThreadLocalRandom; + +import javax.annotation.PostConstruct; + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Bean; +import org.springframework.stereotype.Component; + +import com.nepxion.discovery.plugin.loader.AbstractFileLoader; +import com.nepxion.discovery.plugin.loader.FileLoader; +import com.nepxion.eventbus.core.Event; +import com.nepxion.eventbus.core.EventControllerFactory; + +@Component +public class DiscoveryPluginSimulator { + // 模拟从本地配置和远程配置获取discovery.xml + @Bean + public FileLoader fileLoader() { + return new AbstractFileLoader() { + @Override + public InputStream getRemoteInputStream() throws IOException { + // 本地文件模拟代替远程文件 + return createInputStream("src/main/resources/discovery1.xml"); + } + + @Override + protected String getLocalContextPath() { + // 配置文件放在resources目录下 + return "classpath:discovery1.xml"; + + // 配置文件放在工程根目录下 + // return "file:discovery.xml"; + } + }; + } + + // 模拟每隔15秒通过事件总线接收远程配置中心推送过来的配置更新 + @Autowired + private EventControllerFactory eventControllerFactory; + + @PostConstruct + public void initialize() { + ThreadLocalRandom threadLocalRandom = ThreadLocalRandom.current(); + + Timer timer = new Timer(); + timer.scheduleAtFixedRate(new TimerTask() { + public void run() { + // 本地文件模拟代替远程文件，随机读取 + int index = threadLocalRandom.nextInt(3) + 1; + InputStream inputStream = createInputStream("src/main/resources/discovery" + index + ".xml"); + eventControllerFactory.getAsyncController().post(new Event(inputStream)); + } + }, 0L, 15000L); + } + + private InputStream createInputStream(String fileName) { + File file = new File(fileName); + InputStream inputStream = null; + try { + inputStream = new FileInputStream(file); + } catch (FileNotFoundException e) { + e.printStackTrace(); + } + + return inputStream; + } +} \ No newline at end of file diff --git a/discovery-springcloud-example-a/src/main/resources/application.properties b/discovery-springcloud-example-a/src/main/resources/application.properties new file mode 100644 index 0000000000..b008f0a5c0 --- /dev/null +++ b/discovery-springcloud-example-a/src/main/resources/application.properties @@ -0,0 +1,12 @@ +# Spring cloud config +spring.application.name=discovery-springcloud-example-a +server.port=4321 +eureka.client.serviceUrl.defaultZone=http://10.0.75.1:9528/eureka/ +eureka.instance.metadataMap.version=1.0 + +# Blacklist or Whitelist filter +spring.application.discovery.filter.enabled=false +# Multi-versions control +spring.application.discovery.version.enabled=true +# Get remote config +spring.application.discovery.remote.config.enabled=true \ No newline at end of file diff --git a/discovery-springcloud-example-a/src/main/resources/discovery1.xml b/discovery-springcloud-example-a/src/main/resources/discovery1.xml new file mode 100644 index 0000000000..8e32b3db85 --- /dev/null +++ b/discovery-springcloud-example-a/src/main/resources/discovery1.xml @@ -0,0 +1,24 @@ + +Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import org.springframework.boot.autoconfigure.SpringBootApplication; +import org.springframework.boot.builder.SpringApplicationBuilder; +import org.springframework.cloud.client.discovery.EnableDiscoveryClient; + +@SpringBootApplication +@EnableDiscoveryClient +public class DiscoveryPluginApplicationB1 { + public static void main(String[] args) { + new SpringApplicationBuilder(DiscoveryPluginApplicationB1.class).web(true).run(args); + } +} \ No newline at end of file diff --git a/discovery-springcloud-example-b1/src/main/resources/application.properties b/discovery-springcloud-example-b1/src/main/resources/application.properties new file mode 100644 index 0000000000..9701d39176 --- /dev/null +++ b/discovery-springcloud-example-b1/src/main/resources/application.properties @@ -0,0 +1,5 @@ +# Spring cloud config +spring.application.name=discovery-springcloud-example-b +server.port=4322 +eureka.client.serviceUrl.defaultZone=http://10.0.75.1:9528/eureka/ +eureka.instance.metadataMap.version=1.0 \ No newline at end of file diff --git a/discovery-springcloud-example-b1/src/main/resources/logback.xml b/discovery-springcloud-example-b1/src/main/resources/logback.xml new file mode 100644 index 0000000000..6ce0c9c182 --- /dev/null +++ b/discovery-springcloud-example-b1/src/main/resources/logback.xml @@ -0,0 +1,49 @@ + +Title: Nepxion Discovery
+ *Description: Nepxion Discovery
+ *Copyright: Copyright (c) 2017-2050
+ *Company: Nepxion
+ * @author Haojun Ren + * @version 1.0 + */ + +import org.springframework.boot.autoconfigure.SpringBootApplication; +import org.springframework.boot.builder.SpringApplicationBuilder; +import org.springframework.cloud.client.discovery.EnableDiscoveryClient; + +@SpringBootApplication +@EnableDiscoveryClient +public class DiscoveryPluginApplicationB2 { + public static void main(String[] args) { + new SpringApplicationBuilder(DiscoveryPluginApplicationB2.class).web(true).run(args); + } +} \ No newline at end of file diff --git a/discovery-springcloud-example-b2/src/main/resources/application.properties b/discovery-springcloud-example-b2/src/main/resources/application.properties new file mode 100644 index 0000000000..6387632e11 --- /dev/null +++ b/discovery-springcloud-example-b2/src/main/resources/application.properties @@ -0,0 +1,5 @@ +# Spring cloud config +spring.application.name=discovery-springcloud-example-b +server.port=4323 +eureka.client.serviceUrl.defaultZone=http://10.0.75.1:9528/eureka/ +eureka.instance.metadataMap.version=1.1 \ No newline at end of file diff --git a/discovery-springcloud-example-b2/src/main/resources/logback.xml b/discovery-springcloud-example-b2/src/main/resources/logback.xml new file mode 100644 index 0000000000..6ce0c9c182 --- /dev/null +++ b/discovery-springcloud-example-b2/src/main/resources/logback.xml @@ -0,0 +1,49 @@ + +