测试中启用 pprof 需手动启动 HTTP 服务并注册路由,或使用 -test.cpuprofile 等 flag 生成离线文件;Benchmark 中应手动调用 pprof.StartCPUProfile 确保覆盖压测逻辑。
测试中启用 pprof 服务需要单独启动 HTTP 服务器
Go 的 pprof 默认绑定在运行时的 HTTP 服务上,但 go test 不会自动开启该服务。想在测试期间采集 CPU、内存等数据,必须手动启动一个 net/http 服务并注册 pprof 路由。
常见错误是直接跑 go test -cpuprofile=cpu.out 就以为能访问 /debug/pprof/ —— 实际上这些 flag 只触发采样写入文件,不开启 Web 接口。
- 测试中要交互式查看实时 profile,得在测试代码里加:
go func() { log.Println(http.ListenAndServe("localhost:6060", nil)) }() - 记得在
init()或TestMain中注册路由(1.16+ 默认已注册,旧版本需显式调用pprof.Register()) -
端口被占用时会 panic,建议用
net.Listen("tcp", ":0")动态选端并打印出来
用 -test.cpuprofile 等 flag 生成离线 profile 文件
如果只是想跑完测试就拿到一份 profile 数据用于后续分析,-test.cpuprofile、-test.memprofile、-test.blockprofile 是最轻量的方式。它们不依赖 HTTP,也不要求测试本身做任何修改。
注意这些 flag 属于测试二进制的参数,不是 go test 命令本身的 flag,所以必须写在 -- 后面或紧贴包路径之后:
- 正确:
go test -run TestFoo ./pkg -- -test.cpuprofile=cpu.prof - 错误:
go test -test.cpuprofile=cpu.prof -run TestFoo ./pkg(flag 被 go test 解析,无效) -
-test.memprofile只在测试结束时拍一次堆快照;如需追踪分配热点,应配合-test.memprofilerate=1(设为 1 表示每次分配都记录,仅调试用)
在 Benchmark 中使用 pprof 需主动控制采样周期
Benchmark 函数默认不触发 pprof 的 runtime hook,单纯加 -test.cpuprofile 可能采不到有效数据,尤其当函数执行太快、被内联或优化掉时。
可靠做法是在 BenchmarkXxx 内部手动启停 profiling: