当前位置:   article > 正文

7种常见的生产级负载均衡算法_hcqmvkn

hcqmvkn

准备测试数据

  1. package com.example.demo.balance;
  2. import java.util.*;
  3. /**
  4. * @author liwenchao
  5. */
  6. public class ServerIps {
  7. public static final List<String> LIST = Arrays.asList(
  8. "192.168.0.1",
  9. "192.168.0.2",
  10. "192.168.0.3",
  11. "192.168.0.4",
  12. "192.168.0.5",
  13. "192.168.0.6",
  14. "192.168.0.7",
  15. "192.168.0.8",
  16. "192.168.0.9",
  17. "192.168.0.10"
  18. );
  19. public static final Map<String, Integer> WEIGHT_LIST = new HashMap<String, Integer>();
  20. static {
  21. // 权重之和为50
  22. WEIGHT_LIST.put("192.168.0.1", 1);
  23. WEIGHT_LIST.put("192.168.0.2", 8);
  24. WEIGHT_LIST.put("192.168.0.3", 3);
  25. WEIGHT_LIST.put("192.168.0.4", 6);
  26. WEIGHT_LIST.put("192.168.0.5", 5);
  27. WEIGHT_LIST.put("192.168.0.6", 5);
  28. WEIGHT_LIST.put("192.168.0.7", 4);
  29. WEIGHT_LIST.put("192.168.0.8", 7);
  30. WEIGHT_LIST.put("192.168.0.9", 2);
  31. WEIGHT_LIST.put("192.168.0.10", 9);
  32. }
  33. public static final Map<String, Integer> WEIGHT_ROUND_ROBIN_LIST = new HashMap<String, Integer>();
  34. static {
  35. WEIGHT_ROUND_ROBIN_LIST.put("192.168.0.1", 5);
  36. WEIGHT_ROUND_ROBIN_LIST.put("192.168.0.2", 1);
  37. WEIGHT_ROUND_ROBIN_LIST.put("192.168.0.3", 1);
  38. }
  39. /**
  40. * 服务器当前的活跃数
  41. */
  42. public static final Map<String, Integer> ACTIVITY_LIST = new LinkedHashMap<String, Integer>();
  43. static {
  44. ACTIVITY_LIST.put("192.168.0.1", 2);
  45. ACTIVITY_LIST.put("192.168.0.2", 0);
  46. ACTIVITY_LIST.put("192.168.0.3", 1);
  47. ACTIVITY_LIST.put("192.168.0.4", 3);
  48. ACTIVITY_LIST.put("192.168.0.5", 0);
  49. ACTIVITY_LIST.put("192.168.0.6", 1);
  50. ACTIVITY_LIST.put("192.168.0.7", 4);
  51. ACTIVITY_LIST.put("192.168.0.8", 2);
  52. ACTIVITY_LIST.put("192.168.0.9", 7);
  53. ACTIVITY_LIST.put("192.168.0.10", 3);
  54. }
  55. }

1,随机算法-RandomLoadBalance

  1. package com.example.demo.balance;
  2. import java.util.Random;
  3. /**
  4. * 随机算法
  5. *
  6. * @author liwenchao
  7. */
  8. public class RandomLoadBalance {
  9. public static String getServer() {
  10. // 生成一个随机数作为list的下标值
  11. Random random = new Random();
  12. int randomPos = random.nextInt(ServerIps.LIST.size());
  13. return ServerIps.LIST.get(randomPos);
  14. }
  15. public static void main(String[] args) {
  16. // 连续调用10次
  17. for (int i = 0; i < 10; i++) {
  18. System.out.println(getServer());
  19. }
  20. }
  21. }

运行结果

192.168.0.8
192.168.0.6
192.168.0.6
192.168.0.6
192.168.0.9
192.168.0.3
192.168.0.9
192.168.0.1
192.168.0.1
192.168.0.3

当调用次数比较少时,Random 产生的随机数可能会比较集中,此时多数请求会落到同一台服务器上,只有在经过多次请求后,才能使调用请求进行“均匀”分配。调用量少这一点并没有什么关系,负载均衡机制不正是为了应对请求量多的情况吗,所以随机算法也是用得比较多的一种算法。

