登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

Go benchmark 怎样排除初始化耗时

来源:17golang原创

时间:2026-10-22 01:13:38 266浏览 收藏

Go benchmark 排除初始化耗时的关键,是把一次性准备工作放到计时区间外。最稳妥的写法是先创建查找表、解析模板或准备输入,再进入 benchmark 循环;如果初始化必须写在函数中,就在真正的测试动作开始前调用 b.ResetTimer()。它会清零已计入的耗时和内存分配计数,但不会改变计时器当前是否运行。

官方资料:https://pkg.go.dev/testing

要点速览
  • 一次性初始化优先放在循环前;ResetTimer 适合清除函数前半段已经产生的计时和分配。
  • 每轮都要构造输入时,才考虑用 StopTimer 和 StartTimer 把准备阶段隔开。
  • go test -bench 默认只报告计时区间的平均结果;并行 benchmark 的 worker 不应操作全局计时器。

为什么初始化会混进 ns/op

基准函数开始执行后,testing 包会启动 benchmark 计时。若把昂贵的构造、文件读取、模板解析或连接准备写在循环中,报告的 ns/op 就同时包含了“准备对象”和“执行目标操作”两部分。这样得到的数字可以描述整个工作流,却不能回答“这个函数本身有多快”。

因此,先把测试问题拆成两个边界:初始化只负责准备可复用的输入,计时区间只负责被比较的操作。初始化确实是业务成本时,可以单独再写一个 benchmark;不要把两种成本塞进同一个指标后再凭感觉解释。

Go benchmark 中初始化准备区、ResetTimer 计时边界和目标操作区之间关系的静态说明图
图1:benchmark 初始化与目标操作的计时边界说明图,不是运行截图。

用 ResetTimer 把一次性准备排除出去

下面的例子模拟一个需要先建立索引的查询。buildIndex 只执行一次,Lookup 才是要比较的操作。把初始化放在循环前已经足够清晰;ResetTimer 则适合初始化代码必须位于 benchmark 函数内、并且不想让前面的计时或分配进入结果的场景。

package bench

import "testing"

type index struct {
    values map[string]int
}

func buildIndex() *index {
    // 模拟一次性建立索引;这部分不属于 Lookup 的性能问题。
    return &index{values: map[string]int{"go": 1, "benchmark": 2}}
}

func (idx *index) Lookup(key string) (int, bool) {
    // 基准目标是一次 map 查询,返回值用于避免示例看起来像无效调用。
    value, ok := idx.values[key]
    return value, ok
}

func BenchmarkIndexLookup(b *testing.B) {
    idx := buildIndex()
    b.ResetTimer()

    for i := 0; i 

ResetTimer 不会停止计时器,它只是把已经累计的时间、内存分配计数和用户自定义指标清零。所以如果后面还有一次性准备,仍要把它放在 ResetTimer 之前;如果只需让计时暂停,再恢复,就用下一节的成对方法。

每轮输入必须构造时怎样暂停计时

有些测试不能复用同一个输入,例如每轮要从不同的字节片段创建请求对象。此时可以暂停计时完成准备,再恢复计时执行目标动作。要注意:暂停和恢复本身会增加代码复杂度,只有准备成本确实不属于本次测量目标时才这样做。

func BenchmarkDecodePayload(b *testing.B) {
    for i := 0; i 

如果输入可以提前生成,通常更推荐在计时前准备一个切片,再在循环里按下标取用。这样既减少 StopTimer/StartTimer 的干扰,也让 benchmark 更容易复现。若要观察分配,使用 -benchmem,但不要把“初始化分配”和“每轮操作分配”混为一个结论。

# 只运行目标 benchmark,并同时输出每次操作的分配统计
go test -run '^$' -bench '^BenchmarkIndexLookup$' -benchmem ./...

并行 benchmark 和新版循环写法的边界

RunParallel 会把 b.N 次迭代分配给多个 goroutine,计时器对整个 benchmark 具有全局效果。因此不要在 worker 函数里调用 ResetTimer、StopTimer 或 StartTimer;并行场景的输入准备应放到 RunParallel 之前,或放在每个 worker 自己的局部变量中。

func BenchmarkLookupParallel(b *testing.B) {
    idx := buildIndex()
    b.RunParallel(func(pb *testing.PB) {
        // 每个 worker 使用自己的键,避免共享可变状态造成额外干扰。
        for pb.Next() {
            _, _ = idx.Lookup("benchmark")
        }
    })
}

较新的 Go 版本还提供 b.Loop(),它会在第一次进入循环时重置 benchmark 计时,让循环前的准备自然落在计时区间外。为了兼容仍使用传统 b.N 的项目,ResetTimer 仍然值得掌握;选择哪种写法要和项目支持的 Go 版本保持一致。

复查清单
  • 初始化是否只执行一次,且在 ResetTimer 前完成?
  • 每轮输入是否真的属于测量对象?不属于时是否暂停了计时?
  • 是否使用 -benchmem 分开观察分配,而不是用 ns/op 解释所有成本?
  • 是否在 RunParallel 的 worker 内误用了全局计时器?
Go benchmark 中 ResetTimer、StopTimer、StartTimer、benchmem 和 RunParallel 的职责关系静态说明图
图2:计时器、输入准备和并行 worker 的关系说明图,不是监控平台截图。

常见问题

初始化放在 benchmark 函数里一定要调用 ResetTimer 吗?

不一定。只要初始化发生在目标循环之前,结果通常已经清楚;当函数前段存在会被计时的准备工作时,调用 ResetTimer 能显式清掉它。

StopTimer 和 ResetTimer 有什么区别?

StopTimer 是暂停,之后可以用 StartTimer 继续;ResetTimer 是把已经累计的时间和分配计数归零,并保持当前计时器的运行状态。

为什么 RunParallel 里不能 ResetTimer?

这些计时方法对整个 benchmark 有全局影响。worker 之间同时操作会让测量边界不稳定,所以应在进入并行执行前完成准备。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>