Nginx中使用Lua脚本配置示例

Nginx是一个高性能的Web服务器，可以通过Lua脚本语言扩展功能。下面我们将展示如何在Nginx中使用Lua脚本进行配置，并提供两个实例。

安装Nginx和Lua模块

在开始之前，需要先安装Nginx和Lua模块。可以从Nginx的官方网站上下载Nginx软件包（https://nginx.org/en/download.html），然后在编译时添加--with-http_lua_module选项来启用Lua模块。

Lua脚本文件的编写

我们需要在Nginx配置文件中指定Lua脚本文件的位置，例如：

location /lua {
  default_type text/plain;
  content_by_lua_file /path/to/lua_script.lua;
}

这里将Lua脚本文件存放在/path/to/lua_script.lua文件中，在location指令中使用content_by_lua_file指定Lua脚本文件的位置。

将Lua脚本文件的内容写入到lua_script.lua中，例如：

ngx.say("Hello, Lua!");

这里使用ngx库的say()函数将“Hello, Lua!”输出到客户端。

示例一：根据客户端IP地址重定向

假设需要在Nginx中根据客户端IP地址将请求重定向到不同的服务器。可以编写一个Lua脚本来实现这个功能。

在Nginx配置文件中添加如下代码：

lua_shared_dict ip_list 10m;

server {
  listen 80;
  server_name example.com;

  set $backend "http://backend1.com";

  location / {
    if ($remote_addr != "127.0.0.1") {
      set $backend "http://backend2.com";
    }

    proxy_pass $backend;
  }

  location /update {
    content_by_lua_block {
      -- 读取ip_list字典内容
      local ips = ngx.shared.ip_list:get("ips")

      -- IP列表为空时，读取配置文件中的IP列表
      if not ips then
        local file = io.open("/path/to/ip_list.txt", "r")
        ips = file:read("*all")
        ngx.shared.ip_list:set("ips", ips)
        file:close()
      end

      -- 将IP地址转换为Set存储
      local ip_set = {}
      for ip in string.gmatch(ips, "[%w%.]+") do
        ip_set[ip] = true
      end

      -- 获取客户端IP地址
      local client_ip = ngx.var.remote_addr

      -- 检查客户端IP是否在白名单中
      if ip_set[client_ip] then
        ngx.say("IP " .. client_ip .. " is already whitelisted.")
      else
        -- 将客户端IP地址添加到白名单中，并将新列表写入字典
        table.insert(ip_set, client_ip)
        ips = table.concat(ip_set, ",")
        ngx.shared.ip_list:set("ips", ips)

        ngx.say("IP " .. client_ip .. " has been whitelisted.")
      end
    }
  }
}

这里使用lua_shared_dict存储IP列表，提高性能。当客户端请求/update时，Lua脚本将读取IP列表，将客户端IP添加到白名单中，并将更新后的IP列表写入共享字典中。

示例二：使用Lua扩展Nginx的Access Log

假设需要将每个客户端访问的URL和状态码记录到Access Log中。可以编写一个Lua脚本来实现这个功能。

在Nginx配置文件中添加如下代码：

http {
  log_format lua '$remote_addr - $remote_user [$time_local] '
                  '"$request" $status $body_bytes_sent '
                  '"$http_referer" "$http_user_agent"';

  server {
    listen 80;
    server_name example.com;

    location / {
      access_by_lua_block {
        -- 记录Access Log
        ngx.log(ngx.INFO, ngx.var.remote_addr, " - ", ngx.var.remote_user, " [", ngx.var.time_local, "] \"", ngx.var.request, "\" ", ngx.var.status, " ", ngx.var.body_bytes_sent, " \"", ngx.var.http_referer, "\" \"", ngx.var.http_user_agent, "\"")
      }

      root /path/to/public_html;
    }

    location /admin {
      deny all;
    }
  }
}

这里使用log_format指定Access Log的格式为lua，并在location指令中使用access_by_lua_block指定Lua脚本，在该脚本中使用ngx库的log()函数记录Access Log。

通过以上两个示例，我们可以看到，使用Lua脚本可以轻松扩展Nginx的功能，实现更多定制化的需求。