jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-06-09 12:47:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

opt(ai-load-balancer): update global least request lua script for ai-load-balancer (#2945)

Browse Source
This commit is contained in:
rinfx
2025-09-23 19:24:33 +08:00
committed by GitHub
parent 623c8da8d8
commit 9a3ccff4c8
2 changed files with 16 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
plugins/wasm-go/extensions/ai-load-balancer/global_least_request/lb_policy.go
View File

@@ -28,28 +28,16 @@ local function randomBool()
return math.random() >= 0.5 return math.random() >= 0.5
end end
local function is_healthy(addr)
for i = 4, #KEYS do
if addr == KEYS[i] then
return true
end
end
return false
end
if redis.call('HEXISTS', hset_key, current_target) == 1 then if redis.call('HEXISTS', hset_key, current_target) == 1 then
current_count = redis.call('HGET', hset_key, current_target) current_count = redis.call('HGET', hset_key, current_target)
local hash = redis.call('HGETALL', hset_key) for i = 4, #KEYS do
for i = 1, #hash, 2 do if redis.call('HEXISTS', hset_key, KEYS[i]) == 1 then
local addr = hash[i] local count = redis.call('HGET', hset_key, KEYS[i])
local count = hash[i+1]
if is_healthy(addr) then
if tonumber(count) < tonumber(current_count) then if tonumber(count) < tonumber(current_count) then
current_target = addr current_target = KEYS[i]
current_count = count current_count = count
elseif count == current_count and randomBool() then elseif count == current_count and randomBool() then
current_target = addr current_target = KEYS[i]
current_count = count
end end
end end
end end

13
plugins/wasm-go/extensions/ai-load-balancer/prefix_cache/lb_policy.go
View File

@@ -121,17 +121,14 @@ if target == "" then
target = default_target target = default_target
if redis.call('HEXISTS', hset_key, target) == 1 then if redis.call('HEXISTS', hset_key, target) == 1 then
current_count = redis.call('HGET', hset_key, target) current_count = redis.call('HGET', hset_key, target)
local hash = redis.call('HGETALL', hset_key) for i = 4, #KEYS do
for i = 1, #hash, 2 do if redis.call('HEXISTS', hset_key, KEYS[i]) == 1 then
local addr = hash[i] local count = redis.call('HGET', hset_key, KEYS[i])
local count = hash[i+1]
if is_healthy(addr) then
if tonumber(count) < tonumber(current_count) then if tonumber(count) < tonumber(current_count) then
target = addr target = KEYS[i]
current_count = count current_count = count
elseif count == current_count and randomBool() then elseif count == current_count and randomBool() then
target = addr target = KEYS[i]
current_count = count
end end
end end
end end