Commit 2234bba7 by Nepxion

重构类结构

parent 15323efd
...@@ -32,9 +32,9 @@ import com.nepxion.discovery.plugin.framework.constant.PluginConstant; ...@@ -32,9 +32,9 @@ import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
import com.nepxion.discovery.plugin.framework.entity.CountFilterEntity; import com.nepxion.discovery.plugin.framework.entity.CountFilterEntity;
import com.nepxion.discovery.plugin.framework.entity.DiscoveryEntity; import com.nepxion.discovery.plugin.framework.entity.DiscoveryEntity;
import com.nepxion.discovery.plugin.framework.entity.DiscoveryServiceEntity; import com.nepxion.discovery.plugin.framework.entity.DiscoveryServiceEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterHolderEntity; import com.nepxion.discovery.plugin.framework.entity.FilterHolderEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterType; import com.nepxion.discovery.plugin.framework.entity.FilterType;
import com.nepxion.discovery.plugin.framework.entity.IpAddressFilterEntity;
import com.nepxion.discovery.plugin.framework.entity.RegisterEntity; import com.nepxion.discovery.plugin.framework.entity.RegisterEntity;
import com.nepxion.discovery.plugin.framework.entity.RuleEntity; import com.nepxion.discovery.plugin.framework.entity.RuleEntity;
import com.nepxion.discovery.plugin.framework.entity.VersionEntity; import com.nepxion.discovery.plugin.framework.entity.VersionEntity;
...@@ -115,9 +115,9 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser { ...@@ -115,9 +115,9 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser {
Element childElement = (Element) childElementObject; Element childElement = (Element) childElementObject;
if (StringUtils.equals(childElement.getName(), ConfigConstant.BLACKLIST_ELEMENT_NAME)) { if (StringUtils.equals(childElement.getName(), ConfigConstant.BLACKLIST_ELEMENT_NAME)) {
parseFilter(childElement, ConfigConstant.BLACKLIST_ELEMENT_NAME, registerEntity); parseIpAddressFilter(childElement, ConfigConstant.BLACKLIST_ELEMENT_NAME, registerEntity);
} else if (StringUtils.equals(childElement.getName(), ConfigConstant.WHITELIST_ELEMENT_NAME)) { } else if (StringUtils.equals(childElement.getName(), ConfigConstant.WHITELIST_ELEMENT_NAME)) {
parseFilter(childElement, ConfigConstant.WHITELIST_ELEMENT_NAME, registerEntity); parseIpAddressFilter(childElement, ConfigConstant.WHITELIST_ELEMENT_NAME, registerEntity);
} else if (StringUtils.equals(childElement.getName(), ConfigConstant.COUNT_ELEMENT_NAME)) { } else if (StringUtils.equals(childElement.getName(), ConfigConstant.COUNT_ELEMENT_NAME)) {
parseCountFilter(childElement, registerEntity); parseCountFilter(childElement, registerEntity);
} }
...@@ -133,9 +133,9 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser { ...@@ -133,9 +133,9 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser {
Element childElement = (Element) childElementObject; Element childElement = (Element) childElementObject;
if (StringUtils.equals(childElement.getName(), ConfigConstant.BLACKLIST_ELEMENT_NAME)) { if (StringUtils.equals(childElement.getName(), ConfigConstant.BLACKLIST_ELEMENT_NAME)) {
parseFilter(childElement, ConfigConstant.BLACKLIST_ELEMENT_NAME, discoveryEntity); parseIpAddressFilter(childElement, ConfigConstant.BLACKLIST_ELEMENT_NAME, discoveryEntity);
} else if (StringUtils.equals(childElement.getName(), ConfigConstant.WHITELIST_ELEMENT_NAME)) { } else if (StringUtils.equals(childElement.getName(), ConfigConstant.WHITELIST_ELEMENT_NAME)) {
parseFilter(childElement, ConfigConstant.WHITELIST_ELEMENT_NAME, discoveryEntity); parseIpAddressFilter(childElement, ConfigConstant.WHITELIST_ELEMENT_NAME, discoveryEntity);
} else if (StringUtils.equals(childElement.getName(), ConfigConstant.VERSION_ELEMENT_NAME)) { } else if (StringUtils.equals(childElement.getName(), ConfigConstant.VERSION_ELEMENT_NAME)) {
parseVersion(childElement, discoveryEntity); parseVersion(childElement, discoveryEntity);
} }
...@@ -144,23 +144,23 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser { ...@@ -144,23 +144,23 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser {
} }
@SuppressWarnings("rawtypes") @SuppressWarnings("rawtypes")
private void parseFilter(Element element, String filterTypeValue, FilterHolderEntity filterHolderEntity) { private void parseIpAddressFilter(Element element, String filterTypeValue, FilterHolderEntity filterHolderEntity) {
FilterEntity filterEntity = filterHolderEntity.getFilterEntity(); IpAddressFilterEntity ipAddressFilterEntity = filterHolderEntity.getIpAddressFilterEntity();
if (filterEntity != null) { if (ipAddressFilterEntity != null) {
throw new PluginException("Allow only one filter element to be configed, [" + ConfigConstant.BLACKLIST_ELEMENT_NAME + "] or [" + ConfigConstant.WHITELIST_ELEMENT_NAME + "]"); throw new PluginException("Allow only one filter element to be configed, [" + ConfigConstant.BLACKLIST_ELEMENT_NAME + "] or [" + ConfigConstant.WHITELIST_ELEMENT_NAME + "]");
} }
filterEntity = new FilterEntity(); ipAddressFilterEntity = new IpAddressFilterEntity();
filterEntity.setFilterType(FilterType.fromString(filterTypeValue)); ipAddressFilterEntity.setFilterType(FilterType.fromString(filterTypeValue));
Attribute globalFilterAttribute = element.attribute(ConfigConstant.FILTER_VALUE_ATTRIBUTE_NAME); Attribute globalFilterAttribute = element.attribute(ConfigConstant.FILTER_VALUE_ATTRIBUTE_NAME);
if (globalFilterAttribute != null) { if (globalFilterAttribute != null) {
String globalFilterValue = globalFilterAttribute.getData().toString().trim(); String globalFilterValue = globalFilterAttribute.getData().toString().trim();
List<String> globalFilterValueList = parseList(globalFilterValue); List<String> globalFilterValueList = parseList(globalFilterValue);
filterEntity.setFilterValueList(globalFilterValueList); ipAddressFilterEntity.setFilterValueList(globalFilterValueList);
} }
Map<String, List<String>> filterMap = filterEntity.getFilterMap(); Map<String, List<String>> filterMap = ipAddressFilterEntity.getFilterMap();
for (Iterator elementIterator = element.elementIterator(); elementIterator.hasNext();) { for (Iterator elementIterator = element.elementIterator(); elementIterator.hasNext();) {
Object childElementObject = elementIterator.next(); Object childElementObject = elementIterator.next();
if (childElementObject instanceof Element) { if (childElementObject instanceof Element) {
...@@ -184,7 +184,7 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser { ...@@ -184,7 +184,7 @@ public class ConfigParser extends Dom4JParser implements PluginConfigParser {
} }
} }
filterHolderEntity.setFilterEntity(filterEntity); filterHolderEntity.setIpAddressFilterEntity(ipAddressFilterEntity);
} }
@SuppressWarnings("rawtypes") @SuppressWarnings("rawtypes")
......
...@@ -19,18 +19,18 @@ import org.apache.commons.lang3.builder.ToStringStyle; ...@@ -19,18 +19,18 @@ import org.apache.commons.lang3.builder.ToStringStyle;
public class FilterHolderEntity implements Serializable { public class FilterHolderEntity implements Serializable {
private static final long serialVersionUID = 8767022123685151416L; private static final long serialVersionUID = 8767022123685151416L;
private FilterEntity filterEntity; private IpAddressFilterEntity ipAddressFilterEntity;
public FilterHolderEntity() { public FilterHolderEntity() {
} }
public FilterEntity getFilterEntity() { public IpAddressFilterEntity getIpAddressFilterEntity() {
return filterEntity; return ipAddressFilterEntity;
} }
public void setFilterEntity(FilterEntity filterEntity) { public void setIpAddressFilterEntity(IpAddressFilterEntity ipAddressFilterEntity) {
this.filterEntity = filterEntity; this.ipAddressFilterEntity = ipAddressFilterEntity;
} }
@Override @Override
......
...@@ -19,14 +19,14 @@ import org.apache.commons.lang3.builder.HashCodeBuilder; ...@@ -19,14 +19,14 @@ import org.apache.commons.lang3.builder.HashCodeBuilder;
import org.apache.commons.lang3.builder.ToStringBuilder; import org.apache.commons.lang3.builder.ToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle; import org.apache.commons.lang3.builder.ToStringStyle;
public class FilterEntity implements Serializable { public class IpAddressFilterEntity implements Serializable {
private static final long serialVersionUID = 3830016495318834467L; private static final long serialVersionUID = 3830016495318834467L;
private FilterType filterType; private FilterType filterType;
private List<String> filterValueList; private List<String> filterValueList;
private Map<String, List<String>> filterMap = new LinkedHashMap<String, List<String>>(); private Map<String, List<String>> filterMap = new LinkedHashMap<String, List<String>>();
public FilterEntity() { public IpAddressFilterEntity() {
} }
......
...@@ -19,8 +19,8 @@ import org.springframework.cloud.client.ServiceInstance; ...@@ -19,8 +19,8 @@ import org.springframework.cloud.client.ServiceInstance;
import com.nepxion.discovery.plugin.framework.constant.PluginConstant; import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
import com.nepxion.discovery.plugin.framework.entity.DiscoveryEntity; import com.nepxion.discovery.plugin.framework.entity.DiscoveryEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterType; import com.nepxion.discovery.plugin.framework.entity.FilterType;
import com.nepxion.discovery.plugin.framework.entity.IpAddressFilterEntity;
import com.nepxion.discovery.plugin.framework.entity.RuleEntity; import com.nepxion.discovery.plugin.framework.entity.RuleEntity;
public class IpAddressFilterDiscoveryListener extends AbstractDiscoveryListener { public class IpAddressFilterDiscoveryListener extends AbstractDiscoveryListener {
...@@ -40,15 +40,15 @@ public class IpAddressFilterDiscoveryListener extends AbstractDiscoveryListener ...@@ -40,15 +40,15 @@ public class IpAddressFilterDiscoveryListener extends AbstractDiscoveryListener
return; return;
} }
FilterEntity filterEntity = discoveryEntity.getFilterEntity(); IpAddressFilterEntity ipAddressFilterEntity = discoveryEntity.getIpAddressFilterEntity();
if (filterEntity == null) { if (ipAddressFilterEntity == null) {
return; return;
} }
FilterType filterType = filterEntity.getFilterType(); FilterType filterType = ipAddressFilterEntity.getFilterType();
List<String> globalFilterValueList = filterEntity.getFilterValueList(); List<String> globalFilterValueList = ipAddressFilterEntity.getFilterValueList();
Map<String, List<String>> filterMap = filterEntity.getFilterMap(); Map<String, List<String>> filterMap = ipAddressFilterEntity.getFilterMap();
List<String> filterValueList = filterMap.get(providerServiceId); List<String> filterValueList = filterMap.get(providerServiceId);
List<String> allFilterValueList = new ArrayList<String>(); List<String> allFilterValueList = new ArrayList<String>();
......
...@@ -18,8 +18,8 @@ import org.apache.commons.collections4.CollectionUtils; ...@@ -18,8 +18,8 @@ import org.apache.commons.collections4.CollectionUtils;
import com.nepxion.discovery.plugin.framework.constant.PluginConstant; import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
import com.nepxion.discovery.plugin.framework.entity.DiscoveryEntity; import com.nepxion.discovery.plugin.framework.entity.DiscoveryEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterType; import com.nepxion.discovery.plugin.framework.entity.FilterType;
import com.nepxion.discovery.plugin.framework.entity.IpAddressFilterEntity;
import com.nepxion.discovery.plugin.framework.entity.RuleEntity; import com.nepxion.discovery.plugin.framework.entity.RuleEntity;
import com.netflix.loadbalancer.Server; import com.netflix.loadbalancer.Server;
...@@ -40,15 +40,15 @@ public class IpAddressFilterLoadBalanceListener extends AbstractLoadBalanceListe ...@@ -40,15 +40,15 @@ public class IpAddressFilterLoadBalanceListener extends AbstractLoadBalanceListe
return; return;
} }
FilterEntity filterEntity = discoveryEntity.getFilterEntity(); IpAddressFilterEntity ipAddressFilterEntity = discoveryEntity.getIpAddressFilterEntity();
if (filterEntity == null) { if (ipAddressFilterEntity == null) {
return; return;
} }
FilterType filterType = filterEntity.getFilterType(); FilterType filterType = ipAddressFilterEntity.getFilterType();
List<String> globalFilterValueList = filterEntity.getFilterValueList(); List<String> globalFilterValueList = ipAddressFilterEntity.getFilterValueList();
Map<String, List<String>> filterMap = filterEntity.getFilterMap(); Map<String, List<String>> filterMap = ipAddressFilterEntity.getFilterMap();
List<String> filterValueList = filterMap.get(providerServiceId); List<String> filterValueList = filterMap.get(providerServiceId);
List<String> allFilterValueList = new ArrayList<String>(); List<String> allFilterValueList = new ArrayList<String>();
......
...@@ -17,8 +17,8 @@ import org.apache.commons.collections4.CollectionUtils; ...@@ -17,8 +17,8 @@ import org.apache.commons.collections4.CollectionUtils;
import org.springframework.cloud.client.serviceregistry.Registration; import org.springframework.cloud.client.serviceregistry.Registration;
import com.nepxion.discovery.plugin.framework.constant.PluginConstant; import com.nepxion.discovery.plugin.framework.constant.PluginConstant;
import com.nepxion.discovery.plugin.framework.entity.FilterEntity;
import com.nepxion.discovery.plugin.framework.entity.FilterType; import com.nepxion.discovery.plugin.framework.entity.FilterType;
import com.nepxion.discovery.plugin.framework.entity.IpAddressFilterEntity;
import com.nepxion.discovery.plugin.framework.entity.RegisterEntity; import com.nepxion.discovery.plugin.framework.entity.RegisterEntity;
import com.nepxion.discovery.plugin.framework.entity.RuleEntity; import com.nepxion.discovery.plugin.framework.entity.RuleEntity;
import com.nepxion.discovery.plugin.framework.event.RegisterFailureEvent; import com.nepxion.discovery.plugin.framework.event.RegisterFailureEvent;
...@@ -45,15 +45,15 @@ public class IpAddressFilterRegisterListener extends AbstractRegisterListener { ...@@ -45,15 +45,15 @@ public class IpAddressFilterRegisterListener extends AbstractRegisterListener {
return; return;
} }
FilterEntity filterEntity = registerEntity.getFilterEntity(); IpAddressFilterEntity ipAddressFilterEntity = registerEntity.getIpAddressFilterEntity();
if (filterEntity == null) { if (ipAddressFilterEntity == null) {
return; return;
} }
FilterType filterType = filterEntity.getFilterType(); FilterType filterType = ipAddressFilterEntity.getFilterType();
List<String> globalFilterValueList = filterEntity.getFilterValueList(); List<String> globalFilterValueList = ipAddressFilterEntity.getFilterValueList();
Map<String, List<String>> filterMap = filterEntity.getFilterMap(); Map<String, List<String>> filterMap = ipAddressFilterEntity.getFilterMap();
List<String> filterValueList = filterMap.get(serviceId); List<String> filterValueList = filterMap.get(serviceId);
List<String> allFilterValueList = new ArrayList<String>(); List<String> allFilterValueList = new ArrayList<String>();
......
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