如何通过基准测试评估 Golang 函数的性能？-Golang-PHP中文网

如何通过基准测试评估 Golang 函数的性能？

王林

发布： 2024-09-19 16:30:03

原创

777人浏览过

使用 go 测试包的基准测试来评估函数性能：创建包含要基准测试的代码的基准测试函数。将基准测试函数添加到_test.go文件中。使用 go test -bench=. 命令执行基准测试。分析输出中的运行时间以识别更快、更有效的实现。

如何通过基准测试评估 Golang 函数的性能？

如何通过基准测试评估 Go 函数的性能

基准测试是衡量函数或算法性能的重要工具。在 Go 中，testing 包提供了进行基准测试所需的功能。本文将指导你使用testing 包执行基准测试，并提供一个实战案例。

基准测试函数

要对 Go 函数进行基准测试，需要创建一个包含以下内容的基准测试函数：

立即学习“go语言免费学习笔记（深入）”；

func BenchmarkName(b *testing.B) {
  // 要基准测试的代码
  for i := 0; i < b.N; i++ {
    // 执行要基准测试的代码
  }
}

登录后复制

其中：

b 是一个*testing.B 指针，它提供了对基准测试运行的控制。
Name 是基准测试函数的名称。
b.N 指定基准测试将运行的迭代次数。

创建基准测试

创建基准测试后，需要将其添加到测试文件中。在 Go 中，测试文件通常以_test.go 结尾。基准测试函数需要与其他测试函数一起位于以下代码块中：

func TestBenchmark(t *testing.T) {
  // 基准测试函数
  BenchmarkName(t)
}

登录后复制

执行基准测试

使用go test 命令执行基准测试，后跟-bench 标志：

go test -bench=.

登录后复制

这将运行所有基准测试函数并打印结果。

实战案例：比较排序算法

为了展示基准测试如何用于比较函数的性能，下面是一个比较两种排序算法（冒泡排序和快速排序）的例子：

func BenchmarkBubbleSort(b *testing.B) {
  for i := 0; i < b.N; i++ {
    arr := []int{10, 5, 2, 7, 8, 1}
    bubbleSort(arr)
  }
}

func BenchmarkQuickSort(b *testing.B) {
  for i := 0; i < b.N; i++ {
    arr := []int{10, 5, 2, 7, 8, 1}
    quickSort(arr, 0, len(arr)-1)
  }
}

登录后复制

执行基准测试：