但是,上面的随机算法适用于每天机器的性能差不多的时候,实际上,生产中可能某些机器的性能更高一点,它可以处理更多的请求,所以,我们可以对每台服务器设置一个权重。
在ServerIps类中增加服务器权重对应关系MAP,权重之和为50。

那么现在的随机算法应该要改成权重随机算法,当调用量比较多的时候,服务器使用的分布应该近似对应权重的分布。

2,权重随机算法-RoundRobinLoadBalance

简单的实现思路是,把每个服务器按它所对应的服务器进行复制,具体看代码更加容易理解

  1. package com.example.demo.balance;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import java.util.Random;
  5. /**
  6. * 权重随机算法
  7. *
  8. * @author liwenchao
  9. */
  10. public class WeightRandomLoadBalance {
  11. public static String getServer() {
  12. //生成一个随机数作为list的下标值
  13. List<String> ips = new ArrayList<>();
  14. for (String ip : ServerIps.WEIGHT_LIST.keySet()) {
  15. Integer weight = ServerIps.WEIGHT_LIST.get(ip);
  16. //按权重进行复制
  17. for (int i = 0; i < weight; i++) {
  18. ips.add(ip);
  19. }
  20. }
  21. Random random = new Random();
  22. int randomPos = random.nextInt(ips.size());
  23. return ips.get(randomPos);
  24. }
  25. public static void main(String[] args) {
  26. // 连续调用10次
  27. for (int i = 0; i < 10; i++) {
  28. System.out.println(getServer());
  29. }
  30. }
  31. }

执行结果

192.168.0.1
192.168.0.2
192.168.0.3
192.168.0.4
192.168.0.5
192.168.0.6
192.168.0.7
192.168.0.8
192.168.0.9
192.168.0.10
192.168.0.1

这种实现方法在遇到权重之和特别大的时候就会比较消耗内存,因为需要对ip地址进行复制,权重之和越大那么上文中的ips就需要越多的内存,下面介绍另外一种实现思路。
假设我们有一组服务器 servers = [A, B, C],他们对应的权重为 weights = [5, 3, 2],权重总和为10。现在把这些权重值平铺在一维坐标值上,[0, 5) 区间属于服务器 A,[5, 8) 区间属于服务器 B,[8, 10) 区间属于服务器 C。接下来通过随机数生成器生成一个范围在 [0, 10) 之间的随机数,然后计算这个随机数会落到哪个区间上。比如数字3会落到服务器 A 对应的区间上,此时返回服务器 A 即可。权重越大的机器,在坐标轴上对应的区间范围就越大,因此随机数生成器生成的数字就会有更大的概率落到此区间内。只要随机数生成器产生的随机数分布性很好,在经过多次选择后,每个服务器被选中的次数比例接近其权重比例。比如,经过一万次选择后,服务器 A 被选中的次数大约为5000次,服务器 B 被选中的次数约为3000次,服务器 C 被选中的次数约为2000次。
假设现在随机数offset=7:

1offset<5 is false,所以不在[0, 5)区间,将offset = offset - 5(offset=2)
2offset<3 is true,所以处于[5, 8)区间,所以应该选用B服务器 实现如下:

  1. package com.example.demo.balance;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import java.util.Random;
  5. /**
  6. * 权重随机算法
  7. *
  8. * @author liwenchao
  9. */
  10. public class WeightRandomLoadBalance {
  11. public static String getServer() {
  12. int totalWeight = 0;
  13. //如果所有权重都相等,那么随机一个ip就好了
  14. boolean sameWeight = true;
  15. Object[] weights = ServerIps.WEIGHT_LIST.values().toArray();
  16. for (int i = 0; i < weights.length; i++) {
  17. Integer weight = (Integer) weights[i];
  18. totalWeight += weight;
  19. if (sameWeight && i > 0 && !weight.equals(weights[i - 1])) {
  20. sameWeight = false;
  21. }
  22. }
  23. Random random = new Random();
  24. int randomPos = random.nextInt(totalWeight);
  25. if (!sameWeight) {
  26. for (String ip : ServerIps.WEIGHT_LIST.keySet()) {
  27. Integer value = ServerIps.WEIGHT_LIST.get(ip);
  28. if (randomPos < value) {
  29. return ip;
  30. }
  31. randomPos = randomPos - value;
  32. }
  33. }
  34. return (String) ServerIps.WEIGHT_LIST.keySet().toArray()[new Random().nextInt(ServerIps.WEIGHT_LIST.size())];
  35. }
  36. public static void main(String[] args) {
  37. // 连续调用10次
  38. for (int i = 0; i < 10; i++) {
  39. System.out.println(getServer());
  40. }
  41. }
  42. }

