feat(lua): 实现 ngx.ctx API
提供请求级上下文存储，每请求独立的 Lua table Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
parent
797c4b0a26
commit
d021b0e9fd
60
internal/lua/api_ctx.go
Normal file
60
internal/lua/api_ctx.go
Normal file
@ -0,0 +1,60 @@
|
||||
// Package lua 提供 ngx.ctx API 实现
|
||||
package lua
|
||||
|
||||
import (
|
||||
glua "github.com/yuin/gopher-lua"
|
||||
)
|
||||
|
||||
// ngxCtxAPI ngx.ctx API 实现
|
||||
type ngxCtxAPI struct {
|
||||
// 每个请求独立的 ctx 表
|
||||
// 存储在协程的全局变量中
|
||||
ctxTable *glua.LTable
|
||||
}
|
||||
|
||||
// newNgxCtxAPI 创建 ngx.ctx API 实例
|
||||
func newNgxCtxAPI() *ngxCtxAPI {
|
||||
return &ngxCtxAPI{}
|
||||
}
|
||||
|
||||
// RegisterNgxCtxAPI 在 Lua 状态机中注册 ngx.ctx API
|
||||
// ngx.ctx 是一个普通的 Lua table，每请求独立，支持任意 Lua 值类型
|
||||
func RegisterNgxCtxAPI(L *glua.LState, ngxTable *glua.LTable) {
|
||||
// 创建请求级的 ctx table
|
||||
ctxTable := L.NewTable()
|
||||
|
||||
// 将 ngx.ctx 添加到 ngx 表
|
||||
ngxTable.RawSetString("ctx", ctxTable)
|
||||
}
|
||||
|
||||
// GetCtxTable 获取 ctx table（用于内部访问）
|
||||
func (api *ngxCtxAPI) GetCtxTable(L *glua.LState) *glua.LTable {
|
||||
ngx := L.GetGlobal("ngx")
|
||||
if ngx == glua.LNil {
|
||||
return nil
|
||||
}
|
||||
if ngxTable, ok := ngx.(*glua.LTable); ok {
|
||||
ctx := ngxTable.RawGetString("ctx")
|
||||
if ctxTable, ok := ctx.(*glua.LTable); ok {
|
||||
return ctxTable
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// SetValue 在 Go 层设置 ctx 值
|
||||
func (api *ngxCtxAPI) SetValue(L *glua.LState, key string, value glua.LValue) {
|
||||
tb := api.GetCtxTable(L)
|
||||
if tb != nil {
|
||||
tb.RawSetString(key, value)
|
||||
}
|
||||
}
|
||||
|
||||
// GetValue 在 Go 层获取 ctx 值
|
||||
func (api *ngxCtxAPI) GetValue(L *glua.LState, key string) glua.LValue {
|
||||
tb := api.GetCtxTable(L)
|
||||
if tb != nil {
|
||||
return tb.RawGetString(key)
|
||||
}
|
||||
return glua.LNil
|
||||
}
|
||||
367
internal/lua/api_ctx_test.go
Normal file
367
internal/lua/api_ctx_test.go
Normal file
@ -0,0 +1,367 @@
|
||||
// Package lua 提供 ngx.ctx API 测试
|
||||
package lua
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/valyala/fasthttp"
|
||||
)
|
||||
|
||||
// TestNgxCtxAPI 测试 ngx.ctx API 基础功能
|
||||
func TestNgxCtxAPI(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试 ngx.ctx 存在且是一个 table
|
||||
err = coro.Execute(`
|
||||
if type(ngx) ~= "table" then
|
||||
error("ngx is not a table")
|
||||
end
|
||||
if type(ngx.ctx) ~= "table" then
|
||||
error("ngx.ctx is not a table")
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxStringValue 测试存储字符串值
|
||||
func TestNgxCtxStringValue(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试字符串存储
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.message = "hello world"
|
||||
local msg = ngx.ctx.message
|
||||
if msg ~= "hello world" then
|
||||
error("string value mismatch: " .. tostring(msg))
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxNumberValue 测试存储数字值
|
||||
func TestNgxCtxNumberValue(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试数字存储
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.count = 42
|
||||
ngx.ctx.pi = 3.14159
|
||||
|
||||
local count = ngx.ctx.count
|
||||
local pi = ngx.ctx.pi
|
||||
|
||||
if count ~= 42 then
|
||||
error("count should be 42, got: " .. tostring(count))
|
||||
end
|
||||
|
||||
if math.abs(pi - 3.14159) > 0.00001 then
|
||||
error("pi should be 3.14159, got: " .. tostring(pi))
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxTableValue 测试存储 table 值
|
||||
func TestNgxCtxTableValue(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试 table 存储
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.data = {
|
||||
name = "test",
|
||||
items = {1, 2, 3, 4, 5}
|
||||
}
|
||||
|
||||
local data = ngx.ctx.data
|
||||
if type(data) ~= "table" then
|
||||
error("data should be a table")
|
||||
end
|
||||
|
||||
if data.name ~= "test" then
|
||||
error("data.name should be 'test'")
|
||||
end
|
||||
|
||||
if type(data.items) ~= "table" then
|
||||
error("data.items should be a table")
|
||||
end
|
||||
|
||||
if data.items[1] ~= 1 then
|
||||
error("data.items[1] should be 1")
|
||||
end
|
||||
|
||||
if data.items[5] ~= 5 then
|
||||
error("data.items[5] should be 5")
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxFunctionValue 测试存储函数值
|
||||
func TestNgxCtxFunctionValue(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试函数存储
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.handler = function(x)
|
||||
return x * 2
|
||||
end
|
||||
|
||||
local handler = ngx.ctx.handler
|
||||
if type(handler) ~= "function" then
|
||||
error("handler should be a function")
|
||||
end
|
||||
|
||||
local result = handler(21)
|
||||
if result ~= 42 then
|
||||
error("handler(21) should return 42, got: " .. tostring(result))
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxBooleanValue 测试存储布尔值
|
||||
func TestNgxCtxBooleanValue(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试布尔值存储
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.enabled = true
|
||||
ngx.ctx.disabled = false
|
||||
|
||||
if ngx.ctx.enabled ~= true then
|
||||
error("enabled should be true")
|
||||
end
|
||||
|
||||
if ngx.ctx.disabled ~= false then
|
||||
error("disabled should be false")
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxNilValue 测试存储和读取 nil 值
|
||||
func TestNgxCtxNilValue(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试 nil 值
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.nothing = nil
|
||||
local val = ngx.ctx.nothing
|
||||
if val ~= nil then
|
||||
error("nothing should be nil")
|
||||
end
|
||||
|
||||
-- 读取不存在的键
|
||||
local missing = ngx.ctx.missing_key
|
||||
if missing ~= nil then
|
||||
error("missing_key should be nil")
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxMultipleScripts 测试在同一个脚本中读写 ngx.ctx
|
||||
// 注意：协程在执行后变成 dead 状态，不能多次执行
|
||||
func TestNgxCtxMultipleScripts(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 在同一个脚本中设置和读取值（协程只能执行一次）
|
||||
err = coro.Execute(`
|
||||
-- 设置值
|
||||
ngx.ctx.shared_value = "shared"
|
||||
ngx.ctx.counter = 1
|
||||
|
||||
-- 读取并验证值
|
||||
local val = ngx.ctx.shared_value
|
||||
if val ~= "shared" then
|
||||
error("shared_value should be 'shared'")
|
||||
end
|
||||
|
||||
-- 修改值
|
||||
ngx.ctx.counter = ngx.ctx.counter + 1
|
||||
if ngx.ctx.counter ~= 2 then
|
||||
error("counter should be 2")
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
// TestNgxCtxNestedTable 测试嵌套 table
|
||||
func TestNgxCtxNestedTable(t *testing.T) {
|
||||
var req fasthttp.Request
|
||||
req.Header.SetMethod("GET")
|
||||
req.Header.SetRequestURI("/test")
|
||||
|
||||
ctx := &fasthttp.RequestCtx{}
|
||||
ctx.Init(&req, nil, nil)
|
||||
|
||||
engine, err := NewEngine(DefaultConfig())
|
||||
require.NoError(t, err)
|
||||
defer engine.Close()
|
||||
|
||||
coro, err := engine.NewCoroutine(ctx)
|
||||
require.NoError(t, err)
|
||||
defer coro.Close()
|
||||
|
||||
err = coro.SetupSandbox()
|
||||
require.NoError(t, err)
|
||||
|
||||
// 测试嵌套 table
|
||||
err = coro.Execute(`
|
||||
ngx.ctx.config = {
|
||||
database = {
|
||||
host = "localhost",
|
||||
port = 5432
|
||||
},
|
||||
cache = {
|
||||
ttl = 3600
|
||||
}
|
||||
}
|
||||
|
||||
local host = ngx.ctx.config.database.host
|
||||
if host ~= "localhost" then
|
||||
error("config.database.host should be 'localhost'")
|
||||
end
|
||||
|
||||
local port = ngx.ctx.config.database.port
|
||||
if port ~= 5432 then
|
||||
error("config.database.port should be 5432")
|
||||
end
|
||||
|
||||
ngx.ctx.config.database.port = 3306
|
||||
if ngx.ctx.config.database.port ~= 3306 then
|
||||
error("config.database.port should be updated to 3306")
|
||||
end
|
||||
`)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user