合法基准测试函数需命名为benchmarkxxx,签名固定为func benchmarkxxx(b *testing.b),被测逻辑置于b.resettimer()之后,用b.reportallocs()统计内存分配,避免循环内非必要操作。
Go 的 testing 包自带基准测试能力,无需第三方库,但必须用 go test -bench 运行,且函数名必须是 BenchmarkXxx 形式,否则会被忽略。
如何写一个合法的基准测试函数
基准测试函数签名固定为 func BenchmarkXxx(b *testing.B),b.N 是框架自动调整的循环次数,你不能手动设初值或 break —— 必须让 b.ResetTimer() 和 b.ReportAllocs() 配合实际被测逻辑。
- 被测代码必须放在
b.ResetTimer()之后,否则初始化开销会混入结果 - 若函数有 setup 开销(如构建 map、解析 JSON),应放在
b.ResetTimer()前 - 调用
b.ReportAllocs()才能显示内存分配统计(B/op和allocs/op) - 避免在循环体内做非被测操作(比如拼接字符串再丢弃),这会污染性能数据
func BenchmarkFindInSlice(b *testing.B) {
data := make([]int, 1000)
for i := range data {
data[i] = i
}
b.ResetTimer() // 计时从此开始
b.ReportAllocs()
for i := 0; i < b.N; i++ {
_ = findInSlice(data, 500) // 只测这个函数调用
}
}
为什么 go test -bench=. 跑不出结果
常见原因不是代码写错,而是文件命名或运行方式不对:基准测试必须写在 _test.go 文件里,且该文件不能带 //go:build 约束(除非显式启用),同时要确保当前目录下有对应包的源码(不能只留 test 文件)。
- 文件名不以
_test.go结尾 → 不被识别为测试文件 - 用了
//go:build ignore或//go:build !test→ 编译时被跳过 - 执行命令时不在模块根目录,或
go.mod缺失 →go test找不到包 - 函数名是
TestXxx或benchXxx(大小写/前缀错误)→ 不匹配Benchmark前缀规则
怎么对比两个函数的性能差异
用 -benchmem + -count=N 多次运行取中位数更可靠;想横向比对多个实现,直接在同一个 _test.go 里写多个 BenchmarkXxx 函数,go test -bench=^Benchmark(Linear|Binary)Search$ -benchmem 就能只跑指定几个。