mirror of
https://github.com/wg/wrk
synced 2025-01-24 12:42:53 +08:00
58 lines
1.2 KiB
Lua
58 lines
1.2 KiB
Lua
local wrk = {
|
|
scheme = "http",
|
|
host = "localhost",
|
|
port = nil,
|
|
method = "GET",
|
|
path = "/",
|
|
headers = {},
|
|
body = nil
|
|
}
|
|
|
|
function wrk.format(method, path, headers, body)
|
|
local method = method or wrk.method
|
|
local path = path or wrk.path
|
|
local headers = headers or wrk.headers
|
|
local body = body or wrk.body
|
|
local s = {}
|
|
|
|
if not headers["Host"] then
|
|
headers["Host"] = wrk.headers["Host"]
|
|
end
|
|
|
|
headers["Content-Length"] = body and string.len(body)
|
|
|
|
s[1] = string.format("%s %s HTTP/1.1", method, path)
|
|
for name, value in pairs(headers) do
|
|
s[#s+1] = string.format("%s: %s", name, value)
|
|
end
|
|
|
|
s[#s+1] = ""
|
|
s[#s+1] = body or ""
|
|
|
|
return table.concat(s, "\r\n")
|
|
end
|
|
|
|
function wrk.init(args)
|
|
if not wrk.headers["Host"] then
|
|
local host = wrk.host
|
|
local port = wrk.port
|
|
|
|
host = host:find(":") and ("[" .. host .. "]") or host
|
|
host = port and (host .. ":" .. port) or host
|
|
|
|
wrk.headers["Host"] = host
|
|
end
|
|
req = wrk.format()
|
|
end
|
|
|
|
function wrk.request()
|
|
return req
|
|
end
|
|
|
|
init = wrk.init
|
|
request = wrk.request
|
|
response = nil
|
|
done = nil
|
|
|
|
return wrk
|