HTTP 压测工具 wrk2 安利指南

最近的工作一直在与服务端性能优化打交道，QPS（每秒查询率）的苛刻要求让我这个以前也就用 node.js 写写博客的人深刻地感觉到以前做的东西就是个玩具。所以最近也在尝试了解一些压测方面的知识。对于压测工具，业界常用的有 jmeter、loadrunner、tcpcopy、apache bench、wrk(2) 等。作为压测小白，结合项目实际情况（无需硬件监控、测试请求较简单），在这里选择了上手使用 wrk2。本文记录了使用过程中的一些心得体会。

wrk 简介

wrk2 基于 wrk 进化而来，我们可以先了解一下 wrk 的使用。wrk 类型上与 apache bench（以下简称 ab）类似，都是终端上的工具，其使用多线程设计来进行请求的生成。相比 ab，wrk 最值得称道的应该是它的自定义脚本功能：wrk 支持使用 lua 脚本来进行 HTTP 请求生成、响应处理以及自定义压测报告等。
在基本的配置项上 wrk 也非常简单：

$ wrk
wrk 4.1.0 [kqueue] Copyright (C) 2012 Will Glozer
Usage: wrk <options> <url>
  Options:
    # HTTP 连接数
    -c, --connections <N>  Connections to keep open
    # 测试持续时间，如 2s 2m 2h
    -d, --duration    <T>  Duration of test
    # 开启的线程数
    -t, --threads     <N>  Number of threads to use
    
    # 进阶功能，使用 lua 脚本
    -s, --script      <S>  Load Lua script file
    
    # 添加请求头，如 "User-Agent: wrk"
    -H, --header      <H>  Add header to request
    
        # 打印详细延迟统计
        --latency          Print latency statistics
        
        # 设置请求超时时间，大于该时间的请求将被记录
        --timeout     <T>  Socket/request timeout

对 url http://127.0.0.1:8080/index.html 进行开启 12 个线程，打开 400 个 HTTP 连接，持续 30s 的压测，可表示如下：

wrk -t12 -c400 -d30s http://127.0.0.1:8080/index.html

压测结束后产出的报告内容也还比较详细，都是一些压测方比较关心的数据，比如延迟分布、QPS 等，这里就不过多赘述。

自定义脚本功能

如果 wrk 常规的功能无法满足需求，那么这时就需要用户自行编写脚本去进行处理。wrk 官方提供了一些示例脚本以供参考。当然首先我们要掌握基本的 lua 语法，然后需要参考 wrk 暴露出的 lua 接口。wrk 在 setup、running、done 三个声明周期内暴露了许多方法，分别用于线程的配置、请求和响应的处理以及自定义展示最终生成的测试报告。结合脚本，wrk 能够完成相当程度的复杂压测需求。很多 wrk 的教程对此处并没有详细说明，但个人认为这里才是 wrk 的精华所在。如果想熟练上手使用，应当对自定义脚本的使用有所了解。

例如，如果想对每个请求增加一定延时，可在 wrk 暴露出的 delay 函数中进行设置：

function delay()
   return math.random(10, 50)
end

或者你想对 post 请求作压测，则可以对 table wrk 进行设置：

wrk.method = "POST"
wrk.body   = "foo=bar&baz=quux"
wrk.headers["Content-Type"] = "application/x-www-form-urlencoded"

wrk2

wrk2 是 wrk 的进化版，其号称能够提供稳定的吞吐量以及更精确的延时统计，反映到配置参数上就是 wrk2 增加了 --rate 参数用于设置吞吐量和--u_latency 参数用于显示不正确（统计学角度）的延时统计。其他使用上 wrk2 与 wrk 区别不大，算是更加完善的工具。当然 wrk2 的作者十分谦虚，字里行间流露出对 wrk 满满的崇敬之情，哈哈。

完。