别再只会wrk -t -c -d了！用Lua脚本玩转复杂API压力测试（附实战脚本）

别再只会wrk -t -c -d了！用Lua脚本玩转复杂API压力测试（附实战脚本）

当我们需要评估一个API服务的性能极限时，wrk无疑是开发者工具箱中的一把利器。但大多数人对它的认知还停留在简单的并发测试层面，仅仅通过-t、-c、d这几个基础参数来发起请求。这就像用一台超级计算机只做加减法一样暴殄天物。wrk真正的威力在于其Lua脚本扩展能力，它能模拟真实业务场景中的复杂用户行为，让压力测试结果更具参考价值。

想象一下电商大促时的场景：用户需要先登录获取token，然后浏览商品列表，接着将心仪商品加入购物车，最后完成支付。这种有状态的连续操作，用简单的静态URL测试根本无法反映真实负载。而通过Lua脚本，我们可以完整模拟这个流程，让压力测试真正"活"起来。

1. 为什么需要Lua脚本进行压力测试

传统的压力测试工具往往只能模拟简单的、无状态的请求，这在现代分布式系统和微服务架构面前显得力不从心。一个典型的电商系统可能包含数十个微服务，用户的一次完整操作会涉及多个服务的协同工作。如果仅仅测试单个接口的性能，就像只测试汽车发动机而忽视整车性能一样片面。

Lua脚本在wrk中的应用解决了几个关键问题：

• 状态保持：可以在多个请求间共享变量，模拟真实用户的会话状态
• 动态参数：能够基于前序请求的响应动态构造后续请求的参数
• 复杂逻辑：支持条件判断、循环等编程结构，实现真实的用户行为序列
• 灵活断言：可以对响应内容进行校验，确保测试的有效性

-- 示例：简单的状态保持 token = "" function response(status, headers, body) if status == 200 then token = string.match(body, '"token":"(%w+)"') end end

2. Lua脚本核心功能解析

wrk提供的Lua脚本接口非常丰富，理解这些接口是编写高效测试脚本的基础。脚本主要包含以下几个关键部分：

2.1 请求生成阶段

在这个阶段，我们可以动态构造请求方法、URL、头部和体。wrk提供了几个特殊的全局变量：

wrk.method = "POST" -- 设置请求方法 wrk.headers["Content-Type"] = "application/json" -- 设置请求头 wrk.body = '{"username":"test","password":"123456"}' -- 设置请求体

更高级的用法是通过request函数动态生成请求：

request = function() local uuid = math.random(10000, 99999) local path = "/api/items/" .. uuid return wrk.format("GET", path) end

2.2 响应处理阶段

响应处理允许我们对服务器返回的内容进行解析和校验，常用的回调函数包括：

• response(status, headers, body)：每次收到响应时调用
• done(summary, latency, requests)：测试结束时调用

function response(status, headers, body) if status ~= 200 then print("请求失败，状态码："..status) end -- 解析JSON响应 local json = require("json") local data = json.decode(body) if data.error then print("业务错误："..data.error) end end

2.3 延迟控制

真实的用户请求不会完全同时发出，适当的延迟模拟更符合实际情况。我们可以通过delay函数实现：

function delay() return math.random(100, 500) -- 返回100-500毫秒的随机延迟 end

3. 实战：电商场景压力测试脚本

让我们通过一个完整的电商场景示例，展示如何用Lua脚本模拟真实用户行为。这个场景包含以下步骤：

1. 用户登录获取token
2. 浏览商品列表
3. 查看商品详情
4. 加入购物车
5. 创建订单
6. 支付

3.1 初始化阶段

首先设置一些全局变量和初始化逻辑：

