com.nepxion.discovery.common.entity.WeightEntityWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of discovery-common Show documentation
Nepxion Discovery is a solution for Spring Cloud with blue green, gray, weight, limitation, circuit breaker, degrade, isolation, monitor, tracing, dye, failover, async agent
There is a newer version: 6.22.0
Show newest version
package com.nepxion.discovery.common.entity;

/**
 * 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 com.nepxion.discovery.common.constant.DiscoveryConstant;
import com.nepxion.discovery.common.util.JsonUtil;
import com.nepxion.discovery.common.util.StringUtil;

public class WeightEntityWrapper {
    @SuppressWarnings("unchecked")
    public static List parseWeightEntityList(String weightValue) {
        List weightEntityList = new ArrayList();

        Map jsonWeightMap = JsonUtil.fromJson(weightValue, Map.class);
        for (Map.Entry entry : jsonWeightMap.entrySet()) {
            String providerServiceName = entry.getKey();
            String providerWeightValue = entry.getValue();

            WeightEntity weightEntity = new WeightEntity();
            weightEntity.setProviderServiceName(providerServiceName);

            parseWeightEntity(weightEntity, providerWeightValue);

            weightEntityList.add(weightEntity);
        }

        return weightEntityList;
    }

    public static void parseWeightEntity(MapWeightEntity weightEntity, String weightValue) {
        Map weightMap = weightEntity.getWeightMap();
        List providerWeightValueList = StringUtil.splitToList(weightValue, DiscoveryConstant.SEPARATE);
        for (String value : providerWeightValueList) {
            String[] valueArray = StringUtils.split(value, DiscoveryConstant.EQUALS);
            String key = valueArray[0].trim();
            int weight = 0;
            try {
                weight = Integer.valueOf(valueArray[1].trim());
                if (weight < 0) {
                    weight = 0;
                }
            } catch (NumberFormatException e) {

            }

            weightMap.put(key, weight);
        }
    }

    public static int getWeight(String consumerServiceId, String providerServiceId, String providerKey, Map> weightEntityMap) {
        if (MapUtils.isEmpty(weightEntityMap)) {
            return -1;
        }

        List weightEntityList = weightEntityMap.get(consumerServiceId);

        return getWeight(providerServiceId, providerKey, weightEntityList);
    }

    public static int getWeight(String providerServiceId, String providerKey, List weightEntityList) {
        if (CollectionUtils.isEmpty(weightEntityList)) {
            return -1;
        }

        for (WeightEntity weightEntity : weightEntityList) {
            String providerServiceName = weightEntity.getProviderServiceName();
            if (StringUtils.equalsIgnoreCase(providerServiceName, providerServiceId)) {
                return getWeight(providerKey, weightEntity);
            }
        }

        return -1;
    }

    public static int getWeight(String providerKey, MapWeightEntity weightEntity) {
        if (weightEntity == null) {
            return -1;
        }

        Map weightMap = weightEntity.getWeightMap();
        if (MapUtils.isEmpty(weightMap)) {
            return -1;
        }

        Integer weight = weightMap.get(providerKey);
        if (weight != null) {
            return weight;
        } else {
            return -1;
        }
    }

    public static int getWeight(WeightFilterEntity weightFilterEntity, String providerServiceId, String providerVersion, String providerRegion, String serviceId) {
        int weight = -1;
        if (StringUtils.isNotEmpty(serviceId) && weight < 0) {
            Map> versionWeightEntityMap = weightFilterEntity.getVersionWeightEntityMap();
            weight = WeightEntityWrapper.getWeight(serviceId, providerServiceId, providerVersion, versionWeightEntityMap);
        }
        if (weight < 0) {
            List versionWeightEntityList = weightFilterEntity.getVersionWeightEntityList();
            weight = WeightEntityWrapper.getWeight(providerServiceId, providerVersion, versionWeightEntityList);
        }
        if (weight < 0) {
            VersionWeightEntity versionWeightEntity = weightFilterEntity.getVersionWeightEntity();
            weight = WeightEntityWrapper.getWeight(providerVersion, versionWeightEntity);
        }

        if (StringUtils.isNotEmpty(serviceId) && weight < 0) {
            Map> regionWeightEntityMap = weightFilterEntity.getRegionWeightEntityMap();
            weight = WeightEntityWrapper.getWeight(serviceId, providerServiceId, providerRegion, regionWeightEntityMap);
        }
        if (weight < 0) {
            List regionWeightEntityList = weightFilterEntity.getRegionWeightEntityList();
            weight = WeightEntityWrapper.getWeight(providerServiceId, providerRegion, regionWeightEntityList);
        }
        if (weight < 0) {
            RegionWeightEntity regionWeightEntity = weightFilterEntity.getRegionWeightEntity();
            weight = WeightEntityWrapper.getWeight(providerRegion, regionWeightEntity);
        }

        return weight;
    }
}