mirror of
https://github.com/wg/wrk
synced 2025-01-08 06:52:55 +08:00
23 lines
504 B
Lua
23 lines
504 B
Lua
-- example script that demonstrates use of setup() to pass
|
|
-- a random server address to each thread
|
|
|
|
local addrs = nil
|
|
|
|
function setup(thread)
|
|
if not addrs then
|
|
addrs = wrk.lookup(wrk.host, wrk.port or "http")
|
|
for i = #addrs, 1, -1 do
|
|
if not wrk.connect(addrs[i]) then
|
|
table.remove(addrs, i)
|
|
end
|
|
end
|
|
end
|
|
|
|
thread.addr = addrs[math.random(#addrs)]
|
|
end
|
|
|
|
function init(args)
|
|
local msg = "thread addr: %s"
|
|
print(msg:format(wrk.thread.addr))
|
|
end
|