17370845950

如何使用Golang benchmark测量函数延迟_统计执行耗时
Go testing包原生支持基准测试,需用go test -bench运行且函数名以Benchmark开头、接收*testing.B参数;通过b.N循环执行待测逻辑,用b.ResetTimer()或b.StopTimer()排除初始化开销,b.ReportAllocs()统计内存分配。

Go 的 testing 包原生支持基准测试(benchmark),能准确统计函数执行耗时、吞吐量,并自动进行多次运行取平均,避免单次测量的偶然误差。关键在于用 go test -bench 运行,且函数需符合 BenchmarkXxx(*testing.B) 签名。

编写标准 benchmark 函数

基准函数必须接收 *testing.B 参数,且函数名以 Benchmark 开头。核心是调用 b.N 控制循环次数,把待测逻辑放在循环内:

  • 不要在循环外初始化耗时资源(如打开文件、建连接),除非你明确想测初始化+执行总开销;若只关心纯计算延迟,应把初始化放在 b.ResetTimer() 之前或 b.StopTimer() 区间内
  • b.ReportAllocs() 可开启内存分配统计(如 512 B/op
  • 若函数依赖输入数据,建议在循环外预生成好,避免把数据构造时间算进延迟

示例:

func BenchmarkAdd(b *testing.B) {
    a, bVal := 1, 2
    b.ReportAllocs()
    for i := 0; i < b.N; i++ {
        _ = a + bVal // 待测逻辑
    }
}

运行 benchmark 并解读结果

在包目录下执行:

go test -bench=^BenchmarkAdd$ -benchmem -count=3
  • -bench=^BenchmarkAdd$:正则匹配指定函数(^$ 保证精确匹配)
  • -benchmem:显示内存分配统计
  • -count=3:重复运行 3 次取中位数,减少噪声影响

输出类似:
BenchmarkAdd-8 1000000000 0.34 ns/op 0 B/op 0 allocs/op
含义:在 8 核 CPU 上,执行 10⁹ 次,平均每次耗时 0.34 纳秒,无内存分配。

控制精度与排除干扰

默认情况下,Go 会自动调整 b.N 使单轮测试时长接近 1 秒。若函数极快(如纳秒级),可手动设最小时间避免过短测量:

  • b.SetMinimumTime(500*time.Millisecond):确保至少跑够 500ms 再停止
  • b.RunParallel(func(pb *testing.PB) { ... }):并发运行,适合测并发场景(如 channel 操作、锁竞争)
  • 避免在 benchmark 中打印、写磁盘、调网络 —— 这些 I/O 会严重污染延迟数据

对比多个实现的性能差异

b.Run() 组织子测试,便于横向比较:

func BenchmarkStringConcat(b *testing.B) {
    s1, s2 := "hello", "world"
    b.Run("plus", func(b *testing.B) {
        for i := 0; i < b.N; i++ {
            _ = s1 + s2
        }
    })
    b.Run("fmtSprint", func(b *testing.B) {
        for i := 0; i < b.N; i++ {
            _ = fmt.Sprint(s1, s2)
        }
    })
}

运行后会分别输出 BenchmarkStringConcat/plusBenchmarkStringConcat/fmtSprint 的耗时,直观看出哪种拼接更快。