From d2e72b25be128484133a5e782289cf9e195c162f Mon Sep 17 00:00:00 2001
From: xfy <i@rua.plus>
Date: Thu, 16 Apr 2026 11:44:31 +0800
Subject: [PATCH] =?UTF-8?q?perf(lua):=20=E6=B7=BB=E5=8A=A0=20LuaContext=20?=
 =?UTF-8?q?=E5=92=8C=E5=8D=8F=E7=A8=8B=E6=B1=A0=E5=A4=8D=E7=94=A8=E6=95=88?=
 =?UTF-8?q?=E6=9E=9C=E9=AA=8C=E8=AF=81=E5=9F=BA=E5=87=86=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

添加 BenchmarkLuaContextPoolReuse 测试完整请求生命周期下的池复用率，
添加 BenchmarkLuaCoroutinePoolThroughput 测试高频率协程创建销毁的吞吐量。

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 internal/lua/lua_bench_test.go | 80 ++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)

diff --git a/internal/lua/lua_bench_test.go b/internal/lua/lua_bench_test.go
index a91f59a..54dd750 100644
--- a/internal/lua/lua_bench_test.go
+++ b/internal/lua/lua_bench_test.go
@@ -194,3 +194,83 @@ func BenchmarkTimerGracefulShutdown(b *testing.B) {
 		engine.Close()
 	}
 }
+
+// BenchmarkLuaContextPoolReuse 测试 LuaContext 池复用率。
+// 验证多次获取/释放后池能否有效复用对象，减少分配。
+func BenchmarkLuaContextPoolReuse(b *testing.B) {
+	engine, err := NewEngine(DefaultConfig())
+	if err != nil {
+		b.Fatal(err)
+	}
+	defer engine.Close()
+
+	b.ReportAllocs()
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		// 模拟一次完整的请求生命周期
+		ctx := NewContext(engine, nil)
+		ctx.SetVariable("uri", "/test")
+		ctx.SetVariable("method", "GET")
+		ctx.SetPhase(PhaseContent)
+		ctx.Write([]byte("response body"))
+		ctx.FlushOutput()
+		ctx.Release()
+	}
+}
+
+// BenchmarkLuaCoroutinePoolThroughput 测试协程池吞吐量。
+// 验证协程池在高频率创建/销毁场景下的复用效果。
+func BenchmarkLuaCoroutinePoolThroughput(b *testing.B) {
+	engine, err := NewEngine(DefaultConfig())
+	if err != nil {
+		b.Fatal(err)
+	}
+	defer engine.Close()
+
+	b.ReportAllocs()
+	b.ResetTimer()
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			coro, err := engine.NewCoroutine(nil)
+			if err != nil {
+				continue
+			}
+			coro.Close()
+		}
+	})
+}
+
+// BenchmarkLuaTablePool 测试 Lua table 对象池性能。
+// 验证 table 创建和池复用在频繁 table 操作场景下的表现。
+func BenchmarkLuaTablePool(b *testing.B) {
+	engine, err := NewEngine(DefaultConfig())
+	if err != nil {
+		b.Fatal(err)
+	}
+	defer engine.Close()
+
+	L := engine.L
+
+	b.Run("NewTable_NoPool", func(b *testing.B) {
+		b.ReportAllocs()
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			t := L.NewTable()
+			t.RawSetString("key1", glua.LString("value1"))
+			t.RawSetString("key2", glua.LString("value2"))
+			t.RawSetString("key3", glua.LNumber(42))
+		}
+	})
+
+	b.Run("SharedDict_AsPool", func(b *testing.B) {
+		// 共享字典底层使用对象池存储条目
+		dict := engine.CreateSharedDict("bench_pool", 1000)
+		b.ReportAllocs()
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			dict.Set("key", "value_with_pool", 0)
+			dict.Get("key")
+			dict.Delete("key")
+		}
+	})
+}