当前位置：首页 > article >正文

wrk如何测试post请求

article 2025/1/31 14:05:16

wrk git地址

https://github.com/wg/wrk

wrk 默认是针对 GET 请求的，但它也可以通过添加自定义的 HTTP 请求体和 头部信息来进行 POST 请求的压测。以下是详细的步骤：

wrk -t4 -c100 -d30s -s post.lua http://example.com

-t4：使用 4 个线程。
-c100：模拟 100 个并发连接。
-d30s：测试持续 30 秒。
-s post.lua：加载一个自定义的 Lua 脚本 post.lua，用于指定 POST 请求的内容。

编写 `post.lua` 脚本

创建一个 post.lua 文件并写入以下内容：

wrk.method = "POST"
wrk.body   = '{"key1":"value1", "key2":"value2"}'
wrk.headers["Content-Type"] = "application/json"

参数说明：

wrk.method：指定请求方法为 POST。
wrk.body：定义请求的 JSON 数据。
wrk.headers["Content-Type"]：指定 HTTP 头部的 Content-Type 为 application/json。

完整示例

假设你的 API 是 http://example.com/api，需要发送以下 POST 请求：

URL: http://example.com/api
Body: {"username": "testuser", "password": "testpass"}
Headers: Content-Type: application/json

脚本内容：

wrk.method = "POST"
wrk.body   = '{"username": "testuser", "password": "testpass"}'
wrk.headers["Content-Type"] = "application/json"

执行命令：

wrk -t2 -c50 -d10s -s post.lua http://example.com/api

-t2：使用 2 个线程。
-c50：模拟 50 个并发连接。
-d10s：测试持续 10 秒。

动态参数的高级脚本

如果需要动态生成请求体（例如每次发送不同的参数），可以在 Lua 脚本中自定义逻辑：

counter = 0

request = function()
   counter = counter + 1
   local body = string.format('{"username": "user%d", "password": "pass%d"}', counter, counter)
   wrk.body = body
   wrk.headers["Content-Type"] = "application/json"
   return wrk.format("POST")
end

在这个脚本中，每次请求都会生成不同的 username 和 password。