Go 内存优化：sync.Pool 原理、踩坑与 Benchmark 实测

package main

import (
	"bytes"
	"sync"
)

var bufferPool = sync.Pool{
	New: func() any {
		// 预分配适当的初始容量，避免频繁扩容
		return bytes.NewBuffer(make([]byte, 0, 512))
	},
}

func LogFormat(topic string, data []byte) []byte {
	// 1. 获取对象
	buf := bufferPool.Get().(*bytes.Buffer)
	
	// 2. 极其重要：放回前/取出后必须重置状态！
	buf.Reset()
	
	// 3. 确保在函数退出时放回池中
	defer bufferPool.Put(buf)

	buf.WriteString("topic:")
	buf.WriteString(topic)
	buf.WriteString(";data:")
	buf.Write(data)

	// 注意：如果直接返回 buf.Bytes() 并在外部长期持有，可能会有并发数据竞争问题
	// 这里复制一份干净的数据返回
	res := make([]byte, buf.Len())
	copy(res, buf.Bytes())
	return res
}

package main

import (
	"bytes"
	"sync"
	"testing"
)

var testPool = sync.Pool{
	New: func() any {
		return bytes.NewBuffer(make([]byte, 0, 1024))
	},
}

// 模拟每次新建
func BenchmarkWithoutPool(b *testing.B) {
	b.ReportAllocs()
	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			buf := bytes.NewBuffer(make([]byte, 0, 1024))
			buf.WriteString("user_id=123456&action=click&target=banner")
			_ = buf.Bytes()
		}
	})
}

// 模拟使用 sync.Pool 复用
func BenchmarkWithPool(b *testing.B) {
	b.ReportAllocs()
	b.ResetTimer()
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			buf := testPool.Get().(*bytes.Buffer)
			buf.Reset()
			buf.WriteString("user_id=123456&action=click&target=banner")
			_ = buf.Bytes()
			testPool.Put(buf)
		}
	})
}

goos: darwin
goarch: arm64
pkg: pool-demo
BenchmarkWithoutPool-8   	22435738	        51.23 ns/op	    1024 B/op	       1 allocs/op
BenchmarkWithPool-8      	89452140	        13.15 ns/op	       0 B/op	       0 allocs/op
PASS
ok  	pool-demo	2.412s

if buf.Cap() <= 64*1024 { // 超过 64KB 的大对象直接丢弃，交给 GC 回收
    bufferPool.Put(buf)
}