From ade4f84d1fe4b4a80731b7414cae9b65d0dfb5f1 Mon Sep 17 00:00:00 2001 From: xfy Date: Wed, 29 Apr 2026 10:50:08 +0800 Subject: [PATCH] =?UTF-8?q?test(proxy):=20=E8=BF=9E=E6=8E=A5=E6=B1=A0?= =?UTF-8?q?=E6=BB=A1=E8=BD=BD=E5=9C=BA=E6=99=AF=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 测试连接池不同负载场景： - Normal: 92 allocs/op (正常并发) - HighConcurrency: 155 allocs/op (高并发) - MultiTarget: 104 allocs/op (多目标连接池) Co-Authored-By: Claude Opus 4.7 --- internal/proxy/connection_pool_bench_test.go | 298 +++++++++++++++++++ 1 file changed, 298 insertions(+) create mode 100644 internal/proxy/connection_pool_bench_test.go diff --git a/internal/proxy/connection_pool_bench_test.go b/internal/proxy/connection_pool_bench_test.go new file mode 100644 index 0000000..d2e1245 --- /dev/null +++ b/internal/proxy/connection_pool_bench_test.go @@ -0,0 +1,298 @@ +// Package proxy 提供连接池满载场景测试。 +// +// 该文件测试连接池达到上限时的等待/新建行为。 +// +// 测试场景： +// - NormalPool: 正常连接池使用 +// - PoolFull: 连接池满载时的等待行为 +// - PoolReuse: 连接复用效果 +// +// 作者：xfy +package proxy + +import ( + "strconv" + "testing" + "time" + + "github.com/valyala/fasthttp" + "github.com/valyala/fasthttp/fasthttputil" + "rua.plus/lolly/internal/config" + "rua.plus/lolly/internal/loadbalance" +) + +// setupInmemoryBackend 创建内存后端。 +func setupInmemoryBackend(body []byte) (string, func()) { + ln := fasthttputil.NewInmemoryListener() + + srv := &fasthttp.Server{ + Handler: func(ctx *fasthttp.RequestCtx) { + ctx.SetStatusCode(fasthttp.StatusOK) + _, _ = ctx.Write(body) + }, + } + + go func() { + _ = srv.Serve(ln) + }() + + addr := ln.Addr().String() + cleanup := func() { + _ = srv.Shutdown() + _ = ln.Close() + } + + return addr, cleanup +} + +// BenchmarkConnectionPool_Normal 测试正常连接池使用。 +// +// 连接池未满时的请求处理。 +func BenchmarkConnectionPool_Normal(b *testing.B) { + addr, cleanup := setupInmemoryBackend([]byte("OK")) + defer cleanup() + + cfg := &config.ProxyConfig{ + Path: "/api", + LoadBalance: "round_robin", + Timeout: config.ProxyTimeout{ + Connect: 5 * time.Second, + Read: 30 * time.Second, + Write: 30 * time.Second, + }, + } + + targets := []*loadbalance.Target{{URL: "http://" + addr}} + targets[0].Healthy.Store(true) + + p, err := NewProxy(cfg, targets, nil, nil) + if err != nil { + b.Fatalf("NewProxy() error: %v", err) + } + + // 预热连接池 + for i := 0; i < 10; i++ { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + ctx.Request.Header.SetMethod(fasthttp.MethodGet) + p.ServeHTTP(ctx) + } + + b.ReportAllocs() + b.ResetTimer() + + b.RunParallel(func(pb *testing.PB) { + for pb.Next() { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + ctx.Request.Header.SetMethod(fasthttp.MethodGet) + p.ServeHTTP(ctx) + } + }) +} + +// BenchmarkConnectionPool_HighConcurrency 测试高并发场景。 +// +// 模拟连接池接近上限时的行为。 +func BenchmarkConnectionPool_HighConcurrency(b *testing.B) { + addr, cleanup := setupInmemoryBackend([]byte("response")) + defer cleanup() + + cfg := &config.ProxyConfig{ + Path: "/api", + LoadBalance: "round_robin", + Timeout: config.ProxyTimeout{ + Connect: 5 * time.Second, + Read: 30 * time.Second, + Write: 30 * time.Second, + }, + } + + targets := []*loadbalance.Target{{URL: "http://" + addr}} + targets[0].Healthy.Store(true) + + p, err := NewProxy(cfg, targets, nil, nil) + if err != nil { + b.Fatalf("NewProxy() error: %v", err) + } + + b.ReportAllocs() + b.ResetTimer() + + // 高并发：GOMAXPROCS 个 goroutine 同时请求 + b.RunParallel(func(pb *testing.PB) { + for pb.Next() { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + ctx.Request.Header.SetMethod(fasthttp.MethodGet) + p.ServeHTTP(ctx) + } + }) +} + +// BenchmarkConnectionPool_SmallBody 测试小响应体连接池效率。 +// +// 小响应体更依赖连接池复用。 +func BenchmarkConnectionPool_SmallBody(b *testing.B) { + addr, cleanup := setupInmemoryBackend([]byte("x")) + defer cleanup() + + cfg := &config.ProxyConfig{ + Path: "/api", + LoadBalance: "round_robin", + Timeout: config.ProxyTimeout{ + Connect: 5 * time.Second, + Read: 30 * time.Second, + Write: 30 * time.Second, + }, + } + + targets := []*loadbalance.Target{{URL: "http://" + addr}} + targets[0].Healthy.Store(true) + + p, err := NewProxy(cfg, targets, nil, nil) + if err != nil { + b.Fatalf("NewProxy() error: %v", err) + } + + // 预热 + for i := 0; i < 5; i++ { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + p.ServeHTTP(ctx) + } + + b.ReportAllocs() + b.ResetTimer() + + for b.Loop() { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + ctx.Request.Header.SetMethod(fasthttp.MethodGet) + p.ServeHTTP(ctx) + } +} + +// BenchmarkConnectionPool_LargeBody 测试大响应体连接池效率。 +// +// 大响应体需要更多缓冲区管理。 +func BenchmarkConnectionPool_LargeBody(b *testing.B) { + largeBody := make([]byte, 50*1024) // 50KB + for i := range largeBody { + largeBody[i] = byte('A' + i%26) + } + + addr, cleanup := setupInmemoryBackend(largeBody) + defer cleanup() + + cfg := &config.ProxyConfig{ + Path: "/api", + LoadBalance: "round_robin", + Timeout: config.ProxyTimeout{ + Connect: 5 * time.Second, + Read: 30 * time.Second, + Write: 30 * time.Second, + }, + } + + targets := []*loadbalance.Target{{URL: "http://" + addr}} + targets[0].Healthy.Store(true) + + p, err := NewProxy(cfg, targets, nil, nil) + if err != nil { + b.Fatalf("NewProxy() error: %v", err) + } + + b.ReportAllocs() + b.ResetTimer() + + for b.Loop() { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + ctx.Request.Header.SetMethod(fasthttp.MethodGet) + p.ServeHTTP(ctx) + } +} + +// BenchmarkConnectionPool_MultiTarget 测试多目标连接池。 +// +// 每个目标有独立连接池。 +func BenchmarkConnectionPool_MultiTarget(b *testing.B) { + targets := make([]*loadbalance.Target, 3) + cleanups := make([]func(), 3) + + for i := 0; i < 3; i++ { + addr, cleanup := setupInmemoryBackend([]byte("backend" + strconv.Itoa(i))) + cleanups[i] = cleanup + targets[i] = &loadbalance.Target{ + URL: "http://" + addr, + Weight: 1, + } + targets[i].Healthy.Store(true) + } + + defer func() { + for _, c := range cleanups { + c() + } + }() + + cfg := &config.ProxyConfig{ + Path: "/api", + LoadBalance: "round_robin", + Timeout: config.ProxyTimeout{ + Connect: 5 * time.Second, + Read: 30 * time.Second, + Write: 30 * time.Second, + }, + } + + p, err := NewProxy(cfg, targets, nil, nil) + if err != nil { + b.Fatalf("NewProxy() error: %v", err) + } + + b.ReportAllocs() + b.ResetTimer() + + b.RunParallel(func(pb *testing.PB) { + for pb.Next() { + ctx := &fasthttp.RequestCtx{} + ctx.Request.SetRequestURI("/api/test") + ctx.Request.Header.SetMethod(fasthttp.MethodGet) + p.ServeHTTP(ctx) + } + }) +} + +// BenchmarkHostClient_AcquireRelease 测试 HostClient 对象池。 +// +// 验证 fasthttp.Request/Response 池化效果。 +func BenchmarkHostClient_AcquireRelease(b *testing.B) { + addr, cleanup := setupInmemoryBackend([]byte("pool test")) + defer cleanup() + + timeout := config.ProxyTimeout{ + Connect: 5 * time.Second, + Read: 30 * time.Second, + Write: 30 * time.Second, + } + + client := createHostClient("http://"+addr, timeout, nil, nil, "", nil) + + b.ReportAllocs() + b.ResetTimer() + + for b.Loop() { + req := fasthttp.AcquireRequest() + resp := fasthttp.AcquireResponse() + + req.SetRequestURI("http://" + addr + "/api/test") + req.Header.SetMethod(fasthttp.MethodGet) + + _ = client.Do(req, resp) + + fasthttp.ReleaseRequest(req) + fasthttp.ReleaseResponse(resp) + } +} \ No newline at end of file