com.nepxion.discovery.plugin.strategy.wrapper.StrategyWrapper Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of discovery-plugin-strategy Show documentation
Show all versions of discovery-plugin-strategy Show documentation
Nepxion Discovery is an enhancement for Spring Cloud Discovery
package com.nepxion.discovery.plugin.strategy.wrapper;
/**
* Title: Nepxion Discovery
* Description: Nepxion Discovery
* Copyright: Copyright (c) 2017-2050
* Company: Nepxion
* @author Haojun Ren
* @version 1.0
*/
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.MapUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.apache.commons.lang3.tuple.Pair;
import org.springframework.beans.factory.annotation.Autowired;
import com.nepxion.discovery.common.entity.AddressWeightEntity;
import com.nepxion.discovery.common.entity.MapWeightEntity;
import com.nepxion.discovery.common.entity.RegionWeightEntity;
import com.nepxion.discovery.common.entity.RuleEntity;
import com.nepxion.discovery.common.entity.StrategyConditionBlueGreenEntity;
import com.nepxion.discovery.common.entity.StrategyConditionGrayEntity;
import com.nepxion.discovery.common.entity.StrategyCustomizationEntity;
import com.nepxion.discovery.common.entity.StrategyEntity;
import com.nepxion.discovery.common.entity.StrategyRouteEntity;
import com.nepxion.discovery.common.entity.StrategyType;
import com.nepxion.discovery.common.entity.VersionWeightEntity;
import com.nepxion.discovery.plugin.framework.adapter.PluginAdapter;
import com.nepxion.discovery.plugin.framework.loadbalance.weight.MapWeightRandom;
import com.nepxion.discovery.plugin.strategy.condition.StrategyCondition;
public class StrategyWrapper {
@Autowired
protected PluginAdapter pluginAdapter;
@Autowired
protected StrategyCondition strategyCondition;
// 从远程配置中心或者本地配置文件获取版本路由配置。如果是远程配置中心，则值会动态改变
public String getRouteVersion() {
String routeVersion = getConditionBlueGreenRouteVersion(null);
if (StringUtils.isEmpty(routeVersion)) {
routeVersion = getConditionGrayRouteVersion(null);
if (StringUtils.isEmpty(routeVersion)) {
routeVersion = getGlobalRouteVersion();
}
}
return routeVersion;
}
// 从远程配置中心或者本地配置文件获取区域路由配置。如果是远程配置中心，则值会动态改变
public String getRouteRegion() {
String routeRegion = getConditionBlueGreenRouteRegion(null);
if (StringUtils.isEmpty(routeRegion)) {
routeRegion = getConditionGrayRouteRegion(null);
if (StringUtils.isEmpty(routeRegion)) {
routeRegion = getGlobalRouteRegion();
}
}
return routeRegion;
}
// 从远程配置中心或者本地配置文件获取IP地址和端口路由配置。如果是远程配置中心，则值会动态改变
public String getRouteAddress() {
String routeAddress = getConditionBlueGreenRouteAddress(null);
if (StringUtils.isEmpty(routeAddress)) {
routeAddress = getConditionGrayRouteAddress(null);
if (StringUtils.isEmpty(routeAddress)) {
routeAddress = getGlobalRouteAddress();
}
}
return routeAddress;
}
// 从远程配置中心或者本地配置文件获取版本权重配置。如果是远程配置中心，则值会动态改变
public String getRouteVersionWeight() {
String routeVersionWeight = getConditionBlueGreenRouteVersionWeight(null);
if (StringUtils.isEmpty(routeVersionWeight)) {
routeVersionWeight = getGlobalRouteVersionWeight();
}
return routeVersionWeight;
}
// 从远程配置中心或者本地配置文件获取区域权重配置。如果是远程配置中心，则值会动态改变
public String getRouteRegionWeight() {
String routeRegionWeight = getConditionBlueGreenRouteRegionWeight(null);
if (StringUtils.isEmpty(routeRegionWeight)) {
routeRegionWeight = getGlobalRouteRegionWeight();
}
return routeRegionWeight;
}
// 从远程配置中心或者本地配置文件获取版本路由配置。如果是远程配置中心，则值会动态改变
public String getRouteVersion(Map headerMap) {
String routeVersion = getConditionBlueGreenRouteVersion(headerMap);
if (StringUtils.isEmpty(routeVersion)) {
routeVersion = getConditionGrayRouteVersion(headerMap);
if (StringUtils.isEmpty(routeVersion)) {
routeVersion = getGlobalRouteVersion();
}
}
return routeVersion;
}
// 从远程配置中心或者本地配置文件获取区域路由配置。如果是远程配置中心，则值会动态改变
public String getRouteRegion(Map headerMap) {
String routeRegion = getConditionBlueGreenRouteRegion(headerMap);
if (StringUtils.isEmpty(routeRegion)) {
routeRegion = getConditionGrayRouteRegion(headerMap);
if (StringUtils.isEmpty(routeRegion)) {
routeRegion = getGlobalRouteRegion();
}
}
return routeRegion;
}
// 从远程配置中心或者本地配置文件获取IP地址和端口路由配置。如果是远程配置中心，则值会动态改变
public String getRouteAddress(Map headerMap) {
String routeAddress = getConditionBlueGreenRouteAddress(headerMap);
if (StringUtils.isEmpty(routeAddress)) {
routeAddress = getConditionGrayRouteAddress(headerMap);
if (StringUtils.isEmpty(routeAddress)) {
routeAddress = getGlobalRouteAddress();
}
}
return routeAddress;
}
// 从远程配置中心或者本地配置文件获取版本权重配置。如果是远程配置中心，则值会动态改变
public String getRouteVersionWeight(Map headerMap) {
String routeVersionWeight = getConditionBlueGreenRouteVersionWeight(headerMap);
if (StringUtils.isEmpty(routeVersionWeight)) {
routeVersionWeight = getGlobalRouteVersionWeight();
}
return routeVersionWeight;
}
// 从远程配置中心或者本地配置文件获取区域权重配置。如果是远程配置中心，则值会动态改变
public String getRouteRegionWeight(Map headerMap) {
String routeRegionWeight = getConditionBlueGreenRouteRegionWeight(headerMap);
if (StringUtils.isEmpty(routeRegionWeight)) {
routeRegionWeight = getGlobalRouteRegionWeight();
}
return routeRegionWeight;
}
public String getGlobalRouteVersion() {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyEntity strategyEntity = ruleEntity.getStrategyEntity();
if (strategyEntity != null) {
return strategyEntity.getVersionValue();
}
}
return null;
}
public String getGlobalRouteRegion() {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyEntity strategyEntity = ruleEntity.getStrategyEntity();
if (strategyEntity != null) {
return strategyEntity.getRegionValue();
}
}
return null;
}
public String getGlobalRouteAddress() {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyEntity strategyEntity = ruleEntity.getStrategyEntity();
if (strategyEntity != null) {
return strategyEntity.getAddressValue();
}
}
return null;
}
public String getGlobalRouteVersionWeight() {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyEntity strategyEntity = ruleEntity.getStrategyEntity();
if (strategyEntity != null) {
return strategyEntity.getVersionWeightValue();
}
}
return null;
}
public String getGlobalRouteRegionWeight() {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyEntity strategyEntity = ruleEntity.getStrategyEntity();
if (strategyEntity != null) {
return strategyEntity.getRegionWeightValue();
}
}
return null;
}
public String getConditionBlueGreenRouteVersion(Map headerMap) {
StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity = getTriggeredStrategyConditionBlueGreenEntity(StrategyType.VERSION, headerMap);
if (strategyConditionBlueGreenEntity != null) {
String versionId = strategyConditionBlueGreenEntity.getVersionId();
StrategyRouteEntity strategyRouteEntity = getTriggeredStrategyRouteEntity(versionId, StrategyType.VERSION);
if (strategyRouteEntity != null) {
return strategyRouteEntity.getValue();
}
}
return null;
}
public String getConditionBlueGreenRouteRegion(Map headerMap) {
StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity = getTriggeredStrategyConditionBlueGreenEntity(StrategyType.REGION, headerMap);
if (strategyConditionBlueGreenEntity != null) {
String regionId = strategyConditionBlueGreenEntity.getRegionId();
StrategyRouteEntity strategyRouteEntity = getTriggeredStrategyRouteEntity(regionId, StrategyType.REGION);
if (strategyRouteEntity != null) {
return strategyRouteEntity.getValue();
}
}
return null;
}
public String getConditionBlueGreenRouteAddress(Map headerMap) {
StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity = getTriggeredStrategyConditionBlueGreenEntity(StrategyType.ADDRESS, headerMap);
if (strategyConditionBlueGreenEntity != null) {
String addressId = strategyConditionBlueGreenEntity.getAddressId();
StrategyRouteEntity strategyRouteEntity = getTriggeredStrategyRouteEntity(addressId, StrategyType.ADDRESS);
if (strategyRouteEntity != null) {
return strategyRouteEntity.getValue();
}
}
return null;
}
public String getConditionBlueGreenRouteVersionWeight(Map headerMap) {
StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity = getTriggeredStrategyConditionBlueGreenEntity(StrategyType.VERSION_WEIGHT, headerMap);
if (strategyConditionBlueGreenEntity != null) {
String versionWeightId = strategyConditionBlueGreenEntity.getVersionWeightId();
StrategyRouteEntity strategyRouteEntity = getTriggeredStrategyRouteEntity(versionWeightId, StrategyType.VERSION_WEIGHT);
if (strategyRouteEntity != null) {
return strategyRouteEntity.getValue();
}
}
return null;
}
public String getConditionBlueGreenRouteRegionWeight(Map headerMap) {
StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity = getTriggeredStrategyConditionBlueGreenEntity(StrategyType.REGION_WEIGHT, headerMap);
if (strategyConditionBlueGreenEntity != null) {
String regionWeightId = strategyConditionBlueGreenEntity.getRegionWeightId();
StrategyRouteEntity strategyRouteEntity = getTriggeredStrategyRouteEntity(regionWeightId, StrategyType.REGION_WEIGHT);
if (strategyRouteEntity != null) {
return strategyRouteEntity.getValue();
}
}
return null;
}
private StrategyConditionBlueGreenEntity getTriggeredStrategyConditionBlueGreenEntity(StrategyType strategyType, Map headerMap) {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyCustomizationEntity strategyCustomizationEntity = ruleEntity.getStrategyCustomizationEntity();
if (strategyCustomizationEntity != null) {
List strategyConditionBlueGreenEntityList = strategyCustomizationEntity.getStrategyConditionBlueGreenEntityList();
if (CollectionUtils.isNotEmpty(strategyConditionBlueGreenEntityList)) {
StrategyConditionBlueGreenEntity expressionStrategyConditionBlueGreenEntity = getTriggeredExpressionStrategyConditionBlueGreenEntity(strategyConditionBlueGreenEntityList, strategyType, headerMap);
if (expressionStrategyConditionBlueGreenEntity != null) {
return expressionStrategyConditionBlueGreenEntity;
}
}
}
}
return null;
}
private StrategyConditionBlueGreenEntity getTriggeredExpressionStrategyConditionBlueGreenEntity(List strategyConditionBlueGreenEntityList, StrategyType strategyType, Map headerMap) {
for (StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity : strategyConditionBlueGreenEntityList) {
boolean isValidated = validateBlueGreenStrategyType(strategyConditionBlueGreenEntity, strategyType);
if (isValidated) {
boolean isTriggered = false;
if (headerMap == null) {
isTriggered = strategyCondition.isTriggered(strategyConditionBlueGreenEntity);
} else {
isTriggered = strategyCondition.isTriggered(strategyConditionBlueGreenEntity, headerMap);
}
if (isTriggered) {
return strategyConditionBlueGreenEntity;
}
}
}
return null;
}
private boolean validateBlueGreenStrategyType(StrategyConditionBlueGreenEntity strategyConditionBlueGreenEntity, StrategyType strategyType) {
switch (strategyType) {
case VERSION:
return StringUtils.isNotEmpty(strategyConditionBlueGreenEntity.getVersionId());
case REGION:
return StringUtils.isNotEmpty(strategyConditionBlueGreenEntity.getRegionId());
case ADDRESS:
return StringUtils.isNotEmpty(strategyConditionBlueGreenEntity.getAddressId());
case VERSION_WEIGHT:
return StringUtils.isNotEmpty(strategyConditionBlueGreenEntity.getVersionWeightId());
case REGION_WEIGHT:
return StringUtils.isNotEmpty(strategyConditionBlueGreenEntity.getRegionWeightId());
}
return false;
}
public String getConditionGrayRouteVersion(Map headerMap) {
StrategyConditionGrayEntity strategyConditionGrayEntity = getTriggeredStrategyConditionGrayEntity(headerMap);
if (strategyConditionGrayEntity != null) {
VersionWeightEntity versionWeightEntity = strategyConditionGrayEntity.getVersionWeightEntity();
if (versionWeightEntity != null) {
return getTriggeredStrategyGrayRoute(versionWeightEntity, StrategyType.VERSION);
}
}
return null;
}
public String getConditionGrayRouteRegion(Map headerMap) {
StrategyConditionGrayEntity strategyConditionGrayEntity = getTriggeredStrategyConditionGrayEntity(headerMap);
if (strategyConditionGrayEntity != null) {
RegionWeightEntity regionWeightEntity = strategyConditionGrayEntity.getRegionWeightEntity();
if (regionWeightEntity != null) {
return getTriggeredStrategyGrayRoute(regionWeightEntity, StrategyType.REGION);
}
}
return null;
}
public String getConditionGrayRouteAddress(Map headerMap) {
StrategyConditionGrayEntity strategyConditionGrayEntity = getTriggeredStrategyConditionGrayEntity(headerMap);
if (strategyConditionGrayEntity != null) {
AddressWeightEntity addressWeightEntity = strategyConditionGrayEntity.getAddressWeightEntity();
if (addressWeightEntity != null) {
return getTriggeredStrategyGrayRoute(addressWeightEntity, StrategyType.ADDRESS);
}
}
return null;
}
private StrategyConditionGrayEntity getTriggeredStrategyConditionGrayEntity(Map headerMap) {
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyCustomizationEntity strategyCustomizationEntity = ruleEntity.getStrategyCustomizationEntity();
if (strategyCustomizationEntity != null) {
List strategyConditionGrayEntityList = strategyCustomizationEntity.getStrategyConditionGrayEntityList();
if (CollectionUtils.isNotEmpty(strategyConditionGrayEntityList)) {
StrategyConditionGrayEntity globalStrategyConditionGrayEntity = getTriggeredGlobalStrategyConditionGrayEntity(strategyConditionGrayEntityList);
if (globalStrategyConditionGrayEntity != null) {
return globalStrategyConditionGrayEntity;
} else {
StrategyConditionGrayEntity expressionStrategyConditionGrayEntity = getTriggeredExpressionStrategyConditionGrayEntity(strategyConditionGrayEntityList, headerMap);
if (expressionStrategyConditionGrayEntity != null) {
return expressionStrategyConditionGrayEntity;
}
}
}
}
}
return null;
}
private StrategyConditionGrayEntity getTriggeredGlobalStrategyConditionGrayEntity(List strategyConditionGrayEntityList) {
for (StrategyConditionGrayEntity strategyConditionGrayEntity : strategyConditionGrayEntityList) {
String conditionHeader = strategyConditionGrayEntity.getConditionHeader();
if (StringUtils.isEmpty(conditionHeader)) {
return strategyConditionGrayEntity;
}
}
return null;
}
private StrategyConditionGrayEntity getTriggeredExpressionStrategyConditionGrayEntity(List strategyConditionGrayEntityList, Map headerMap) {
for (StrategyConditionGrayEntity strategyConditionGrayEntity : strategyConditionGrayEntityList) {
boolean isTriggered = false;
if (headerMap == null) {
isTriggered = strategyCondition.isTriggered(strategyConditionGrayEntity);
} else {
isTriggered = strategyCondition.isTriggered(strategyConditionGrayEntity, headerMap);
}
if (isTriggered) {
return strategyConditionGrayEntity;
}
}
return null;
}
private String getTriggeredStrategyGrayRoute(MapWeightEntity mapWeightEntity, StrategyType strategyType) {
Map weightMap = mapWeightEntity.getWeightMap();
if (MapUtils.isEmpty(weightMap)) {
return null;
}
List> weightList = new ArrayList>();
for (Map.Entry entry : weightMap.entrySet()) {
String id = entry.getKey();
StrategyRouteEntity strategyRouteEntity = getTriggeredStrategyRouteEntity(id, strategyType);
if (strategyRouteEntity != null) {
String strategyRoute = strategyRouteEntity.getValue();
Double weight = Double.valueOf(entry.getValue());
weightList.add(new ImmutablePair(strategyRoute, weight));
}
}
MapWeightRandom weightRandom = new MapWeightRandom(weightList);
return weightRandom.random();
}
private StrategyRouteEntity getTriggeredStrategyRouteEntity(String id, StrategyType strategyType) {
if (StringUtils.isEmpty(id)) {
return null;
}
RuleEntity ruleEntity = pluginAdapter.getRule();
if (ruleEntity != null) {
StrategyCustomizationEntity strategyCustomizationEntity = ruleEntity.getStrategyCustomizationEntity();
if (strategyCustomizationEntity != null) {
List strategyRouteEntityList = strategyCustomizationEntity.getStrategyRouteEntityList();
if (CollectionUtils.isNotEmpty(strategyRouteEntityList)) {
for (StrategyRouteEntity strategyRouteEntity : strategyRouteEntityList) {
if (StringUtils.equals(strategyRouteEntity.getId(), id) && strategyRouteEntity.getType() == strategyType) {
return strategyRouteEntity;
}
}
}
}
}
return null;
}
}