Go的sub-benchmark通过b.Run()将基准测试拆分为多个子测试，便于对比不同参数或实现的性能，提升代码清晰度与可维护性。
在Golang中，使用sub-benchmark可以将一个基准测试拆分为多个子基准，便于对不同参数或实现方式的性能进行对比。这种方式让代码更清晰、结构更模块化，也方便维护和扩展。
理解sub-benchmark的基本结构
Go的testing包支持通过b.Run()方法创建子基准测试。每个子基准独立运行，并在结果中单独显示性能数据。
基准函数接收一个*testing.B类型的指针，调用Run方法时传入子测试名和对应的函数。
示例：
func BenchmarkConcat(b *testing.B) {
b.Run("StringBuilder", func(b *testing.B) {
for i := 0; i < b.N; i++ {
var sb strings.Builder
sb.WriteString("hello")
sb.WriteString("world")
_ = sb.String()
}
})
b.Run("ByteSlice", func(b *testing.B) {
for i := 0; i < b.N; i++ {
bs := []byte{}
bs = append(bs, "hello"...)
bs = append(bs, "world"...)
_ = string(bs)
}
})
}
})
}
动态生成sub-benchmark
当需要测试多种输入规模时，可以用循环动态创建子基准，避免重复代码。
子基准名称通常包含变量值，以便区分输出结果。
示例：测试不同字符串长度下的性能
func BenchmarkStringRepeat(b *testing.B) {
for _, size := range []int{10, 100, 1000} {
b.Run(fmt.Sprintf("Size%d", size), func(b *testing.B) {
s := strings.Repeat("a", size)
b.ResetTimer()
for i := 0; i < b.N; i++ {
_ = strings.ToUpper(s)
}
})
}
}
注意：b.ResetTimer()用于排除准备数据的时间，确保只测量核心逻辑。
组织多个相关基准
你可以把一组相关的性能测试归在一个顶级benchmark下，比如比较JSON、Gob、XML等序列化方式的性能。
这样运行go test -bench=.时，会清晰展示每种子测试的结果。
示例：序列化性能对比
func BenchmarkSerialization(b *testing.B) {
data := struct{ Name string }{Name: "gopher"}
b.Run("JSON", func(b *testing.B) {
for i := 0; i < b.N; i++ {
bytes, _ := json.Marshal(data)
_ = bytes
}
})
b.Run("Gob", func(b *testing.B) {
for i := 0; i < b.N; i++ {
buf := new(bytes.Buffer)
enc := gob.NewEncoder(buf)
enc.Encode(data)
}
})
}
运行后输出类似：
BenchmarkSerialization/JSON-8 5000000 210 ns/op BenchmarkSerialization/Gob-8 3000000 400 ns/op
运行特定子基准
使用-bench标志可运行特定子测试。例如：
- go test -bench=Concat$ —— 只运行顶层BenchmarkConcat
- go test -bench=Concat/StringBuilder —— 只运行指定子项
正则匹配需注意斜杠路径格式。调试时可结合-run跳过单元测试，如-run=^$。
基本上就这些。用好sub-benchmark能让性能测试更有条理，结果更易读。