这就是另外一种权重随机算法。

3,轮询算法-RoundRobinLoadBalance

  1. package com.example.demo.balance;
  2. import java.util.concurrent.atomic.AtomicInteger;
  3. /**
  4. * @author liwenchao
  5. */
  6. public class RoundRobinLoadBalance {
  7. /**
  8. * 当前循环的位置
  9. */
  10. private static final AtomicInteger POS = new AtomicInteger(0);
  11. public static String getServer() {
  12. if (POS.get() >= ServerIps.LIST.size()) {
  13. POS.set(0);
  14. }
  15. return ServerIps.LIST.get(POS.getAndIncrement());
  16. }
  17. public static void main(String[] args) {
  18. // 连续调用10次
  19. for (int i = 0; i < 11; i++) {
  20. System.out.println(getServer());
  21. }
  22. }
  23. }

执行结果

192.168.0.1
192.168.0.2
192.168.0.3
192.168.0.4
192.168.0.5
192.168.0.6
192.168.0.7
192.168.0.8
192.168.0.9
192.168.0.10
192.168.0.1

这种算法很简单,也很公平,每台服务轮流来进行服务,但是有的机器性能好,所以能者多劳,和随机算法一下,加上权重这个维度之后,其中一种实现方法就是复制法,这里就不演示了,这种复制算法的缺点和随机算法的是一样的,比较消耗内存,那么自然就有其他实现方法。

4,权重轮询算法-WeightRoundRobinLoadBalance

这种算法需要加入一个概念:调用编号,比如第1次调用为1, 第2次调用为2, 第100次调用为100,调用编号是递增的,所以我们可以根据这个调用编号推算出服务器。
假设我们有三台服务器 servers = [A, B, C],对应的权重为 weights = [ 2, 5, 1], 总权重为8,我们可以理解为有8台“服务器”,这是8台“不具有并发功能”,其中有2台为A,5台为B,1台为C,一次调用过来的时候,需要按顺序访问,比如有10次调用,那么服务器调用顺序为AABBBBBCAA,调用编号会越来越大,而服务器是固定的,所以需要把调用编号“缩小”,这里对调用编号进行取余,除数为总权重和,比如:

1号调用,1%8=1;
2号调用,2%8=2;
3号调用,3%8=3;
8号调用,8%8=0;
9号调用,9%8=1;
100号调用,100%8=4; 我们发现调用编号可以被缩小为0-7之间的8个数字,问题是怎么根据这个8个数字找到对应的服务器呢?和我们随机算法类似,这里也可以把权重想象为一个坐标轴

“0-----2-----7-----8”

