第7章 - OpenResty 入门

-- 获取请求参数
local args = ngx.req.get_uri_args()
local name = args.name or "World"

-- 设置响应头
ngx.header["Content-Type"] = "application/json"

-- 返回 JSON 响应
local cjson = require("cjson")
local response = {
    message = "Hello, " .. name .. "!",
    method = ngx.req.get_method(),
    uri = ngx.var.uri,
    time = ngx.now()
}

ngx.say(cjson.encode(response))

-- lua/rate_limit.lua
-- 基于令牌桶的接口限流

local limit_count = ngx.shared.limit_count    -- 需要在 nginx.conf 中声明共享内存

local function rate_limit(key, limit, window)
    local count, err = limit_count:get(key)
    
    if not count then
        -- 第一次请求
        limit_count:set(key, 1, window)
        return true
    end
    
    if count >= limit then
        return false    -- 超过限制
    end
    
    limit_count:incr(key, 1)
    return true
end

-- 获取客户端 IP
local client_ip = ngx.var.remote_addr
local key = "rate:" .. client_ip

-- 每秒最多 10 次请求
if not rate_limit(key, 10, 1) then
    ngx.status = 429
    ngx.say('{"error": "Too Many Requests"}')
    return ngx.exit(429)
end

-- 通过限流，继续处理...

-- lua/auth.lua
-- JWT Token 验证

local cjson = require("cjson")

-- 获取 Authorization 头
local auth_header = ngx.req.get_headers()["Authorization"]
if not auth_header then
    ngx.status = 401
    ngx.say('{"error": "Missing Authorization header"}')
    return ngx.exit(401)
end

-- 提取 Token
local token = auth_header:match("Bearer%s+(.+)")
if not token then
    ngx.status = 401
    ngx.say('{"error": "Invalid token format"}')
    return ngx.exit(401)
end

-- 简单的 Token 验证（实际项目中应该验证 JWT 签名）
-- 这里只是演示，生产环境请使用 lua-resty-jwt 库
local decoded = ngx.decode_base64(token)
if not decoded then
    ngx.status = 401
    ngx.say('{"error": "Invalid token"}')
    return ngx.exit(401)
end

-- 验证通过，将用户信息传给后端
ngx.req.set_header("X-User-Id", "extracted-user-id")

-- lua/gray_route.lua
-- 灰度发布路由

local gray_users = ngx.shared.gray_users    -- 灰度用户列表

local function is_gray_user(user_id)
    if gray_users:get(user_id) then
        return true
    end
    
    -- 也可以按百分比灰度
    local hash = ngx.crc32_long(user_id)
    local percent = hash % 100
    return percent < 10    -- 10% 的用户走灰度
end

-- 获取用户 ID（从 header 或 cookie）
local user_id = ngx.req.get_headers()["X-User-Id"] or "anonymous"

if is_gray_user(user_id) then
    -- 灰度用户 → 新版本服务
    ngx.var.backend = "http://service-v2:8080"
else
    -- 正常用户 → 稳定版本服务
    ngx.var.backend = "http://service-v1:8080"
end