-- 全局变量 token = "" cartId = "" orderId = "" userIds = {"user1", "user2", "user3", "user4", "user5"} currentUser = "" -- 初始化函数，每个线程开始时调用一次 function init(args) currentUser = userIds[math.random(#userIds)] end

3.2 登录接口

模拟用户登录并保存token：

request = function() if token == "" then -- 首次请求为登录 wrk.method = "POST" wrk.headers["Content-Type"] = "application/json" local body = string.format('{"username":"%s","password":"123456"}', currentUser) return wrk.format(nil, "/api/login", nil, body) else -- 后续请求随机选择业务接口 local r = math.random(1, 100) if r <= 30 then return browseItems() elseif r <= 60 then return viewItemDetail() elseif r <= 80 then return addToCart() elseif r <= 95 then return createOrder() else return makePayment() end end end function response(status, headers, body) if token == "" and status == 200 then -- 登录成功，保存token token = string.match(body, '"token":"(%w+)"') wrk.headers["Authorization"] = "Bearer " .. token end end

3.3 商品浏览接口

function browseItems() local page = math.random(1, 10) local size = math.random(5, 20) local path = string.format("/api/items?page=%d&size=%d", page, size) return wrk.format("GET", path) end

3.4 加入购物车接口

function addToCart() local itemId = math.random(1000, 9999) local quantity = math.random(1, 3) local body = string.format('{"itemId":"%d","quantity":%d}', itemId, quantity) return wrk.format("POST", "/api/cart", nil, body) end function response(status, headers, body) -- ...之前的响应处理... if string.find(wrk.path, "/api/cart") and status == 200 then cartId = string.match(body, '"cartId":"(%w+)"') end end

4. 高级技巧与性能优化

编写高效的Lua脚本不仅关乎功能实现，还需要考虑性能影响。以下是一些实用技巧：

4.1 减少Lua脚本中的计算量

压力测试时，脚本本身的执行时间也会影响测试结果。应该尽量减少脚本中的复杂计算：

-- 不推荐：每次请求都生成随机用户 function request() local user = "user"..math.random(1,100) -- ... end -- 推荐：初始化时生成用户列表，请求时随机选择 users = {} function init(args) for i=1,100 do users[i] = "user"..i end end function request() local user = users[math.random(1,100)] -- ... end

4.2 合理使用延迟

适当的延迟可以更真实地模拟用户行为，但过多的延迟会降低测试强度：

-- 根据测试目标调整延迟策略 function delay() -- 压力测试：返回较小值 -- return math.random(10, 50) -- 真实场景模拟：返回较大值 return math.random(100, 1000) end

4.3 错误处理与重试机制

完善的错误处理可以避免因个别失败请求影响整体测试：

retryCount = 0 maxRetry = 3 function response(status, headers, body) if status >= 500 and retryCount < maxRetry then retryCount = retryCount + 1 wrk.thread:stop() wrk.thread:start() else retryCount = 0 end end

5. 测试结果分析与解读

使用Lua脚本进行压力测试后，我们需要从几个维度分析结果：

5.1 关键指标对比

5.2 常见问题定位

• Token失效问题：检查token刷新逻辑是否正确
• 商品库存不足：压力测试前预置足够测试数据
• 订单重复创建：确保幂等性处理逻辑完善

-- 示例：检测库存不足错误 function response(status, headers, body) if string.find(body, "INSUFFICIENT_STOCK") then print("WARNING: 库存不足") -- 可以在这里调整测试逻辑 end end

6. 实战脚本合集

最后，我们提供几个常用场景的完整脚本，可以直接用于实际测试：

6.1 OAuth2授权流程测试

-- 配置参数 client_id = "test_client" client_secret = "test_secret" auth_code = "" access_token = "" refresh_token = "" -- 请求逻辑 request = function() if access_token == "" then if auth_code == "" then -- 第一步：获取授权码 local path = "/oauth/authorize?response_type=code&client_id="..client_id return wrk.format("GET", path) else -- 第二步：获取access token local body = "grant_type=authorization_code&code="..auth_code.. "&client_id="..client_id.."&client_secret="..client_secret wrk.headers["Content-Type"] = "application/x-www-form-urlencoded" return wrk.format("POST", "/oauth/token", nil, body) end else -- 使用access token访问受保护资源 wrk.headers["Authorization"] = "Bearer "..access_token return wrk.format("GET", "/api/protected") end end -- 响应处理 function response(status, headers, body) if access_token == "" then if auth_code == "" then auth_code = string.match(body, "code=(%w+)") else access_token = string.match(body, '"access_token":"(%w+)"') refresh_token = string.match(body, '"refresh_token":"(%w+)"') end end end

6.2 WebSocket压力测试

虽然wrk本身不支持WebSocket，但可以通过HTTP升级模拟：

-- WebSocket升级请求 ws_key = "" request = function() if ws_key == "" then -- 发送WebSocket升级请求 ws_key = math.random(1000000000, 9999999999) wrk.headers["Connection"] = "Upgrade" wrk.headers["Upgrade"] = "websocket" wrk.headers["Sec-WebSocket-Key"] = ws_key wrk.headers["Sec-WebSocket-Version"] = "13" return wrk.format("GET", "/ws") else -- 已建立连接，可以发送数据帧(需要base64编码) local payload = "data_"..math.random(1000) wrk.headers["Content-Type"] = "application/octet-stream" return wrk.format("POST", "/ws/send", nil, payload) end end