jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-03-01 15:10:56 +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

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

@@ -28,31 +28,19 @@ local function randomBool()
return math.random() >= 0.5
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
current_count = redis.call('HGET', hset_key, current_target)
local hash = redis.call('HGETALL', hset_key)
for i = 1, #hash, 2 do
local addr = hash[i]
local count = hash[i+1]
if is_healthy(addr) then
if tonumber(count) < tonumber(current_count) then
current_target = addr
current_count = count
elseif count == current_count and randomBool() then
current_target = addr
current_count = count
end
end
end
for i = 4, #KEYS do
if redis.call('HEXISTS', hset_key, KEYS[i]) == 1 then
local count = redis.call('HGET', hset_key, KEYS[i])
if tonumber(count) < tonumber(current_count) then
current_target = KEYS[i]
current_count = count
elseif count == current_count and randomBool() then
current_target = KEYS[i]
end
end
end
end
redis.call("HINCRBY", hset_key, current_target, 1)

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
if redis.call('HEXISTS', hset_key, target) == 1 then
current_count = redis.call('HGET', hset_key, target)
local hash = redis.call('HGETALL', hset_key)
for i = 1, #hash, 2 do
local addr = hash[i]
local count = hash[i+1]
if is_healthy(addr) then
for i = 4, #KEYS do
if redis.call('HEXISTS', hset_key, KEYS[i]) == 1 then
local count = redis.call('HGET', hset_key, KEYS[i])
if tonumber(count) < tonumber(current_count) then
target = addr
target = KEYS[i]
current_count = count
elseif count == current_count and randomBool() then
target = addr
current_count = count
target = KEYS[i]
end
end
end