e625c2507a
* Fix chashsubset backend with canary * use floor for buckets calculation to remove hot spot * Remove testing code
106 lines
2.8 KiB
Lua
106 lines
2.8 KiB
Lua
function mock_ngx(mock)
|
|
local _ngx = mock
|
|
setmetatable(_ngx, {__index = _G.ngx})
|
|
_G.ngx = _ngx
|
|
end
|
|
|
|
local function get_test_backend(n_endpoints)
|
|
local backend = {
|
|
name = "my-dummy-backend",
|
|
["upstreamHashByConfig"] = {
|
|
["upstream-hash-by"] = "$request_uri",
|
|
["upstream-hash-by-subset"] = true,
|
|
["upstream-hash-by-subset-size"] = 3
|
|
},
|
|
endpoints = {}
|
|
}
|
|
|
|
for i = 1, n_endpoints do
|
|
backend.endpoints[i] = { address = "10.184.7." .. tostring(i), port = "8080", maxFails = 0, failTimeout = 0 }
|
|
end
|
|
|
|
return backend
|
|
end
|
|
|
|
describe("Balancer chash subset", function()
|
|
local balancer_chashsubset
|
|
|
|
before_each(function()
|
|
mock_ngx({ var = { request_uri = "/alma/armud" }})
|
|
balancer_chashsubset = require("balancer.chashsubset")
|
|
end)
|
|
|
|
describe("balance()", function()
|
|
it("returns peers from the same subset", function()
|
|
|
|
local backend = get_test_backend(9)
|
|
|
|
local instance = balancer_chashsubset:new(backend)
|
|
|
|
instance:sync(backend)
|
|
|
|
local first_node = instance:balance()
|
|
local subset_id
|
|
local endpoint_strings
|
|
|
|
local function has_value (tab, val)
|
|
for _, value in ipairs(tab) do
|
|
if value == val then
|
|
return true
|
|
end
|
|
end
|
|
|
|
return false
|
|
end
|
|
|
|
for id, endpoints in pairs(instance["subsets"]) do
|
|
endpoint_strings = {}
|
|
for _, endpoint in pairs(endpoints) do
|
|
local endpoint_string = endpoint.address .. ":" .. endpoint.port
|
|
table.insert(endpoint_strings, endpoint_string)
|
|
if first_node == endpoint_string then
|
|
-- found the set of first_node
|
|
subset_id = id
|
|
end
|
|
end
|
|
if subset_id then
|
|
break
|
|
end
|
|
end
|
|
|
|
-- multiple calls to balance must return nodes from the same subset
|
|
for i = 0, 10 do
|
|
assert.True(has_value(endpoint_strings, instance:balance()))
|
|
end
|
|
end)
|
|
end)
|
|
describe("new(backend)", function()
|
|
it("fills last subset correctly", function()
|
|
|
|
local backend = get_test_backend(7)
|
|
|
|
local instance = balancer_chashsubset:new(backend)
|
|
|
|
instance:sync(backend)
|
|
for id, endpoints in pairs(instance["subsets"]) do
|
|
assert.are.equal(#endpoints, 3)
|
|
end
|
|
end)
|
|
|
|
it("set alternative backends", function()
|
|
local backend = get_test_backend(7)
|
|
backend.trafficShapingPolicy = {
|
|
weight = 0,
|
|
header = "",
|
|
headerValue = "",
|
|
cookie = ""
|
|
}
|
|
backend.alternativeBackends = {
|
|
"my-dummy-canary-backend"
|
|
}
|
|
local instance = balancer_chashsubset:new(backend)
|
|
assert.not_equal(instance.traffic_shaping_policy, nil)
|
|
assert.not_equal(instance.alternative_backends, nil)
|
|
end)
|
|
end)
|
|
end)
|