本文内容总结于 字节跳动青年训练营 第五届后端组
Go自带了一些性能测试工具,其中BenchMark是较为重要的一个。
我们以计算斐波那契数列的示例来展示BenchMark的使用
package Benchmark
import "testing"
func Fib(n int) int {
if n < 2 {
return n
}
return Fib(n-1) + Fib(n-2)
}
// from fib_test.go
func BenchmarkFib(b *testing.B) {
// 运行Fib 函数b.N次
for n := 0; n < b.N; n++ {
Fib(10)
}
}
其中func BenchmarkFib()为测试函数,传入的参数为b *testing.B,当运行Benchmark基准测试的时候,BenchMark会抓取参数为 *testing.B的函数进行测试
- 基准测试代码文件必须是_test.go结尾,和单元测试一样;
- 基准测试的函数以Benchmark开头;
- 参数须为 *testing.B;
- 基准测试函数不能有返回值;
- b.N是基准测试指定的循环次数,不能由用户指定
然后使用命令行输入:
go test -bench=. -benchmem -count=3
其中go test用于启动测试,使用-bench指明启动的是基准测试,并且-bench=也用于指明测试的是什么函数,-bench=.的意思是测试所有Bench函数。 -benchmem用于指明基准测试的内存相关设置,现在暂时不使用。-count=3指明运行多少个基准测试。
运行结果如下
其中:
- BenchmarkFib-8 为测试的函数名
- 第二列表示基准测试进行了多少次测试,这就是b.N的值
- 第三列表示每次执行花费了多少时间
- 第四列表示每次执行需要多少内存
- 第五列表示每次执行申请了多少次内存