Golang基准测试内存使用量统计示例-Golang-PHP中文网

Golang基准测试内存使用量统计示例

P粉602998670

发布： 2025-09-15 14:19:01

原创

367人浏览过

基准测试可统计内存分配，通过b.ReportAllocs()记录每次操作的内存分配次数和字节数，结合ResetTimer确保数据准确。

golang基准测试内存使用量统计示例

Go语言的基准测试不仅能测量函数执行时间，还能统计内存分配情况。通过

testing.B

登录后复制

提供的方法，可以获取每次操作的内存分配次数和字节数，帮助优化性能关键代码。

编写基础基准测试

以一个简单的字符串拼接函数为例，展示如何启用内存统计：

// example.go
package main

func ConcatStrings(strs []string) string {
    var result string
    for _, s := range strs {
        result += s
    }
    return result
} // example_test.go
package main

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

func BenchmarkConcatStrings(b *testing.B) {
    inputs := make([]string, 100)
    for i := range inputs {
        inputs[i] = randomString(10)
    }

    b.ResetTimer()
    for i := 0; i         ConcatStrings(inputs)
    }
}

func randomString(n int) string {
    const letters = "abcdefghijklmnopqrstuvwxyz"
    sb := strings.Builder{}
    sb.Grow(n)
    for i := 0; i         sb.WriteByte(letters[rand.Intn(len(letters))])
    }
    return sb.String()
}

启用内存统计输出

运行基准测试时添加

-benchmem

登录后复制

标志，显示内存分配数据：

go test -bench=BenchmarkConcatStrings -benchmem

输出示例：

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

BenchmarkConcatStrings-8 1000000 1250 ns/op 9600 B/op 99 allocs/op

其中：

Otter.ai

一个自动的会议记录和笔记工具，会议内容生成和实时转录

查看详情

B/op：每次操作分配的字节数
allocs/op：每次操作的内存分配次数

对比优化方案

使用

strings.Builder

登录后复制

改进拼接逻辑：

func ConcatWithBuilder(strs []string) string {
    var sb strings.Builder
    sb.Grow(1000) // 预分配容量减少重新分配
    for _, s := range strs {
        sb.WriteString(s)
    }
    return sb.String()
}

添加对应基准测试：

func BenchmarkConcatWithBuilder(b *testing.B) {
    inputs := make([]string, 100)
    for i := range inputs {
        inputs[i] = randomString(10)
    }

    b.ResetTimer()
    for i := 0; i         ConcatWithBuilder(inputs)
    }
}

运行对比：

BenchmarkConcatStrings-8 1000000 1250 ns/op 9600 B/op 99 allocs/op
BenchmarkConcatWithBuilder-8 5000000 300 ns/op 1200 B/op 2 allocs/op

结果显示，使用