Golang 函数测试中的性能基准测试

go 中的性能基准测试衡量函数效率，通过在以 benchmark 开头的函数中编写基准测试代码实现。testing.b 类型提供 resettimer()、stoptimer() 和 n 属性控制基准测试行为。例如，计算斐波那契数的函数基准测试表明执行 fib(30) 需要约 2,767,425 纳秒。优化基准测试代码以避免开销，多次运行以获得准确结果。

Go 函数测试中的性能基准测试

性能基准测试是衡量函数效率的重要工具。在 Go 中，testing 包提供了针对函数执行时间的基准测试功能。

编写性能基准测试

编写一个性能基准测试需要创建一个以 Benchmark 开头的函数，后面跟上要测试的函数名：

func BenchmarkFib(b *testing.B) {
    // 基准测试代码
}

使用 testing.B

testing.B 类型提供以下方法来控制基准测试：

• ResetTimer(): 重置计时器。
• StopTimer(): 停止计时器并记录时间。
• N: 执行基准测试的次数。

实战案例

让我们对一个计算斐波那契数的函数进行基准测试：

func Fib(n int) int {
    if n <= 1 {
        return n
    }
    return Fib(n-1) + Fib(n-2)
}

func BenchmarkFib(b *testing.B) {
    for i := 0; i < b.N; i++ {
        Fib(30)
    }
}

在终端中运行测试：

go test -bench=.

输出将如下所示：

BenchmarkFib  2767425 ns/op

这意味着用 30 作为参数执行 Fib 函数的基准测试需要大约 2,767,425 纳秒（2767 毫秒）。

提示

• 使用 -benchmem 标志来测量基准测试的内存分配。
• 优化基准测试代码以避免开销，例如创建不必要的变量。
• 多次运行基准测试以获得更准确的结果。