Golang性能测试实现 基准测试写法

testing

_test.go

my_package_test.go

BenchmarkXxx

*testing.B

package mypackage

import (
    "testing"
    "time"
)

// 一个简单的示例函数，我们想测试它的性能
func SomeExpensiveOperation(n int) int {
    sum := 0
    for i := 0; i < n; i++ {
        sum += i
    }
    // 模拟一些耗时操作
    time.Sleep(time.Microsecond * time.Duration(n/1000))
    return sum
}

// BenchmarkSomeExpensiveOperation 是对 SomeExpensiveOperation 的基准测试
func BenchmarkSomeExpensiveOperation(b *testing.B) {
    // b.N 是基准测试框架为我们确定的循环次数，确保测量结果的统计学意义
    for i := 0; i < b.N; i++ {
        SomeExpensiveOperation(10000) // 每次迭代都执行这个操作
    }
}

// 另一个例子，可能涉及内存分配
func generateSlice(size int) []int {
    s := make([]int, size)
    for i := 0; i < size; i++ {
        s[i] = i
    }
    return s
}

func BenchmarkGenerateSlice(b *testing.B) {
    b.ReportAllocs() // 报告内存分配情况
    for i := 0; i < b.N; i++ {
        generateSlice(1000)
    }
}

// 如果你的测试函数有准备阶段，可以用 b.ResetTimer() 来排除准备时间
func BenchmarkSomeOperationWithSetup(b *testing.B) {
    // 假设这里有一些耗时的初始化工作
    _ = make([]byte, 1024*1024) // 比如分配一个大缓冲区
    b.ResetTimer()              // 重置计时器，确保只测量下面的循环
    for i := 0; i < b.N; i++ {
        // 实际要测试的代码
        _ = SomeExpensiveOperation(100)
    }
}

// 有时候你可能想在测试过程中暂停计时器
func BenchmarkOperationWithPause(b *testing.B) {
    for i := 0; i < b.N; i++ {
        b.StopTimer() // 暂停计时器
        // 模拟一个不希望计入性能测量的操作，比如数据准备
        time.Sleep(time.Millisecond)
        b.StartTimer() // 重新开始计时
        _ = SomeExpensiveOperation(10)
    }
}

go test -bench=.

.

go test -bench=SomeExpensiveOperation

_test.go

testing

*testing.T

t.Error()

t.Fail()

*testing.B

b.N

b.ResetTimer()

b.ReportAllocs()

go test

-bench

b.N

b.ResetTimer()

b.N

b.RunParallel

b.RunParallel

// 假设有一个并发安全的计数器
type ConcurrentCounter struct {
    count int
    mu    sync.Mutex
}

func (c *ConcurrentCounter) Increment() {
    c.mu.Lock()
    c.count++
    c.mu.Unlock()
}

func BenchmarkConcurrentIncrement(b *testing.B) {
    counter := &ConcurrentCounter{}
    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() {
            counter.Increment()
        }
    })
}

b.ReportAllocs()

go test -bench=.

goos: darwin
goarch: arm64
pkg: my_project/mypackage
cpu: Apple M1
BenchmarkSomeExpensiveOperation-8           100000             10373 ns/op
BenchmarkGenerateSlice-8                    100000             10373 ns/op           8000 B/op          2 allocs/op
BenchmarkSomeOperationWithSetup-8          1000000              1037 ns/op
BenchmarkOperationWithPause-8              1000000              1037 ns/op
BenchmarkConcurrentIncrement-8            10000000               100 ns/op
PASS
ok      my_project/mypackage    1.234s

benchstat