压测Benchmark基准测试 - Jxy 博客

Benchmark基准测试

简单基测

当我们尝试去优化代码的性能时，首先得知道当前的性能怎么样。

Go 语言标准库内置的 testing 测试框架提供了基准测试(benchmark)的能力，能让我们很容易地对某一段代码进行性能测试。

性能测试受环境的影响很大，为了保证测试的可重复性，在进行性能测试时，尽可能地保持测试环境的稳定。

• 机器处于闲置状态，测试时不要执行其他任务，也不要和其他人共享硬件资源。
• 机器是否关闭了节能模式，一般笔记本会默认打开这个模式，测试时关闭。
• 避免使用虚拟机和云主机进行测试，一般情况下，为了尽可能地提高资源的利用率，虚拟机和云主机 CPU 和内存一般会超分配，超分机器的性能表现会非常地不稳定。

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

Benchmark是如何工作的

单元测试：

• 运行当前package内的所有用例：go test 【包名】或 go test .
• 递归执行当前目录下所有用例：go test ./... 或 go test 【目录名】/...
• 运行指定文件单元测试： go test -v [单元测试文件]. // 如 go test -v foo_test.go
• 运行指定单元测试用例：go test -v [单元测试文件] -run [单元测试函数]. //如 go test -v foo_test.go -run TestFoo
• 压测：go test -bench .

import (
    "math/rand"
    "testing"
    "time"
)

func generateWithCap(n int) []int {
    rand.Seed(time.Now().UnixNano())
    nums := make([]int, 0, n)
    for i := 0; i < n; i++ {
        nums = append(nums, rand.Int())
    }
    return nums
}

func generate(n int) []int {
    rand.Seed(time.Now().UnixNano())
    nums := make([]int, 0)
    for i := 0; i < n; i++ {
        nums = append(nums, rand.Int())
    }
    return nums
}

func BenchmarkGenerateWithCap(b *testing.B) {
    for n := 0; n < b.N; n++ {
        generateWithCap(1000000)
    }
}

func BenchmarkGenerate(b *testing.B) {
    for n := 0; n < b.N; n++ {
        generate(1000000)
    }
}

$ go test -bench='Generate' -benchmem  -cpu=2,4,8  .    // 执行的指令

goos: darwin   // 操作系统
goarch: arm64  // cpu处理器
pkg: itxj.com/xj-utils/utils/algorithm   // 测试执行所在的包
BenchmarkGenerateWithCap-2            82          13937680 ns/op         8003584 B/op           1 allocs/op
BenchmarkGenerateWithCap-4            84          14083593 ns/op         8003595 B/op           1 allocs/op
BenchmarkGenerateWithCap-8            85          13938689 ns/op         8003616 B/op           1 allocs/op
BenchmarkGenerate-2                   68          15586489 ns/op        45188365 B/op           40 allocs/op
BenchmarkGenerate-4                   70          15646168 ns/op        45188417 B/op           40 allocs/op
BenchmarkGenerate-8                   73          15948826 ns/op        45188464 B/op           41 allocs/op
// 执行测试函数所用CPU核数               执行次数   每次花费时间左移动9位为秒s 每次分配内存字节B左移6位为m  每次分配内存次数
// 如用2核CPU执行BenchmarkGenerate函数               0014083593ns=0.014s    8003584B=7.9m          

PASS
ok      itxj.com/xj-utils/utils/algorithm       7.682s    // 总共执行耗时

运行测试用例：

func fib(n int) int {
    if n == 0 || n == 1 {
        return n
    }
    return fib(n-2) + fib(n-1)
}

import "testing"

func BenchmarkFib(b *testing.B) {
    for n := 0; n < b.N; n++ {
        fib(30) // run fib(30) b.N times
    }
}

提高准确度

内存分配情况

不同输入复杂度测试

其他注意事项

• ResetTimer
• StopTimer
• StartTimer