十三:Dubbo负载均衡(四)LeastActiveLoadBalance

    //org.apache.dubbo.rpc.cluster.loadbalance.LeastActiveLoadBalance#doSelect
protected <T> Invoker<T> doSelect(List<Invoker<T>> invokers, URL url, Invocation invocation) {
    // 总个数
    int length = invokers.size(); 
    // 最小的活跃数
    int leastActive = -1; 
    // 相同最小活跃数的个数
    int leastCount = 0; 
    // 相同最小活跃数的下标
    int[] leastIndexs = new int[length]; 
    // 总权重
    int totalWeight = 0; 
    // 第一个权重,用于于计算是否相同
    int firstWeight = 0; 
    // 是否所有权重相同
    boolean sameWeight = true; 
    // 遍历所有 provider
    for (int i = 0; i < length; i++) {
        // 获取当前 provider
        Invoker<T> invoker = invokers.get(i);
        // 活跃数
        int active = RpcStatus.getStatus(invoker.getUrl(), invocation.getMethodName()).getActive(); 
        // 获取权重
        int weight = invoker.getUrl().getMethodParameter(invocation.getMethodName(), Constants.WEIGHT_KEY, Constants.DEFAULT_WEIGHT);
        if (leastActive == -1 || active < leastActive) { 
            // 发现更小的活跃数,重新开始
            // 记录最小活跃数
            leastActive = active; 
            // 重新统计相同最小活跃数的个数
            leastCount = 1; 
            // 重新记录最小活跃数下标
            leastIndexs[0] = i; 
            // 重新累计总权重
            totalWeight = weight; 
            // 记录第一个权重
            firstWeight = weight; 
            // 还原权重相同标识
            sameWeight = true; 
        } else if (active == leastActive) { 
            // 累计相同最小的活跃数
            // 累计相同最小活跃数下标
            leastIndexs[leastCount ++] = i;
            // 累计总权重
            totalWeight += weight; 
            // 判断所有权重是否一样
            if (sameWeight && i > 0  && weight != firstWeight) {
                sameWeight = false;
            }
        }
    }
   
    if (leastCount == 1) {
        // 如果只有一个最小则直接返回
        return invokers.get(leastIndexs[0]);
    }
    
    // 最小活跃数有多个 provider
    // 所有 provider 的权重不一致情况,
    if (!sameWeight && totalWeight > 0) {
        // 如果权重不相同且权重大于0则按总权重数随机
        int offsetWeight = random.nextInt(totalWeight);
        // 并确定随机值落在哪个片断上
        for (int i = 0; i < leastCount; i++) {
            int leastIndex = leastIndexs[i];
            offsetWeight -= getWeight(invokers.get(leastIndex), invocation);
            if (offsetWeight <= 0)
                return invokers.get(leastIndex);
        }
    }
    // 如果权重相同或权重为0则均等随机
    return invokers.get(leastIndexs[random.nextInt(leastCount)]);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值