RandomLoadBalance算法
public class RandomLoadBalance extends AbstractLoadBalance {
public static final String NAME = "random";
private final Random random = new Random();
protected <T> Invoker<T> doSelect(List<Invoker<T>> invokers, URL url, Invocation invocation) {
int length = invokers.size(); // 总个数
int totalWeight = 0; // 总权重
boolean sameWeight = true; // 权重是否都一样
for (int i = 0; i < length; i++) {
int weight = getWeight(invokers.get(i), invocation);
totalWeight += weight; // 累计总权重
if (sameWeight && i > 0
&& weight != getWeight(invokers.get(i - 1), invocation)) {
sameWeight = false; // 计算所有权重是否一样
}
}
if (totalWeight > 0 && ! sameWeight) {
// 如果权重不相同且权重大于0则按总权重数随机
int offset = random.nextInt(totalWeight);
// 并确定随机值落在哪个片断上
for (int i = 0; i < length; i++) {
offset -= getWeight(invokers.get(i), invocation);
if (offset < 0) {
return invokers.get(i);
}
}
}
// 如果权重相同或权重为0则均等随机
return invokers.get(random.nextInt(length));
}
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
这一段代码的可能刚开始看的时候会有点懵,不知道offset那个参数主要是有什么作用,这里面其实只是设计了一点数学的味道。
具体的思路入下图所示:
假设现在有三个节点,A,B,C,然后我们给他赋予一定的权重,分别是1,2,3,那么我们就可以计算出来三个节点的总权重为1+2+3=6,那么A,B,C三个节点的访问可能性分别是1/6,1/3,1/2。
那么有为什么需要按照权重获取随机数来减去相应的权重获取随机调用的节点呢?
if (totalWeight > 0 && ! sameWeight) {
// 如果权重不相同且权重大于0则按总权重数随机
int offset = random.nextInt(totalWeight);
// 并确定随机值落在哪个片断上
for (int i = 0; i < length; i++) {
offset -= getWeight(invokers.get(i), invocation);
if (offset < 0) {
return invokers.get(i);
}
}
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
我们可以用画图表的方式来较好地理解这段代码里面的思路:
通过对于A,B,C三个几点的权重计算和统计,当总权重为6的时候,offset的可能数值为0,1,2,3,4,5,通过计算发现,当offset为0的时候,访问A节点,当offset为1和2的时候,会访问B节点,当offset为3,4,5的时候会访问C节点,正好他们三个节点的访问概率分别为1/6,1/3,1/2。这样的分配正好满足了我们之前所涉及的权重比例为1:2:3。