xfy
f145a8770e
Apply modern Go patterns across the codebase:
- Replace `interface{}` with `any` (Go 1.18+)
- Use `for range n` instead of `for i := 0; i < n; i++` (Go 1.22+)
- Replace `sort.Slice` with `slices.Sort` from slices package
- Simplify sync.WaitGroup patterns with errgroup where appropriate
- Add Makefile targets for modernize analyzer
Total: 84 files updated, net reduction of 79 lines
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
148 lines
3.1 KiB
Go
148 lines
3.1 KiB
Go
// Package bodylimit 提供请求体大小限制中间件的基准测试。
|
||
//
|
||
// 作者：xfy
|
||
package bodylimit
|
||
|
||
import (
|
||
"bytes"
|
||
"testing"
|
||
|
||
"github.com/valyala/fasthttp"
|
||
)
|
||
|
||
// BenchmarkBodyLimitProcess 基准测试限制检查（无超限情况）。
|
||
//
|
||
// 测试中间件处理正常请求（不触发限制）的性能。
|
||
func BenchmarkBodyLimitProcess(b *testing.B) {
|
||
bl, err := New("1mb")
|
||
if err != nil {
|
||
b.Fatalf("创建中间件失败: %v", err)
|
||
}
|
||
|
||
nextHandler := func(ctx *fasthttp.RequestCtx) {
|
||
ctx.SetStatusCode(fasthttp.StatusOK)
|
||
}
|
||
handler := bl.Process(nextHandler)
|
||
|
||
// 准备请求上下文
|
||
body := []byte("test body content")
|
||
|
||
b.ResetTimer()
|
||
b.ReportAllocs()
|
||
|
||
for b.Loop() {
|
||
ctx := &fasthttp.RequestCtx{}
|
||
ctx.Request.Header.SetMethod("POST")
|
||
ctx.Request.Header.SetContentLength(len(body))
|
||
ctx.Request.SetBodyStream(bytes.NewReader(body), len(body))
|
||
|
||
handler(ctx)
|
||
}
|
||
}
|
||
|
||
// BenchmarkBodyLimitGetLimit 基准测试获取路径限制。
|
||
//
|
||
// 测试 GetLimit 方法的性能，该方法在每次请求时都会被调用。
|
||
func BenchmarkBodyLimitGetLimit(b *testing.B) {
|
||
bl, err := New("1mb")
|
||
if err != nil {
|
||
b.Fatalf("创建中间件失败: %v", err)
|
||
}
|
||
|
||
// 添加多个路径配置
|
||
paths := []string{
|
||
"/api/v1/users",
|
||
"/api/v1/posts",
|
||
"/api/v2/upload",
|
||
"/admin/settings",
|
||
"/static/images",
|
||
}
|
||
limits := []string{"10kb", "100kb", "5mb", "50kb", "20mb"}
|
||
|
||
for i, path := range paths {
|
||
if err := bl.AddPathLimit(path, limits[i]); err != nil {
|
||
b.Fatalf("添加路径限制失败: %v", err)
|
||
}
|
||
}
|
||
|
||
testPaths := []string{
|
||
"/api/v1/users/123",
|
||
"/api/v2/upload/file",
|
||
"/other/path",
|
||
"/admin/settings/general",
|
||
}
|
||
|
||
b.ResetTimer()
|
||
b.ReportAllocs()
|
||
|
||
for i := 0; b.Loop(); i++ {
|
||
path := testPaths[i%len(testPaths)]
|
||
_ = bl.GetLimit(path)
|
||
}
|
||
}
|
||
|
||
// BenchmarkBodyLimitPathMatching 基准测试多路径配置匹配。
|
||
//
|
||
// 测试在有大量路径配置时 GetLimit 的性能（包括 RWMutex 锁开销）。
|
||
func BenchmarkBodyLimitPathMatching(b *testing.B) {
|
||
bl, err := New("1mb")
|
||
if err != nil {
|
||
b.Fatalf("创建中间件失败: %v", err)
|
||
}
|
||
|
||
// 添加大量路径配置
|
||
for i := range 100 {
|
||
path := "/api/v" + string(rune('0'+i%10)) + "/resource" + string(rune('0'+i%10))
|
||
size := "1mb"
|
||
if i%3 == 0 {
|
||
size = "10mb"
|
||
} else if i%5 == 0 {
|
||
size = "100kb"
|
||
}
|
||
if err := bl.AddPathLimit(path, size); err != nil {
|
||
b.Fatalf("添加路径限制失败: %v", err)
|
||
}
|
||
}
|
||
|
||
testPaths := []string{
|
||
"/api/v1/resource1/123",
|
||
"/api/v5/resource5/upload",
|
||
"/other/path",
|
||
"/api/v9/resource9/data",
|
||
}
|
||
|
||
b.ResetTimer()
|
||
b.ReportAllocs()
|
||
|
||
for i := 0; b.Loop(); i++ {
|
||
path := testPaths[i%len(testPaths)]
|
||
_ = bl.GetLimit(path)
|
||
}
|
||
}
|
||
|
||
// BenchmarkParseSize 基准测试大小字符串解析。
|
||
//
|
||
// 测试 ParseSize 函数解析各种大小字符串的性能。
|
||
func BenchmarkParseSize(b *testing.B) {
|
||
sizes := []string{
|
||
"1024",
|
||
"1kb",
|
||
"10kb",
|
||
"1mb",
|
||
"10mb",
|
||
"1.5mb",
|
||
"1gb",
|
||
}
|
||
|
||
b.ResetTimer()
|
||
b.ReportAllocs()
|
||
|
||
for i := 0; b.Loop(); i++ {
|
||
sizeStr := sizes[i%len(sizes)]
|
||
_, err := ParseSize(sizeStr)
|
||
if err != nil {
|
||
b.Fatalf("ParseSize(%q) 失败: %v", sizeStr, err)
|
||
}
|
||
}
|
||
}
|