153 lines
3.2 KiB
Go
153 lines
3.2 KiB
Go
// Package compression 提供压缩 writer 池化效果测试。
|
||
//
|
||
// 该文件对比新建 vs 池化复用的分配差异。
|
||
//
|
||
// 测试场景：
|
||
// - NewWriter: 每次新建 gzip.Writer
|
||
// - PoolWriter: 从 sync.Pool 获取复用
|
||
//
|
||
// 目标：池化后 ≤2 allocs/op
|
||
//
|
||
// 作者：xfy
|
||
package compression
|
||
|
||
import (
|
||
"bytes"
|
||
"compress/gzip"
|
||
"sync"
|
||
"testing"
|
||
|
||
"github.com/valyala/fasthttp"
|
||
"rua.plus/lolly/internal/config"
|
||
"rua.plus/lolly/internal/testutil"
|
||
)
|
||
|
||
// BenchmarkGzipPool_GetPut 测试 gzip.Writer 池直接操作。
|
||
//
|
||
// 验证 sync.Pool 的分配效果。
|
||
func BenchmarkGzipPool_GetPut(b *testing.B) {
|
||
pool := &sync.Pool{
|
||
New: func() any {
|
||
return gzip.NewWriter(nil)
|
||
},
|
||
}
|
||
|
||
buf := bytes.NewBuffer(nil)
|
||
|
||
b.ReportAllocs()
|
||
b.ResetTimer()
|
||
|
||
for b.Loop() {
|
||
w := pool.Get().(*gzip.Writer)
|
||
w.Reset(buf)
|
||
// 模拟使用
|
||
_, _ = w.Write([]byte("test data"))
|
||
_ = w.Close()
|
||
buf.Reset()
|
||
pool.Put(w)
|
||
}
|
||
}
|
||
|
||
// BenchmarkGzipWriter_New 测试每次新建 gzip.Writer。
|
||
//
|
||
// 对比基准：每次创建新 Writer。
|
||
func BenchmarkGzipWriter_New(b *testing.B) {
|
||
data := testutil.GenerateTestData(testutil.Size1KB)
|
||
|
||
b.ReportAllocs()
|
||
b.ResetTimer()
|
||
|
||
for b.Loop() {
|
||
buf := bytes.NewBuffer(nil)
|
||
w := gzip.NewWriter(buf)
|
||
_, _ = w.Write(data)
|
||
_ = w.Close()
|
||
}
|
||
}
|
||
|
||
// BenchmarkGzipWriter_Pool 测试池化复用 gzip.Writer。
|
||
//
|
||
// 目标：≤2 allocs/op
|
||
func BenchmarkGzipWriter_Pool(b *testing.B) {
|
||
data := testutil.GenerateTestData(testutil.Size1KB)
|
||
|
||
cfg := &config.CompressionConfig{
|
||
Type: "gzip",
|
||
Level: 6,
|
||
MinSize: 100,
|
||
Types: []string{"application/json"},
|
||
}
|
||
mw, _ := New(cfg)
|
||
|
||
b.ReportAllocs()
|
||
b.ResetTimer()
|
||
|
||
for b.Loop() {
|
||
_ = mw.compressWithPool(data, mw.gzipPool)
|
||
}
|
||
}
|
||
|
||
// BenchmarkCompressionMiddleware_Pool 测试中间件池化效果。
|
||
//
|
||
// 模拟完整请求处理路径。
|
||
func BenchmarkCompressionMiddleware_Pool(b *testing.B) {
|
||
cfg := &config.CompressionConfig{
|
||
Type: "gzip",
|
||
Level: 6,
|
||
MinSize: 100,
|
||
Types: []string{"text/html", "application/json"},
|
||
}
|
||
mw, _ := New(cfg)
|
||
|
||
// 创建测试响应
|
||
data := testutil.GenerateTestData(testutil.Size10KB)
|
||
|
||
b.ReportAllocs()
|
||
b.ResetTimer()
|
||
|
||
for b.Loop() {
|
||
ctx := &fasthttp.RequestCtx{}
|
||
ctx.Request.Header.SetMethod(fasthttp.MethodGet)
|
||
ctx.Request.SetRequestURI("/test")
|
||
ctx.Request.Header.Set("Accept-Encoding", "gzip")
|
||
ctx.Response.Header.Set("Content-Type", "application/json")
|
||
ctx.Response.SetBody(data)
|
||
|
||
// 模拟中间件处理
|
||
handler := mw.Process(func(ctx *fasthttp.RequestCtx) {
|
||
ctx.Response.SetBody(data)
|
||
})
|
||
handler(ctx)
|
||
}
|
||
}
|
||
|
||
// BenchmarkGzipCompress_Sizes 测试不同数据大小的池化效果。
|
||
func BenchmarkGzipCompress_Sizes(b *testing.B) {
|
||
cfg := &config.CompressionConfig{
|
||
Type: "gzip",
|
||
Level: 6,
|
||
MinSize: 100,
|
||
}
|
||
mw, _ := New(cfg)
|
||
|
||
sizes := []struct {
|
||
name string
|
||
data []byte
|
||
}{
|
||
{"100B", testutil.GenerateTestData(100)},
|
||
{"1KB", testutil.GenerateTestData(testutil.Size1KB)},
|
||
{"10KB", testutil.GenerateTestData(testutil.Size10KB)},
|
||
{"100KB", testutil.GenerateTestData(testutil.Size100KB)},
|
||
}
|
||
|
||
for _, tc := range sizes {
|
||
b.Run(tc.name, func(b *testing.B) {
|
||
b.ReportAllocs()
|
||
b.ResetTimer()
|
||
for b.Loop() {
|
||
_ = mw.compressWithPool(tc.data, mw.gzipPool)
|
||
}
|
||
})
|
||
}
|
||
}
|