DevFW-CICD/ingress-nginx-helm
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix ewma.balance and add unit tests for it

Browse source
This commit is contained in:
Elvin Efendi 2018-05-25 23:53:22 -04:00
parent da3a87646a
commit 04b7356190
2 changed files with 113 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
rootfs/etc/nginx/lua/balancer/ewma.lua
View file

@ -119,12 +119,14 @@ end
function _M.balance(self)
local peers = self.peers
if #peers == 1 then
return peers[1]
local endpoint = peers[1]
if #peers > 1 then
local k = (#peers < PICK_SET_SIZE) and #peers or PICK_SET_SIZE
local peer_copy = util.deepcopy(peers)
endpoint = pick_and_score(peer_copy, k)
end
local k = (#peers < PICK_SET_SIZE) and #peers or PICK_SET_SIZE
local peer_copy = util.deepcopy(peers)
local endpoint = pick_and_score(peer_copy, k)
return endpoint.address, endpoint.port
end
@ -145,6 +147,9 @@ function _M.sync(self, backend)
if not changed then
return
end
self.peers = backend.endpoints
-- TODO: Reset state of EWMA per backend
ngx.shared.balancer_ewma:flush_all()
ngx.shared.balancer_ewma_last_touched_at:flush_all()