Go 实现 API 自动化压测方法

本文深入探讨了使用 Go 语言进行 API 自动化压测的实战要点，强调压测绝非简单循环调用 http.Post 或 http.Get，而需前置严谨校验 API 的三大关键特征：Content-Type 是否严格要求 application/json、是否依赖需持久化的 session/token（警惕默认空 CookieJar 导致登录态丢失）、以及接口是否幂等（避免非幂等 POST 引发数据爆炸）；同时倡导摒弃黑盒第三方压测工具，转而基于 net/http + sync.WaitGroup 构建透明可控的原生压测脚本，精细管控超时（区分 DNS、连接、TLS、读响应阶段）、强制关闭 resp.Body 防止连接池失效、精准捕获真实错误根源（如状态码分布、分段耗时日志），并严格遵循 goroutine 生命周期管理规范以杜绝泄漏——这是一份聚焦可调试性、可观测性与生产级稳定性的 Go 压测实践指南。

压测前必须确认的三个 API 特征

Go 做 API 压测不是写个 http.Post 循环就完事——API 的实际行为会直接决定你压测结果是否可信。重点看这三点：

• Content-Type 是否为 application/json？若后端严格校验，发 text/plain 会被 400 拦截，但 Go 默认不设 header，容易漏
• 是否依赖 session 或 token？用 http.Client 复用时，CookieJar 默认为空，每次请求都是“新用户”，压不出真实登录态下的并发瓶颈
• 接口是否幂等？比如 POST /orders 每次都创建订单，压测一跑完数据库就爆了——得先确认能否走 GET /health 或带固定 id 的查询类接口

用 net/http + sync.WaitGroup 实现可控并发

别碰第三方压测库（如 vegeta）封装太深，调试时连超时是 DNS 还是 TCP 都分不清。原生组合最稳：

client := &http.Client{
    Timeout: 5 * time.Second,
}
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
    wg.Add(1)
    go func() {
        defer wg.Done()
        resp, err := client.Get("https://api.example.com/health")
        if err != nil {
            // 记录 err.Error()，别只打 "failed"
            return
        }
        resp.Body.Close() // 必须关，否则 fd 耗尽
    }()
}
wg.Wait()

• 并发数别硬写死循环次数——用 runtime.GOMAXPROCS(0) 参考值，但最终以压测目标 QPS 为准（比如要压 200 QPS，就每秒启 200 goroutine）
• http.Client.Timeout 是整个请求生命周期，含 DNS、连接、TLS、读响应；若需更细粒度控制，用 context.WithTimeout 包裹 req.Context()
• 漏掉 resp.Body.Close() 会导致连接不复用，http.Client 底层的连接池失效，很快触发 dial tcp: lookup xxx: no such host

如何捕获真实失败原因而非笼统的 “timeout”

Go 默认错误信息太模糊，context deadline exceeded 可能是 DNS 卡住、TCP 连不上、TLS 握手失败、或服务端真卡住。得拆开看：

• 加 DNS 日志：在 http.Client.Transport 里替换 DialContext，用 net.Resolver 手动解析并计时
• 区分连接超时和读超时：自定义 http.Transport，设 DialContext: (&net.Dialer{Timeout: 1 * time.Second}).DialContext 和 ResponseHeaderTimeout: 2 * time.Second
• 检查状态码分布：别只统计 “成功/失败”，用 map[int]int 记录 resp.StatusCode，突然大量 429 或 503 比单纯 timeout 更说明问题

避免 Goroutine 泄漏的两个硬约束

压测脚本跑完进程不退，十有八九是 goroutine 没收干净。盯死这两处：

• 所有 go func() { ... }() 必须配 defer wg.Done()，且 wg.Wait() 后不能有阻塞操作（比如没关闭的 channel receive）
• 用 http.Client 时禁用长连接：设 Transport.MaxIdleConnsPerHost = 0，否则压测结束时 idle 连接还挂着，pprof/goroutine 里能看到一堆 transport.dialConn

复杂点在于：真实压测要持续几分钟甚至小时，中间可能需要动态调速、采样日志、熔断失败率——这些逻辑一旦掺进 goroutine，泄漏概率指数上升。建议把压测主循环抽成独立函数，用 select + time.After 控制生命周期，别靠 sleep 硬等。

以上就是《Go 实现 API 自动化压测方法》的详细内容，更多关于的资料请关注golang学习网公众号！