1号调用,1%8=1,offset = 1, offset <= 2 is true,取A;
2号调用,2%8=2;offset = 2,offset <= 2 is true, 取A;
3号调用,3%8=3;offset = 3, offset <= 2 is false, offset = offset - 2, offset = 1, offset <= 5,取B
8号调用,8%8=0;offset = 0, 特殊情况,offset = 8,offset <= 2 is false, offset = offset - 2, offset = 6, offset  <= 5 is false, offset = offset - 5, offset = 1, offset <= 1 is true, 取C;
9号调用,9%8=1;// ...
100号调用,100%8=4; //...

  1. package com.example.demo.balance;
  2. import java.util.Random;
  3. import java.util.concurrent.atomic.AtomicInteger;
  4. /**
  5. * @author liwenchao
  6. */
  7. public class WeightRoundRobinLoadBalance {
  8. private static final AtomicInteger POS = new AtomicInteger(0);
  9. public static String getServer() {
  10. int totalWeight = 0;
  11. //如果所有权重都相等,那么随机一个ip就好了
  12. boolean sameWeight = true;
  13. Object[] weights = ServerIps.WEIGHT_LIST.values().toArray();
  14. for (int i = 0; i < weights.length; i++) {
  15. Integer weight = (Integer) weights[i];
  16. totalWeight += weight;
  17. if (sameWeight && i > 0 && !weight.equals(weights[i - 1])) {
  18. sameWeight = false;
  19. }
  20. }
  21. int sequenceNum = POS.getAndIncrement();
  22. int offset = sequenceNum % totalWeight;
  23. offset = offset == 0 ? totalWeight : offset;
  24. if (!sameWeight) {
  25. for (String ip : ServerIps.WEIGHT_LIST.keySet()) {
  26. Integer weight = ServerIps.WEIGHT_LIST.get(ip);
  27. if (offset <= weight) {
  28. return ip;
  29. }
  30. offset = offset - weight;
  31. }
  32. }
  33. if (sequenceNum >= ServerIps.WEIGHT_LIST.size() - 1) {
  34. POS.set(0);
  35. }
  36. return (String) ServerIps.WEIGHT_LIST.keySet().toArray()[sequenceNum];
  37. }
  38. public static void main(String[] args) {
  39. // 连续调用11次
  40. for (int i = 0; i < 12; i++) {
  41. System.out.println(getServer());
  42. }
  43. }
  44. }

执行结果

192.168.0.7
192.168.0.2
192.168.0.2
192.168.0.2
192.168.0.2
192.168.0.2
192.168.0.2
192.168.0.2
192.168.0.2
192.168.0.1
192.168.0.4
192.168.0.4

但是这种算法有一个缺点:一台服务器的权重特别大的时候,他需要连续的的处理请求,但是实际上我们想达到的效果是,对于100次请求,只要有100*8/50=16次就够了,这16次不一定要连续的访问,比如假设我们有三台服务器 servers = [A, B, C],对应的权重为 weights = [5, 1, 1] , 总权重为7,那么上述这个算法的结果是:AAAAABC,那么如果能够是这么一个结果呢:AABACAA,把B和C平均插入到5个A中间,这样是比较均衡的了。

我们这里可以改成平滑加权轮询。

5,平滑加权轮询算法-WeightRoundSmoothRobinLoadBalance

思路:每个服务器对应两个权重,分别为 weight 和 currentWeight。其中 weight 是固定的,currentWeight 会动态调整,初始值为0。当有新的请求进来时,遍历服务器列表,让它的 currentWeight 加上自身权重。遍历完成后,找到最大的 currentWeight,并将其减去权重总和,然后返回相应的服务器即可。

请求编号

currentWeight 数组 (current_weight += weight)

选择结果(max(currentWeight))

减去权重总和后的currentWeight 数组(max(currentWeight) -= sum(weight))

1

[5, 1, 1]

A

[-2, 1, 1]

2

[3, 2, 2] 

A

[-4, 2, 2]

3

[1, 3, 3]

B

[1, -4, 3]

4

[6, -3, 4]

A

[-1, -3, 4]

5

[4, -2, 5]

C

[4, -2, -2]

6

[9, -1, -1]

A

[2, -1, -1]

7

[7, 0, 0]

A

[0, 0, 0]

如上,经过平滑性处理后,得到的服务器序列为 [A, A, B, A, C, A, A],相比之前的序列 [A, A, A, A, A, B, C],分布性要好一些。初始情况下 currentWeight = [0, 0, 0],第7个请求处理完后,currentWeight 再次变为 [0, 0, 0]。

  1. package com.example.demo.balance;
  2. /**
  3. * 增加一个Weight类,用来保存ip, weight(固定不变的原始权重), currentWeight(当前会变化的权重)
  4. */
  5. public class Weight {
  6. private String ip;
  7. private Integer weight;
  8. private Integer currentWeight;
  9. public Weight(String ip, Integer weight, Integer currentWeight) {
  10. this.ip = ip;
  11. this.weight = weight;
  12. this.currentWeight = currentWeight;
  13. }
  14. public String getIp() {
  15. return ip;
  16. }
  17. public void setIp(String ip) {
  18. this.ip = ip;
  19. }
  20. public Integer getWeight() {
  21. return weight;
  22. }
  23. public void setWeight(Integer weight) {
  24. this.weight = weight;
  25. }
  26. public Integer getCurrentWeight() {
  27. return currentWeight;
  28. }
  29. public void setCurrentWeight(Integer currentWeight) {
  30. this.currentWeight = currentWeight;
  31. }
  32. }
  1. package com.example.demo.balance;
  2. import java.util.HashMap;
  3. import java.util.Map;
  4. /**
  5. * @author liwenchao
  6. */
  7. public class WeightRoundSmoothRobinLoadBalance {
  8. private static Map<String, Weight> weightMap = new HashMap<>();
  9. public static String getServer() {
  10. int totalWeight = ServerIps.WEIGHT_ROUND_ROBIN_LIST.values().stream().reduce(0, Integer::sum);
  11. //初始化weightMap,初始时将currentWeight赋值为weight
  12. if (weightMap.isEmpty()) {
  13. ServerIps.WEIGHT_ROUND_ROBIN_LIST.forEach((key, value) -> {
  14. weightMap.put(key, new Weight(key, value, value));
  15. });
  16. }
  17. // 找出currentWeight最大值
  18. Weight maxCurrentWeight = null;
  19. for (Weight weight : weightMap.values()) {
  20. if (maxCurrentWeight == null || weight.getCurrentWeight() > maxCurrentWeight.getCurrentWeight()) {
  21. maxCurrentWeight = weight;
  22. }
  23. }
  24. // 将maxCurrentWeight减去总权重和
  25. maxCurrentWeight.setCurrentWeight(maxCurrentWeight.getCurrentWeight() - totalWeight);
  26. // 所有的ip的currentWeight统一加上原始权重
  27. for (Weight weight : weightMap.values()) {
  28. weight.setCurrentWeight(weight.getCurrentWeight() + weight.getWeight());
  29. }
  30. // 返回maxCurrentWeight所对应的ip
  31. return maxCurrentWeight.getIp();
  32. }
  33. public static void main(String[] args) {
  34. // 连续调用10次
  35. for (int i = 0; i < 10; i++) {
  36. System.out.println(getServer());
  37. }
  38. }
  39. }

执行结果

192.168.0.1
192.168.0.1
192.168.0.2
192.168.0.1
192.168.0.3
192.168.0.1
192.168.0.1
192.168.0.1
192.168.0.1
192.168.0.2

这就是轮询算法,一个循环很简单,但是真正在实际运用的过程中需要思考更多。

6,一致性哈希算法-ConsistentHashLoadBalance

服务器集群接收到一次请求调用时,可以根据根据请求的信息,比如客户端的ip地址,或请求路径与请求参数等信息进行哈希,可以得出一个哈希值,特点是对于相同的ip地址,或请求路径和请求参数哈希出来的值是一样的,只要能再增加一个算法,能够把这个哈希值映射成一个服务端ip地址,就可以使相同的请求(相同的ip地址,或请求路径和请求参数)落到同一服务器上。
因为客户端发起的请求情况是无穷无尽的(客户端地址不同,请求参数不同等等),所以对于的哈希值也是无穷大的,所以我们不可能把所有的哈希值都进行映射到服务端ip上,所以这里需要用到哈希环。如下图:

