Go语言table-driven基准测试通过testing.B对多组输入数据独立计时,支持算法性能对比与边界case评估;需用b.Run()为每组参数创建子基准,命名含关键参数,并避免预处理污染计时。

Go 语言的 table-driven 基准测试(benchmarks)是验证函数在不同输入规模或数据特征下性能表现的高效方式。它不像单元测试那样只关注“对错”,而是聚焦“快慢”——尤其适合对比算法优化、评估边界 case、发现隐式性能退化。
用 testing.B 驱动多组数据循环
基准测试函数签名固定为 func BenchmarkXxx(*testing.B),其中 *testing.B 提供了计时、迭代控制和报告能力。要测多组数据,只需在 B.Run() 中为每组参数创建子基准:
- 每个子基准独立计时、独立运行(默认至少 1 秒,自动调整迭代次数)
- 子基准名建议含关键参数(如
"100"、"sorted"),便于识别 - 避免在
B.ResetTimer()前做耗时预处理(如生成大数据),否则会污染测量
构造清晰的测试表(table)
定义一个结构体切片,每项包含输入数据、预期行为(可选)、描述性标签。例如测字符串查找函数:
var benchCases = []struct {
name string
data string
want int
}{
{"short", "hello world", 6},
{"long", strings.Repeat("a", 10000) + "x", 10000},
{"empty", "", 0},
}
然后遍历运行:
立即学习“go语言免费学习笔记(深入)”;
func BenchmarkFindIndex(b *testing.B) {
for _, tc := range benchCases {
b.Run(tc.name, func(b *testing.B) {
b.ReportAllocs() // 可选:记录内存分配
for i := 0; i < b.N; i++ {
_ = findIndex(tc.data, 'x') // 实际被测函数
}
})
}
}
注意初始化与复位时机
若每次迭代需重建状态(如新建 map、重置 slice),应放在 b.ResetTimer() 之后;若只需一次初始化(如预编译正则),放在 b.ResetTimer() 之前:
-
b.ResetTimer()重置计时器和分配计数器,通常放在循环前 - 耗时初始化(如读文件、生成百万元素 slice)必须在
ResetTimer前完成,且只做一次 - 避免在循环内重复初始化——那测的是初始化开销,不是目标函数
运行与解读结果
执行 go test -bench=. -benchmem:
-
BenchmarkFindIndex/short-8 10000000 124 ns/op 0 B/op 0 allocs/op表示每操作平均 124 纳秒,无内存分配 - 子基准名用斜杠分隔(
/short),方便-bench=FindIndex/long单独运行某组 -
-benchmem显示每次操作的平均内存分配字节数和次数,对排查 GC 压力很关键










