在Golang的测试体系中,除了常用的单元测试功能,标准库testing包还提供了专门用于性能基准测试的工具testing.B,它可以帮助开发者量化代码的执行效率,为性能优化提供可靠的数据支撑。

testing.B基础概念
testing.B是基准测试函数的接收器类型,所有基准测试函数的签名都必须符合func BenchmarkXxx(b *testing.B)的格式,其中Xxx为自定义的测试名称,首字母需要大写。基准测试的核心逻辑是在函数中执行待测代码,testing.B会自动调整循环次数,确保测试结果具有统计意义。
编写基础基准测试
下面以一个简单的字符串拼接函数为例,演示如何编写基准测试:
package main
import (
"strings"
"testing"
)
// 待测函数:拼接多个字符串
func concatStrings(strs []string) string {
var result strings.Builder
for _, s := range strs {
result.WriteString(s)
}
return result.String()
}
// 基准测试函数
func BenchmarkConcatStrings(b *testing.B) {
// 准备测试数据
testStrs := []string{"hello", " ", "world", "!", "golang"}
// 重置定时器,排除准备数据的耗时
b.ResetTimer()
// 循环执行待测代码,b.N由testing.B自动调整
for i := 0; i < b.N; i++ {
concatStrings(testStrs)
}
}运行该基准测试的命令为go test -bench=.,执行后会输出类似如下的结果:
BenchmarkConcatStrings-8 1000000 1056 ns/op 512 B/op 8 allocs/op
结果中各部分含义如下:
- BenchmarkConcatStrings-8:测试名称,8表示GOMAXPROCS的值
- 1000000:测试循环执行的次数b.N
- 1056 ns/op:每次操作的平均耗时,单位为纳秒
- 512 B/op:每次操作平均分配的内存字节数
- 8 allocs/op:每次操作的平均内存分配次数
testing.B常用方法
控制测试流程
testing.B提供了多个方法帮助开发者更灵活地控制基准测试流程:
| 方法名 | 作用说明 |
|---|---|
| ResetTimer() | 重置计时器,用于排除测试数据准备阶段的耗时,只计算待测代码的执行时间 |
| StopTimer() | 暂停计时器,可在待测代码中插入非测试逻辑的耗时操作时使用 |
| StartTimer() | 恢复计时器,配合StopTimer()使用 |
| RunParallel() | 开启并行基准测试,模拟多协程并发场景下的性能表现 |
并行基准测试示例
当需要测试并发场景下的代码性能时,可以使用RunParallel方法:
func BenchmarkConcatStringsParallel(b *testing.B) {
testStrs := []string{"hello", " ", "world"}
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
concatStrings(testStrs)
}
})
}运行并行基准测试时会自动利用多个CPU核心,模拟真实的并发执行场景。
基准测试常用参数
在执行基准测试时,可以通过命令行参数调整测试行为:
-bench=.:运行所有基准测试,也可以指定正则表达式匹配特定测试,比如-bench=Concat只运行名称包含Concat的基准测试-benchtime=5s:设置每个基准测试的最小运行时间,默认是1秒,可适当延长以获得更稳定的结果-benchmem:输出内存分配相关的指标,默认已经开启,如需关闭可添加-benchmem=false-count=3:每个基准测试重复执行的次数,用于减少偶然误差,取多次结果的平均值
基准测试注意事项
编写基准测试时需要注意以下几点,避免结果出现偏差:
- 不要在循环内执行非待测逻辑,如果必须执行,要使用StopTimer和StartTimer控制计时范围
- 待测代码的结果如果没有被使用,编译器可能会优化掉整个代码逻辑,导致测试结果不准确,可以通过
_ = 结果变量的方式保留结果 - 基准测试的结果受运行环境的影响较大,不同机器、不同负载下的结果没有可比性,仅适合在同一环境下对比优化前后的性能差异
- 不要过度依赖单次基准测试的结果,建议多次运行并取稳定值作为参考
完整测试示例
下面是一个包含内存指标输出的完整基准测试示例:
package main
import (
"strings"
"testing"
)
func concatStrings(strs []string) string {
var result strings.Builder
for _, s := range strs {
result.WriteString(s)
}
return result.String()
}
func BenchmarkConcatStringsWithMem(b *testing.B) {
testStrs := []string{"a", "b", "c", "d", "e"}
b.ResetTimer()
for i := 0; i < b.N; i++ {
res := concatStrings(testStrs)
// 避免编译器优化掉待测代码
_ = res
}
}执行命令go test -bench=BenchmarkConcatStringsWithMem -benchmem即可得到包含内存分配数据的基准测试结果。