今天在看《大型分布式网站架构设计与实践》一书中, 看到了一种比较简单的加权的算法, 在这里记下来:
var serverWeightMap = new Dictionary<string, int>();
serverWeightMap.Add("192.168.1.100", 1);
serverWeightMap.Add("192.168.1.101", 1);
// 权重为 4
serverWeightMap.Add("192.168.1.102", 4);
serverWeightMap.Add("192.168.1.103", 1);
serverWeightMap.Add("192.168.1.104", 1);
// 权重为 3
serverWeightMap.Add("192.168.1.105", 3);
serverWeightMap.Add("192.168.1.106", 1);
// 权重为 2
serverWeightMap.Add("192.168.1.107", 2);
serverWeightMap.Add("192.168.1.108", 1);
serverWeightMap.Add("192.168.1.109", 1);
serverWeightMap.Add("192.168.1.110", 1);
int pos = 0;
// 加权轮询
public static string getRoundRobin()
{
List<string> ipList = new List<string>();
foreach(var key in serverWeightMap.Keys)
{
var weight = serverWeightMap[key];
for(var i = 0; i < weight; i++)
ipList.Add(key);
}
var ip = string.Empty;
lock(pos)
{
if(pos > ipList.Count)
pos = 0;
ip = ipList[pos];
pos++;
}
return ip;
}
// 加权随机
public static string getRandom()
{
List<string> ipList = new List<string>();
foreach(var key in serverWeightMap.Keys)
{
var weight = serverWeightMap[key];
for(var i = 0; i < weight; i++)
ipList.Add(key);
}
var randPos = Convert.ToInt32((new Random()).Next(ipList.Count));
var ip = ipList[randPos];
return ip;
}
上面的两个方法中, 就处理服务器 IP 地址的时候, 根据权重的不同, 在 IP 列表中重复添加 IP 值,权重越大, IP 列表中 IP 值的重复数就越多。