xfy
8ed800271d
- 所有 *_bench_test.go 文件从 for i := 0; i < b.N; i++ 改为 for b.Loop() - 部分测试文件从 for i := 0; i < N; ... 改为 for range N 或 for i := range N - 涵盖模块: cache, handler, http2, http3, loadbalance, logging, lua, middleware/accesslog, middleware/bodylimit, middleware/rewrite, middleware/security, netutil, resolver, server, ssl, stream Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
// Package accesslog 提供访问日志中间件的基准测试。
|
||
//
|
||
// 该文件测试日志记录的性能开销。
|
||
//
|
||
// 作者：xfy
|
||
package accesslog
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/valyala/fasthttp"
|
||
"rua.plus/lolly/internal/config"
|
||
)
|
||
|
||
// BenchmarkAccessLogProcess 测试访问日志中间件处理性能。
|
||
func BenchmarkAccessLogProcess(b *testing.B) {
|
||
cfg := &config.LoggingConfig{
|
||
Access: config.AccessLogConfig{
|
||
Path: "/dev/null",
|
||
Format: "combined",
|
||
},
|
||
}
|
||
al := New(cfg)
|
||
defer func() { _ = al.Close() }()
|
||
|
||
mockHandler := func(ctx *fasthttp.RequestCtx) {
|
||
ctx.SetStatusCode(fasthttp.StatusOK)
|
||
ctx.SetBodyString("Hello, World!")
|
||
}
|
||
|
||
handler := al.Process(mockHandler)
|
||
|
||
b.ResetTimer()
|
||
for b.Loop() {
|
||
ctx := &fasthttp.RequestCtx{}
|
||
ctx.Request.Header.SetMethod(fasthttp.MethodGet)
|
||
ctx.Request.SetRequestURI("/api/test")
|
||
ctx.Request.Header.SetHost("example.com")
|
||
handler(ctx)
|
||
}
|
||
}
|
||
|
||
// BenchmarkAccessLogProcessParallel 测试并发场景下的访问日志性能。
|
||
func BenchmarkAccessLogProcessParallel(b *testing.B) {
|
||
cfg := &config.LoggingConfig{
|
||
Access: config.AccessLogConfig{
|
||
Path: "/dev/null",
|
||
Format: "combined",
|
||
},
|
||
}
|
||
al := New(cfg)
|
||
defer func() { _ = al.Close() }()
|
||
|
||
mockHandler := func(ctx *fasthttp.RequestCtx) {
|
||
ctx.SetStatusCode(fasthttp.StatusOK)
|
||
ctx.SetBodyString("OK")
|
||
}
|
||
|
||
handler := al.Process(mockHandler)
|
||
|
||
b.ResetTimer()
|
||
b.RunParallel(func(pb *testing.PB) {
|
||
for pb.Next() {
|
||
ctx := &fasthttp.RequestCtx{}
|
||
ctx.Request.Header.SetMethod(fasthttp.MethodGet)
|
||
ctx.Request.SetRequestURI("/api/test")
|
||
handler(ctx)
|
||
}
|
||
})
|
||
}
|