如何在 Go 中获取并保存基准测试结果

本文介绍如何使用 go 标准库中的 `testing.benchmark` 函数主动执行基准测试，并通过 `testing.benchmarkresult` 结构体捕获、格式化和持久化测试结果，替代默认的命令行输出方式。

在 Go 中，go test -bench 命令会自动发现并运行以 Benchmark 开头的函数，将结果打印到终端。但若需程序化访问、比较或导出基准数据（例如生成报告、写入 JSON/CSV 文件、集成 CI 指标监控），则需绕过默认流程，手动调用 testing.Benchmark —— 它接收一个 func(*testing.B) 类型的基准函数，并返回完整的 testing.BenchmarkResult 实例。

testing.BenchmarkResult 是一个公开结构体，包含以下关键字段：

• N: 实际执行的迭代次数；
• T: 总耗时（纳秒）；
• Bytes: 每次操作处理的字节数（常用于内存密集型 benchmark）；
• MemAllocs / MemBytes: 内存分配统计（需配合 -benchmem 使用）；
• Extra: 用户自定义的额外指标（如 b.ReportMetric(123.4, "ms/op") 可写入）。

下面是一个完整可运行示例，展示如何执行基准、提取结果并写入文件：

package main

import (
    "fmt"
    "os"
    "testing"
    "time"
)

func Add(a, b int) int {
    time.Sleep(10 * time.Microsecond) // 模拟轻量计算开销
    return a + b
}

func BenchAdd(b *testing.B) {
    for i := 0; i < b.N; i++ {
        _ = Add(1, 2)
    }
}

func main() {
    // 主动执行基准测试（注意：此调用会重置计时器、自动调整 b.N）
    res := testing.Benchmark(BenchAdd)

    // 格式化为标准 benchmark 输出风格（如 "120000\t10000 ns/op"）
    summary := fmt.Sprintf("%d\t%d ns/op", res.N, res.T/int64(res.N))

    // 打印到控制台
    fmt.Println("Benchmark result:", summary)
    fmt.Printf("Full result: %+v\n", res)

    // ✅ 写入文件（生产环境建议添加错误处理）
    err := os.WriteFile("benchmark_result.txt", []byte(summary+"\n"), 0644)
    if err != nil {
        panic("failed to write benchmark result: " + err.Error())
    }
    fmt.Println("✅ Saved to benchmark_result.txt")
}

⚠️ 注意事项：testing.Benchmark 只能在非测试主程序中调用（即 main 包的 main() 函数），不能在 *_test.go 文件中直接用于 go test 流程，否则会引发 panic 或行为异常；若需同时支持 go test -bench 和程序化导出，建议将基准逻辑封装为独立函数（如 runAddBenchmark(b *testing.B)），并在 BenchmarkXxx 和 main() 中分别调用；要获取内存分配数据，需在 BenchAdd 中调用 b.ReportAllocs()，并在 testing.Benchmark 调用前确保环境支持（通常无问题）；testing.Benchmark 会自动执行多次采样并选择最优结果（类似 -benchmem 的逻辑），因此其 res.T 是稳定、去噪后的总耗时。

通过这种方式，你可以灵活地将基准结果序列化为 JSON、追加至日志、上传至监控平台，或与历史数据对比，真正实现可编程、可审计、可自动化的性能验证闭环。