使用Run方法可在基准测试中运行多个子基准测试，分别测量不同切片大小的拷贝性能，每个子测试独立输出结果。
在Go语言中，Benchmark 通常用于测量代码性能。从 Go 1.7 开始，Go 引入了 子测试（subtests） 和 子基准测试（sub-benchmarks） 的概念，使得我们可以更灵活地组织和运行测试与基准测试。本文将详细介绍如何在 Golang 中实现 Benchmark 子测试，并汇总常用方法。
使用 Run 方法执行子基准测试
通过 *testing.B 的 Run 方法，可以在一个基准测试函数中运行多个子基准测试。每个子测试可以独立执行并输出各自的性能数据。
示例：假设我们要比较不同大小的切片拷贝性能：
func BenchmarkCopySlice(b *testing.B) {
sizes := []int{10, 100, 1000}
for _, n := range sizes {
b.Run(fmt.Sprintf("Size_%d", n), func(b *testing.B) {
src := make([]int, n)
dst := make([]int, n)
for i := 0; i < b.N; i++ {
copy(dst, src)
}
})
}
}
运行结果会显示每个子测试的独立性能指标：
立即学习“go语言免费学习笔记（深入）”；
BenchmarkCopySlice/Size_10 BenchmarkCopySlice/Size_10-8 100000000 10.5 ns/op BenchmarkCopySlice/Size_100 BenchmarkCopySlice/Size_100-8 10000000 125 ns/op BenchmarkCopySlice/Size_1000 BenchmarkCopySlice/Size_1000-8 1000000 1250 ns/op
为不同算法实现做性能对比
子基准测试非常适合用于对比不同实现方式的性能差异，比如比较 map 与 switch 在查找操作中的表现。
示例：func BenchmarkLookup_Map(b *testing.B) {
m := map[string]int{"a": 1, "b": 2, "c": 3}
b.Run("MapLookup", func(b *testing.B) {
for i := 0; i < b.N; i++ {
_ = m["b"]
}
})
}
func BenchmarkLookup_Switch(b *testing.B) {
key := "b"
b.Run("SwitchLookup", func(b *testing.B) {
var val int
for i := 0; i < b.N; i++ {
switch key {
case "a":
val = 1
case "b":
val = 2
case "c":
val = 3
}
}
_ = val
})
}
这样可以在同一组测试中清晰看到两种方式的性能差异。
控制并发与重置计时器
在子基准测试中，有时需要手动控制性能统计范围，例如排除初始化开销或测试并发场景。
- b.ResetTimer()：重置计时器，常用于跳过初始化耗时。
- b.SetParallelism() 和 b.RunParallel()：用于并发基准测试。
func BenchmarkConcurrentMapAccess(b *testing.B) {
m := sync.Map{}
b.Run("Concurrent_Write_Read", func(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
i := 0
for pb.Next() {
key := fmt.Sprintf("key_%d", i%100)
m.Store(key, i)
_, _ = m.Load(key)
i++
}
})
})
}
该方式能有效模拟高并发读写场景，评估数据结构的并发性能。
避免常见陷阱
- 不要在循环外定义会被捕获的变量：闭包可能导致意外共享。
- 确保 b.N 被实际使用：空跑或未正确循环会影响结果准确性。
- 子测试名称应唯一且可读：建议使用有意义的命名规则，如 b.Run("InputType_Size", ...)
例如错误写法：
// 错误：i 被所有子测试共享
for i := range tests {
b.Run(fmt.Sprintf("Test%d", i), func(b *testing.B) {
// 使用了外部 i，值可能错乱
})
}
正确做法是传参或局部复制：
for i := range tests {
i := i // 创建局部副本
b.Run(fmt.Sprintf("Test%d", i), func(b *testing.B) {
// 使用局部 i
})
}
基本上就这些。通过合理使用 b.Run，你可以将基准测试模块化、参数化，并生成结构清晰的性能报告。子基准测试不仅提升可维护性，也便于定位性能瓶颈。不复杂但容易忽略的是命名规范与变量捕获问题，稍加注意即可写出高效可靠的 benchmark 代码。
