跳转到主内容
极星编程网:以代码为星,赴技术山海!

如何在Golang中使用sub-benchmark组织测试

Go的sub-benchmark通过b.Run()将基准测试拆分为多个子测试,便于对比不同参数或实现的性能,提升代码清晰度与可维护性。 在Golang中,使用sub-benchmark可以将一个基准测试拆分为多个子基准,便于对不同参数或实现方式的性能进行对比。这种方式让代码更清晰、结构更模块化,也方便维护和扩展。 理解sub-benchmark的基本结构 Go的 testing 包支持通过 b.Run() 方法创建子基准测试。每个子基准独立运行,并在结果中单独显示性能数据。 基准函数接收一个 *testing.B 类型的指针,调用 Run 方法时传入子测试名和对应的函数。 示例:立即学习“go语言免费学习笔记(深入)”;
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) } }) }
动态生成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数组安全、编译下载
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) } }) } }
注意: b.ResetTimer() 用于排除准备数据的时间,确保只测量核心逻辑。 组织多个相关基准 你可以把一组相关的 性能测试 归在一个顶级benchmark下,比如比较JSON、Gob、XML等序列化方式的性能。 这样运行 go test -bench=. 时,会清晰展示每种子测试的结果。 示例:序列化性能对比
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
运行特定子基准 使用 -bench 标志可运行特定子测试。例如: go test -bench=Concat$ —— 只运行顶层 BenchmarkConcat go test -bench=Concat/StringBuilder —— 只运行指定子项 正则匹配需注意斜杠路径格式。调试时可结合 -run 跳过单元测试,如 -run=^$ 。 基本上就这些。用好sub-benchmark能让性能测试更有条理,结果更易读。

相关文章