Go的sub-benchmark通过b.Run()将基准测试拆分为多个子测试,便于对比不同参数或实现的性能,提升代码清晰度与可维护性。
在Golang中,使用sub-benchmark可以将一个基准测试拆分为多个子基准,便于对不同参数或实现方式的性能进行对比。这种方式让代码更清晰、结构更模块化,也方便维护和扩展。
理解sub-benchmark的基本结构
Go的
testing
包支持通过
b.Run()
方法创建子基准测试。每个子基准独立运行,并在结果中单独显示性能数据。
基准函数接收一个
*testing.B
类型的指针,调用
Run
方法时传入子测试名和对应的函数。
示例:立即学习“go语言免费学习笔记(深入)”;
动态生成sub-benchmark
当需要测试多种输入规模时,可以用循环动态创建子基准,避免重复代码。
子基准名称通常包含变量值,以便区分输出结果。
示例:测试不同字符串长度下的性能go语言参考手册 中文CHM版Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译下载
注意:
b.ResetTimer()
用于排除准备数据的时间,确保只测量核心逻辑。
组织多个相关基准
你可以把一组相关的
性能测试
归在一个顶级benchmark下,比如比较JSON、Gob、XML等序列化方式的性能。
这样运行
go test -bench=.
时,会清晰展示每种子测试的结果。
示例:序列化性能对比
运行后输出类似:
运行特定子基准
使用
-bench
标志可运行特定子测试。例如:
go test -bench=Concat$
—— 只运行顶层
BenchmarkConcat
go test -bench=Concat/StringBuilder
—— 只运行指定子项
正则匹配需注意斜杠路径格式。调试时可结合
-run
跳过单元测试,如
-run=^$
。
基本上就这些。用好sub-benchmark能让性能测试更有条理,结果更易读。
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)
}
})
}
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)
}
})
}
}
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
