Go语言中*testing.B用于基准测试,通过go test -bench=.运行。示例对比字符串拼接方式,结果显示strings.Builder比+=更高效,编写时需注意避免编译器优化、重置计时器排除初始化开销,并可使用b.RunParallel进行并发测试。

在Go语言中,testing.B 是用于编写基准测试的核心类型。通过 go test -bench=. 命令运行,它会自动循环执行被测代码,帮助你评估性能表现。下面是一个使用 *testing.B 进行循环基准测试的实用示例。
基本语法与结构
基准测试函数名必须以 Benchmark 开头,参数为 *testing.B。系统会自动调用 b.N 次目标代码,b.N 的值由测试运行器动态调整,以获得稳定的时间测量。
func BenchmarkExample(b *testing.B) {
for i := 0; i < b.N; i++ {
// 被测函数调用
}
}
对字符串拼接进行基准测试
以下对比两种常见字符串拼接方式:使用 += 和 strings.Builder。
package main
import (
"strings"
"testing"
)
func concatWithPlus(n int) string {
s := ""
for i := 0; i < n; i++ {
s += "a"
}
return s
}
func concatWithBuilder(n int) string {
var sb strings.Builder
for i := 0; i < n; i++ {
sb.WriteString("a")
}
return sb.String()
}
func BenchmarkConcatPlus(b *testing.B) {
for i := 0; i < b.N; i++ {
concatWithPlus(1000)
}
}
func BenchmarkConcatBuilder(b *testing.B) {
for i := 0; i < b.N; i++ {
concatWithBuilder(1000)
}
}
保存为 main_test.go 后,运行命令:
立即学习“go语言免费学习笔记(深入)”;
go test -bench=.
输出类似:
BenchmarkConcatPlus-8 10000 124124567 ns/op BenchmarkConcatBuilder-8 500000 3456 ns/op
可见 strings.Builder 明显更快。
优化建议与注意事项
编写高效可靠的基准测试需要注意以下几点:
-
避免编译器优化掉无用结果:如果返回值未使用,编译器可能优化掉整个调用。可通过
blackhole = fn()或runtime.KeepAlive防止。 -
重置计时器处理初始化开销:若准备阶段耗时较长,可用
b.ResetTimer()排除干扰。 -
控制并发测试:使用
b.RunParallel测试多协程场景下的性能。
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 := 0
for _, v := range data[:1000] {
sum += v
}
}
}
基本上就这些。写好基准测试能帮你发现性能瓶颈,做出更优实现选择。










