testing.B用于go性能测试,通过go test -bench=.运行,可测量函数吞吐量、耗时及内存分配;编写时函数名以Benchmark开头,接收*testing.B参数,b.N由运行时自动调整以稳定测试时长;可用-benchtime设置最短运行时间,-count指定重复次数;调用b.ReportAllocs()输出内存分配情况;使用b.ResetTimer()等控制计时范围,排除初始化开销;通过b.Run()创建子基准测试对比不同场景;需防止编译器优化影响结果,如将结果赋值给全局变量;结合-benchmem和pprof可深入分析性能。

在go语言中,testing.B 是 testing 包提供的用于性能基准测试的核心类型。通过 go test -bench=. 命令可以运行这些测试,帮助开发者评估函数的执行效率,比如每秒能执行多少次操作(即吞吐量),以及每次操作消耗的时间和内存资源。
1. 编写基本的性能测试函数
性能测试函数与普通测试函数类似,但函数名以 Benchmark 开头,并接收 *testing.B 参数。
示例:测试字符串拼接性能
func BenchmarkStringConcat(b *testing.B) {
for i := 0; i < b.N; i++ {
var s string
for j := 0; j < 1000; j++ {
s += “x”
}
}
}
b.N 表示测试循环的次数,由Go运行时自动调整,目标是让测试持续约1秒以上,从而获得稳定的性能数据。
立即学习“go语言免费学习笔记(深入)”;
2. 控制测试时间和迭代次数
默认情况下,Go会动态调整 b.N 来获取可靠的测量结果。你也可以手动控制测试时长:
-
go test -bench=.—— 运行所有基准测试 -
go test -bench=BenchmarkStringConcat -benchtime=5s—— 每个测试至少运行5秒 -
go test -bench=BenchmarkStringConcat -count=3—— 重复测试3次,用于观察波动
3. 内存分配测试与性能分析
使用 b.ReportAllocs() 可输出每次操作的内存分配次数和字节数。
func BenchmarkWithAlloc(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
_ = make([]byte, 1024)
}
}
输出示例:
BenchmarkWithAlloc-8 1000000 1000 ns/op 1024 B/op 1 allocs/op
其中:
1024 B/op 表示每次操作分配了1024字节内存,1 allocs/op 表示每次操作发生1次内存分配。
4. 重置计时器与分离准备逻辑
如果测试前需要初始化大量数据,应避免将这部分时间计入性能统计。可使用 b.ResetTimer()、b.StartTimer() 和 b.StopTimer() 精确控制计时范围。
func BenchmarkWithSetup(b *testing.B) {
data := make([]int, 1e6)
for i := range data {
data[i] = i
}
b.ResetTimer() // 开始计时前完成初始化
for i := 0; i < b.N; i++ {
_ = sum(data)
}
}
也可以在循环中暂停和恢复计时,例如模拟I/O等待:
b.StopTimer()
// 模拟清理或准备操作
b.StartTimer()
5. 子基准测试(Sub-Benchmarks)
当需要对比不同输入规模或实现方式时,可使用 b.Run() 创建子测试。
func BenchmarkSmallLarge(b *testing.B) {
sizes := []int{10, 100, 1000}
for _, n := range sizes {
b.Run(fmt.Sprintf(“Size-%d”, n), func(b *testing.B) {
data := make([]int, n)
for i := 0; i < b.N; i++ {
_ = sum(data)
}
})
} }
运行后会分别输出每个子测试的结果,便于横向比较。
6. 避免编译器优化导致的误判
Go编译器可能优化掉“无副作用”的计算。为防止这种情况,应将结果赋值给 blackhole 变量或使用 runtime.ReadMemStats 强制保留。
推荐做法:
var result int
func BenchmarkSum(b *testing.B) {
data := []int{1, 2, 3, 4, 5}
var r int
for i := 0; i < b.N; i++ {
r = sum(data)
}
result = r // 防止被优化掉
}
基本上就这些。合理使用 *testing.B 能帮你发现性能瓶颈,验证优化效果,是Go项目中不可或缺的工具。记得结合 -benchmem 和 pprof 做深入分析。不复杂但容易忽略细节。


