Nginx配置中如何巧妙运用Lua脚本？（Lua脚本在Nginx中的使用教程）

赵其鑫 • 2024年 11月 1日 16:25:19 • 中间件

Nginx 配置中巧用 Lua 脚本：实战指南

Nginx 的 LuaJIT 集成让开发者能够在 Nginx 配置文件内直接使用 Lua 脚本，极大地提升了 Nginx 的灵活性和功能性。下面是如何在 Nginx 中配置和使用 Lua 脚本的一些实践指导。

准备阶段

首先，确保你的 Nginx 包含 Lua 模块。可以通过重新编译 Nginx 加上 LuaJIT 模块，或者使用预先打包好的版本，如 OpenResty。

基础语法

在 Nginx 配置文件中嵌入 Lua 脚本，主要有两种方式：

content_by_lua_block 和 rewrite_by_lua*:
- content_by_lua_block 用于处理 HTTP 响应体。
- rewrite_by_lua*, access_by_lua*, 等用于不同阶段的数据处理。

location / {
    content_by_lua_block {
        ngx.say("Hello from Lua!");
    }
}

实战案例

让我们通过几个具体的例子来看看 Lua 脚本在 Nginx 中的强大功能。

1. 数据转换

使用 Lua 进行 JSON 格式的转换，比如：

location /json {
    content_by_lua_block {
        local json = require "cjson"
        local data = { name="John", age=30 }
        ngx.say(json.encode(data))
    }
}

2. 限流

实现基本的限流逻辑：

location /limited {
    access_by_lua_block {
        local limit_count = tonumber(ngx.shared.limit:get("counter") or 0)
        if limit_count >= 10 then
            ngx.exit(ngx.HTTP_TOO_MANY_REQUESTS)
        end
        ngx.shared.limit:set("counter", limit_count + 1, 10)
    }
}

3. 复杂的 URL 重写

利用 Lua 实现更复杂的 URL 重写规则：

rewrite_by_lua_block {
    if ngx.re.match(ngx.var.request_uri, "^/(.+)$") then
        ngx.redirect("/api/" .. ngx.re.capt[1])
    end
}

4. 请求过滤

拒绝特定类型的请求：

access_by_lua_block {
    if ngx.var.http_user_agent == 'BadBot' then
        ngx.return(ngx.HTTP_FORBIDDEN)
    end
}

5. 个性化响应

根据条件生成不同的响应：

content_by_lua_block {
    local user_agent = ngx.var.http_user_agent
    if user_agent and user_agent:find('Mobile') then
        ngx.header['Content-Type'] = 'text/html'
        ngx.print("<html><body>Hello Mobile User!</body></html>")
    else
        ngx.print("Hello Desktop User!")
    end
}

6. 日志分析

在响应前或后插入日志信息：

log_by_lua_block {
    ngx.exec("echo Request processed")
}

7. 高级数据处理

整合外部数据库或 REST API 调用：

local resty_core = require "resty.core"
local http = require "resty.http"

content_by_lua_block {
    local cli = http.new()
    local resp, err = cli:request_uri("https://api.example.com/data", {method = "GET"})
    if not err then
        ngx.print(resp.body)
    end
}

小结

通过 Lua 脚本，Nginx 变得异常强大，可以处理复杂的业务逻辑而不牺牲性能。上述只是冰山一角，Lua 脚本在 Nginx 中的应用潜力无限。掌握了这些基础知识，就可以开始在实际项目中探索更多的可能性了！