使用testing.B结合go关键字和sync.WaitGroup或chan,通过b.RunParallel或手动协程模拟并发场景;2. b.RunParallel自动分发迭代,适合无状态并行逻辑;3. 手动并发需用b.ResetTimer和wg.Wait确保计时准确。

在 Go 中做并发基准测试,核心是用 testing.B 结合 go 关键字和同步机制(如 sync.WaitGroup 或 chan),让 Benchmark 函数真实模拟多 goroutine 并发场景,同时避免因初始化、阻塞或计数错误导致结果失真。
用 b.RunParallel 实现轻量级并发压测
b.RunParallel 是标准库提供的专用并发测试方式,它会自动分发迭代到多个 goroutine 中执行,无需手动管理协程生命周期,适合测试无状态、可并行的逻辑(如加解密、JSON 编码、哈希计算)。
- 每个 goroutine 独立调用传入的函数,
b.N是总迭代次数,由所有 goroutine 共同完成 - 不支持自定义 goroutine 数量,但可通过
-cpu参数间接影响(实际效果有限),更推荐用-benchtime和-count控制稳定性 - 示例:测试字符串转大写的并发性能
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
strings.ToUpper("hello world")
}
})
}
手动启动 goroutine 时务必正确计数
若需控制并发数、共享状态或模拟真实业务流程(如并发请求、DB 查询),就得手动启 goroutine,此时必须用 b.ResetTimer() 排除 setup 开销,并用 sync.WaitGroup 确保所有任务完成后再结束计时。
- 切勿在 goroutine 内直接调用
b.N—— 它只在主 goroutine 中有效;应把总工作量均分,或用原子计数器协调 - 忘记
wg.Wait()会导致测试提前退出,结果不准;忘记b.ResetTimer()会让初始化时间计入耗时 - 示例:10 个 goroutine 并发执行 1000 次操作
const concurrency = 10
b.ResetTimer()
wg := sync.WaitGroup{}
for i := 0; i wg.Add(1)
go func() {
defer wg.Done()
for j := 0; j // 实际待测逻辑
time.Sleep(1 * time.Microsecond)
}
}()
}
wg.Wait()
}
避免常见陷阱:共享变量、锁竞争与 GC 干扰
并发基准测试容易放大隐藏问题。例如多个 goroutine 写同一变量未加锁,或高频分配触发 GC,都会让结果偏离真实吞吐能力。
立即学习“go语言免费学习笔记(深入)”;
- 读写共享资源必须加锁(
sync.Mutex)或用原子操作(atomic),否则数据竞争会导致 panic 或错误结果 - 避免在循环中创建大量临时对象;可复用对象池(
sync.Pool)或预分配切片减少 GC 压力 - 运行时加上
-gcflags="-m"观察逃逸分析,确认关键结构是否栈上分配
结合 pprof 定位瓶颈
单纯看 ns/op 不够,要配合 CPU / heap profile 找出热点。在 benchmark 中启用 pprof 很简单:
- 加
import _ "net/http/pprof",然后go tool pprof http://localhost:6060/debug/pprof/profile?seconds=30 - 或直接用
go test -cpuprofile cpu.out -bench=.生成文件后分析 - 重点关注高占比的函数调用栈,尤其是锁等待、channel 阻塞、内存分配点
基本上就这些。并发 benchmark 不是跑得越快越好,而是要测得准、稳、可复现 —— 控制变量、隔离干扰、验证逻辑,比堆 goroutine 数量重要得多。