●哈希值如果需要ip1和ip2之间的,则应该选择ip2作为结果;
●哈希值如果需要ip2和ip3之间的,则应该选择ip3作为结果;
●哈希值如果需要ip3和ip4之间的,则应该选择ip4作为结果;
●哈希值如果需要ip4和ip1之间的,则应该选择ip1作为结果;

上面这情况是比较均匀情况,如果出现ip4服务器不存在,那就是这样了:

会发现,ip3和ip1直接的范围是比较大的,会有更多的请求落在ip1上,这是不“公平的”,解决这个问题需要加入虚拟节点,比如: 

其中ip2-1, ip3-1就是虚拟结点,并不能处理节点,而是等同于对应的ip2和ip3服务器。 实际上,这只是处理这种不均衡性的一种思路,实际上就算哈希环本身是均衡的,你也可以增加更多的虚拟节点来使这个环更加平滑,比如: 

 这个彩环也是“公平的”,并且只有ip1,2,3,4是实际的服务器ip,其他的都是虚拟ip。 那么我们怎么来实现呢? 对于我们的服务端ip地址,我们肯定知道总共有多少个,需要多少个虚拟节点也有我们自己控制,虚拟节点越多则流量越均衡,另外哈希算法也是很关键的,哈希算法越散列流量也将越均衡。 实现:

  1. package com.example.demo.balance;
  2. import java.util.SortedMap;
  3. import java.util.TreeMap;
  4. /**
  5. * 一致性哈希算法
  6. *
  7. * @author liwenchao
  8. */
  9. public class ConsistentHashLoadBalance {
  10. private static final SortedMap<Integer, String> INTEGER_STRING_TREE_MAP = new TreeMap<>();
  11. /**
  12. * 虚拟节点个数
  13. */
  14. private static final int VIRTUAL_NODES = 160;
  15. static {
  16. //对每个真实节点添加虚拟节点,虚拟节点会根据哈希算法进行散列
  17. for (String ip : ServerIps.LIST) {
  18. for (int i = 0; i < VIRTUAL_NODES; i++) {
  19. int hash = getHash(ip + "VN" + i);
  20. INTEGER_STRING_TREE_MAP.put(hash, ip);
  21. }
  22. }
  23. }
  24. private static String getServer(String client) {
  25. int hash = getHash(client);
  26. //得到大于该Hash值的排好序的Map
  27. SortedMap<Integer, String> subMap = INTEGER_STRING_TREE_MAP.tailMap(hash);
  28. //大于该hash值的第一个元素的位置
  29. Integer nodeIndex = subMap.firstKey();
  30. //如果不存在大于该hash值的元素,则返回根节点
  31. if (nodeIndex == null) {
  32. nodeIndex = INTEGER_STRING_TREE_MAP.firstKey();
  33. }
  34. // 返回对应的虚拟节点名称
  35. return subMap.get(nodeIndex);
  36. }
  37. private static int getHash(String str) {
  38. final int p = 16777619;
  39. int hash = (int) 2166136261L;
  40. for (int i = 0; i < str.length(); i++) {
  41. hash = (hash ^ str.charAt(i)) * p;
  42. }
  43. hash += hash << 13;
  44. hash ^= hash >> 7;
  45. hash += hash << 3;
  46. hash ^= hash >> 17;
  47. hash += hash << 5;
  48. // 如果算出来的值为负数则取其绝对值
  49. if (hash < 0) {
  50. hash = Math.abs(hash);
  51. }
  52. return hash;
  53. }
  54. public static void main(String[] args) {
  55. // 连续调用10次,随机10个client
  56. for (int i = 0; i < 10; i++) {
  57. System.out.println(getServer("client" + i));
  58. }
  59. }
  60. }

执行结果

192.168.0.9
192.168.0.9
192.168.0.6
192.168.0.5
192.168.0.10
192.168.0.3
192.168.0.4
192.168.0.5
192.168.0.5
192.168.0.8

7,最小活跃数算法-LeastActiveLoadBalance

