Go testing包原生支持基准测试,需用go test -bench运行且函数名以Benchmark开头、接收*testing.B参数;通过b.N循环执行待测逻辑,用b.ResetTimer()或b.StopTimer()排除初始化开销,b.ReportAllocs()统计内存分配。
Go 的 testing 包原生支持基准测试(benchmark),能准确统计函数执行耗时、吞吐量,并自动进行多次运行取平均,避免单次测量的偶然误差。关键在于用 go test -bench 运行,且函数需符合 BenchmarkXxx(*testing.B) 签名。
基准函数必须接收 *testing.B 参数,且函数名以 Benchmark 开头。核心是调用 b.N 控制循环次数,把待测逻辑放在循环内:
b.ResetTimer() 之前或 b.StopTimer() 区间内b.ReportAllocs() 可开启内存分配统计(如 512 B/op)示例:
func BenchmarkAdd(b *testing.B) {
a, bVal := 1, 2
b.ReportAllocs()
for i := 0; i < b.N; i++ {
_ = a + bVal // 待测逻辑
}
}
在包目录下执行:
go test -bench=^BenchmarkAdd$ -benchmem -count=3
-bench=^BenchmarkAdd$:正则匹配指定函数(^ 和 $ 保证精确匹配)-benchmem:显示内存分配统计-count=3:重复运行 3 次取中位数,减少噪声影响输出类似:BenchmarkAdd-8 1000000000 0.34 ns/op 0 B/op 0 allocs/op
含义:在 8 核 CPU 上,执行 10⁹ 次,平均每次耗时 0.34 纳秒,无内存分配。
默认情况下,Go 会自动调整 b.N 使单轮测试时长接近 1 秒。若函数极快(如纳秒级),可手动设最小时间避免过短测量:
b.SetMinimumTime(500*time.Millisecond):确保至少跑够 500ms 再停止b.RunParallel(func(pb *testing.PB) { ...
}):并发运行,适合测并发场景(如 channel 操作、锁竞争)用 b.Run() 组织子测试,便于横向比较:
func BenchmarkStringConcat(b *testing.B) {
s1, s2 := "hello", "world"
b.Run("plus", func(b *testing.B) {
for i := 0; i < b.N; i++ {
_ = s1 + s2
}
})
b.Run("fmtSprint", func(b *testing.B) {
for i := 0; i < b.N; i++ {
_ = fmt.Sprint(s1, s2)
}
})
}
运行后会分别输出 BenchmarkStringConcat/plus 和 BenchmarkStringConcat/fmtSprint 的耗时,直观看出哪种拼接更快。