子基准测试是Go中通过*testing.B的Run方法实现的机制,可在单个基准函数内运行多个独立测试用例,每个子测试独立计时并输出结果,适用于对比不同数据规模、算法或优化效果。例如可测试字符串拼接在不同输入长度下的性能,或比较map遍历方式:通过b.Run定义多个子测试,合理命名以反映场景,如"Small"、"Medium",并在内部构造数据、调用b.ResetTimer()确保准确计时,最终生成清晰的分级结果输出,便于性能分析与优化验证。

在Go语言中,使用子基准测试(sub-benchmark)可以更细致地对函数的不同场景或输入规模进行性能测量。通过
Run</samp>方法,你可以将一个基准测试拆分为多个子任务,便于横向比较不同实现或参数下的表现。</p><H3>什么是sub-benchmark?</H3><p>子基准测试是<pre class="brush:php;toolbar:false;">*testing.B提供的
Run方法所支持的一种机制。它允许你在单个基准函数内运行多个独立的基准测试用例,每个子测试都有独立的计时和结果输出。
这特别适合以下场景:
- 测试同一函数处理不同数据规模的性能
- 对比多种算法在同一问题上的表现
- 验证优化版本是否真的提升了性能
如何编写sub-benchmark
下面是一个使用sub-benchmark的例子,测试字符串拼接在不同方式下的性能:
立即学习“go语言免费学习笔记(深入)”;
func BenchmarkStringConcat(b *testing.B) {
inputs := []struct {
name string
size int
}{
{"Small", 10},
{"Medium", 100},
{"Large", 1000},
}
<pre class='brush:php;toolbar:false;'>for _, tc := range inputs {
b.Run(tc.name, func(b *testing.B) {
data := make([]string, tc.size)
for i := range data {
data[i] = "x"
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
var result string
for _, s := range data {
result += s
}
}
})
}}
执行命令:
go test -bench=.
输出示例:
BenchmarkStringConcat/Small-8 10000000 150 ns/op BenchmarkStringConcat/Medium-8 1000000 1500 ns/op BenchmarkStringConcat/Large-8 100000 15000 ns/op
可以看到每个子测试都有独立的结果行,清晰展示随输入增长的性能变化。
使用技巧与注意事项
写好 sub-benchmark 需要注意几个关键点:
- 合理命名子测试:名字应反映测试条件,如"JSON_Unmarshal_SmallPayload"
-
避免在子测试外做耗时初始化:大对象应在
b.Run
内部构造,或使用Setup
逻辑分离准备阶段 -
适时调用ResetTimer:如果预处理较重,记得在循环前调用
b.ResetTimer()
-
控制并发与P值影响:可结合
b.SetParallelism
测试并发场景
也可以嵌套多层
Run来组织复杂测试结构,比如先分算法类型,再分数据规模。
实用案例:比较map遍历方式
用sub-benchmark对比两种遍历map的方式:
func BenchmarkMapRange(b *testing.B) {
m := make(map[int]int)
for i := 0; i < 1000; i++ {
m[i] = i * 2
}
<pre class='brush:php;toolbar:false;'>b.Run("Range", func(b *testing.B) {
for i := 0; i < b.N; i++ {
sum := 0
for _, v := range m {
sum += v
}
}
})
b.Run("KeysSlice", func(b *testing.B) {
keys := make([]int, 0, len(m))
for k := range m {
keys = append(keys, k)
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
sum := 0
for _, k := range keys {
sum += m[k]
}
}
})}
这种对比能帮助你判断哪种方式在特定场景下更高效。
基本上就这些。sub-benchmark让基准测试更有条理,也更容易发现性能拐点。只要结构清晰,就能为性能调优提供有力支持。


