前面几种方法主要目标是使服务端分配到的调用次数尽量均衡,但是实际情况是这样吗?调用次数相同,服务器的负载就均衡吗?当然不是,这里还要考虑每次调用的时间,而最小活跃数算法则是解决这种问题的。
活跃调用数越小,表明该服务提供者效率越高,单位时间内可处理更多的请求。此时应优先将请求分配给该服务提供者。在具体实现中,每个服务提供者对应一个活跃数。初始情况下,所有服务提供者活跃数均为0。每收到一个请求,活跃数加1,完成请求后则将活跃数减1。在服务运行一段时间后,性能好的服务提供者处理请求的速度更快,因此活跃数下降的也越快,此时这样的服务提供者能够优先获取到新的服务请求、这就是最小活跃数负载均衡算法的基本思想。除了最小活跃数,最小活跃数算法在实现上还引入了权重值。所以准确的来说,最小活跃数算法是基于加权最小活跃数算法实现的。举个例子说明一下,在一个服务提供者集群中,有两个性能优异的服务提供者。某一时刻它们的活跃数相同,则会根据它们的权重去分配请求,权重越大,获取到新请求的概率就越大。如果两个服务提供者权重相同,此时随机选择一个即可。
实现:
因为活跃数是需要服务器请求处理相关逻辑配合的,一次调用开始时活跃数+1,结束是活跃数-1,所以这里就不对这部分逻辑进行模拟了,直接使用一个map来进行模拟。

  1. package com.example.demo.balance;
  2. import java.util.*;
  3. /**
  4. * 最小活跃数算法
  5. *
  6. * @author liwenchao
  7. */
  8. public class LeastActiveLoadBalance {
  9. private static String getServer() {
  10. // 找出当前活跃数最小的服务器
  11. Optional<Integer> minValue = ServerIps.ACTIVITY_LIST.values().stream().min(Comparator.naturalOrder());
  12. if (minValue.isPresent()) {
  13. List<String> minActivityIps = new ArrayList<>();
  14. ServerIps.ACTIVITY_LIST.forEach((ip, activity) -> {
  15. if (activity.equals(minValue.get())) {
  16. minActivityIps.add(ip);
  17. }
  18. });
  19. // 最小活跃数的ip有多个,则根据权重来选,权重大的优先
  20. if (minActivityIps.size() > 1) {
  21. // 过滤出对应的ip和权重
  22. Map<String, Integer> weightList = new LinkedHashMap<String, Integer>();
  23. ServerIps.WEIGHT_LIST.forEach((ip, weight) -> {
  24. if (minActivityIps.contains(ip)) {
  25. weightList.put(ip, ServerIps.WEIGHT_LIST.get(ip));
  26. }
  27. });
  28. int totalWeight = 0;
  29. // 如果所有权重都相等,那么随机一个ip就好了
  30. boolean sameWeight = true;
  31. Object[] weights = weightList.values().toArray();
  32. for (int i = 0; i < weights.length; i++) {
  33. Integer weight = (Integer) weights[i];
  34. totalWeight += weight;
  35. if (sameWeight && i > 0 && !weight.equals(weights[i - 1])) {
  36. sameWeight = false;
  37. }
  38. }
  39. Random random = new Random();
  40. int randomPos = random.nextInt(totalWeight);
  41. if (!sameWeight) {
  42. for (String ip : weightList.keySet()) {
  43. Integer value = weightList.get(ip);
  44. if (randomPos < value) {
  45. return ip;
  46. }
  47. randomPos = randomPos - value;
  48. }
  49. }
  50. return (String) weightList.keySet().toArray()[new Random().nextInt(weightList.size())];
  51. } else {
  52. return minActivityIps.get(0);
  53. }
  54. } else {
  55. return (String) ServerIps.WEIGHT_LIST.keySet().toArray()[new Random().nextInt(ServerIps.WEIGHT_LIST.size())];
  56. }
  57. }
  58. public static void main(String[] args) {
  59. // 连续调用10次,随机10个client
  60. for (int i = 0; i < 10; i++) {
  61. System.out.println(getServer());
  62. }
  63. }
  64. }

这里因为不会对活跃数进行操作,所以结果是固定的(担任在随机权重的时候会随机)。

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号