-
在静态代码块中,定义了一个
ipMap
,其中key为服务器的IP地址,value为服务器的权重。这里使用了HashMap
来存储服务器的信息。 -
weightRandom()
方法是主要的负载均衡方法。在该方法中,首先创建了一个ConcurrentHashMap
类型的ipServerMap
,用于保存ipMap
中的信息。 -
将
ipMap
中的key(即服务器的IP地址)添加到ipArrayList
中,并根据权重的数量重复添加。例如,如果某服务器的权重为2,则将其添加两次到ipArrayList
中。 -
通过使用
Random
类生成随机数,从ipArrayList
中随机选择一个位置,然后返回对应的服务器IP地址。 -
在
main
方法中,创建了一个fuzaijunheng02
对象testWeightRandom
,并使用循环调用weightRandom()
方法来模拟负载均衡过程。打印出随机选择的服务器IP地址。
总体而言,这段代码使用带有权重的随机数生成算法从给定的服务器列表中按照权重随机选择一个服务器,实现了加权负载均衡的功能。
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
/*** @author Administrator* @Date 2019/8/20 15:11* 加权随机法:* 获取带有权重的随机数字,随机这种东西,不能看绝对,只能看相对。*/
public class fuzaijunheng02 {
// 1.定义map, key-ip,value-weightstatic
static Map<String, Integer> ipMap = new HashMap<>();
static {
ipMap.put("192.168.13.1", 1);
ipMap.put("192.168.13.2", 2);
ipMap.put("192.168.13.3", 4);
}
public String weightRandom() {
Map<String, Integer> ipServerMap = new ConcurrentHashMap<>();
ipServerMap.putAll(ipMap);
Set<String> ipSet = ipServerMap.keySet();
Iterator<String> ipIterator = ipSet.iterator();
//定义一个list放所有server
ArrayList<String> ipArrayList = new ArrayList<String>();
//循环set,根据set中的可以去得知map中的value,给list中添加对应数字的server数量
while(ipIterator.hasNext()) {
String serverName = ipIterator.next();
Integer weight = ipServerMap.get(serverName);
for (int i = 0; i < weight; i++) {
ipArrayList.add(serverName);//加入的serverName的个数和权重一致
}
}
//循环随机数
Random random = new Random();
//随机数在list数量中取(1-list.size)
int pos = random.nextInt(ipArrayList.size());
String serverNameReturn = ipArrayList.get(pos);
return serverNameReturn;
}
public static void main(String[] args) {
fuzaijunheng02 testWeightRandom = new fuzaijunheng02();
for (int i = 0; i < 10; i++) {
String server = testWeightRandom.weightRandom();
System.out.println(server);
}
}
}