利用Lua实现基于延迟动态调整HAProxy服务器权重(weight)

easyt***

将下面的Lua代码上传到路由器上，假设命名为dynamic_set_weight.lua。

1. local function calculate_weight(ping_value,eresp)
   
2.     -- from 1 to 256
   
3.     return math.ceil(256/math.exp(0.1*(ping_value+eresp)))
   
4. end
   
5. local function adjust_weight()
   
6.     while true do
   
7.         for _, backend in pairs(core.backends) do
   
8.             for _, server in pairs(backend.servers) do
   
9.                 local stat = server:get_stats()
   
10.                 if stat['check_desc'] == 'Layer4 check passed' and stat['bin'] > 1000 then
    
11.                     -- adjust weight based on the ping value and the number of eresp
    
12.                     server:set_weight(calculate_weight(stat['check_duration'],stat['eresp']))
    
13.                 end
    
14.             end
    
15.         end
    
16.         core.msleep(10000) -- 10s
    
17.     end
    
18. end
    
19. 
    
20. core.register_task(adjust_weight)

复制代码

假设上传的位置为/usr/bin，则在HAProxy的配置文件里，在global下添加一行

1. lua-load /usr/bin/dynamic_set_weight.lua

复制代码

重启HAProxy（OpenWRT上的命令为/etc/init.d/haproxy restart），一段时间后效果如图所示：

可以看到权重已经根据延迟做了调整。调整每10s进行一次，可以修改core.msleep括号内的数字对运行间隔进行调整。

更新：
2020年2月14日：修改节点在线状态判断方法

easyt***

更新了权重计算方法

easyt***

修改节点在线状态判断方法