go-基准测试

基准测试

Demo

// fib_test.go
package mainimport "testing"func BenchmarkFib(b *testing.B) {for n := 0; n < b.N; n++ {fib(30) // run fib(30) b.N times}
}func fib(n int) int {if n == 0 || n == 1 {return n}return fib(n-2) + fib(n-1)
}

• benchmark 和普通的单元测试用例一样，都位于 _test.go 文件中。
• 函数名以 Benchmark 开头，参数是 b *testing.B。和普通的单元测试用例很像，单元测试函数名以 Test 开头，参数是 t *testing.T。

执行基准测试

• go test 命令默认不运行 benchmark 用例的，如果我们想运行 benchmark 用例，则需要加上 -bench 参数。

  $ go test -bench="$Fib" . // 通过-bench指定测试函数，或者通过正则匹配一类函数,比如说`-bench='Fib$'`
  goos: windows
  goarch: amd64
  pkg: gan
  cpu: Intel(R) Core(TM) i5-10400 CPU @ 2.90GHz
  BenchmarkFib-12              306           3925085 ns/op
  PASS
  ok      gan     2.218s
  

• 如果想使用多个cpu执行

  $ go test -bench='Fib$' -cpu=2,4 .
  

• 如果想提高测试时间和次数

  $ go test -bench='Fib$' -benchtime=50x .  //执行50次
  $ go test -bench='Fib$' -benchtime=5s .  //执行5s
  $ go test -bench='Fib$' -benchtime=5s -count=3 . //执行5s，一共执行3轮
  

• 查看内存分配情况

  $ go test -bench='Fib$' -benchmem .
  

去除干扰因素

• 重置计时器

  func BenchmarkFib(b *testing.B) {time.Sleep(time.Second * 3) // 模拟耗时准备任务b.ResetTimer() // 重置定时器for n := 0; n < b.N; n++ {fib(30) // run fib(30) b.N times}
  }
  

• 去除准备和后置处理工作耗时影响

  func BenchmarkBubbleSort(b *testing.B) {for n := 0; n < b.N; n++ {b.StopTimer()nums := generateWithCap(10000)b.StartTimer()bubbleSort(nums)}
  }
  

参考： https://geektutu.com/post/hpg-benchmark.html