Lua on Ricky

Go实战指南：使用 go-redis 执行 Lua 脚本

Tue, 22 Jul 2025 16:56:00 +0800

Go 实战指南：使用 go-redis 执行 Lua 脚本

Redis command Eval
Redis package redis.NewScript, script.Run

openresty+redis 攔截高頻訪問 IP

Thu, 27 Jun 2019 11:18:46 +0800

openresty+redis 攔截高頻訪問 IP

init_by_lua_block {
 redis = require "redis"
 client = redis.connect('127.0.0.1', 6379)
}
server {
 listen 8080;
 location / {
 access_by_lua_file /usr/local/nginx/conf/lua/block.lua;
 proxy_pass http://192.168.1.102:8000;
 }
}

-- Redis-based IP rate limiting / blocking for OpenResty (ngx_lua)

-- NOTE:
-- This script assumes a global `client` variable is used/stored.
-- Make sure `redis` module is available and `client` is initialized somewhere.

local function isConnected()
 return client:ping()
end

local function createRedisConnection()
 return redis.connect("127.0.0.1", 6379)
end

-- 如果發生 redis 連線失敗，將停止攔截（直接放行）
if pcall(isConnected) then
 -- already connected (or ping succeeded)
else
 -- not connected; try reconnect
 if pcall(createRedisConnection) then
 -- 斷開重連：會導致每次訪問都需要重連 redis
 -- 訪問量大時建議：關閉重連邏輯（pcall 不執行），直接 ngx.exit 放行/終止
 client = createRedisConnection()
 else
 ngx.exit(ngx.OK)
 end
end

local ttl = 60 -- 監測週期（秒）
local bktimes = 30 -- 在監測週期內達到觸發攔截的訪問量
local block_ttl = 600 -- 觸發攔截後攔截時間（秒）

local ip = ngx.var.remote_addr
local ipvtimes = client:get(ip)

if ipvtimes then
 if ipvtimes == "-1" then
 -- blocked
 return ngx.exit(403)
 else
 local last_ttl = client:ttl(ip)
 -- ngx.say("key exist.ttl is ", last_ttl)

 if last_ttl == -1 then
 client:set(ip, 0)
 client:expire(ip, ttl)
 -- ngx.say("ttl & vtimes recount")
 return ngx.exit(ngx.OK)
 end

 local vtimes = tonumber(client:get(ip)) + 1

 if vtimes < bktimes then
 client:set(ip, vtimes)
 client:expire(ip, last_ttl)
 -- ngx.say(ip, " view ", vtimes, " times")
 return ngx.exit(ngx.OK)
 else
 -- ngx.say(ip, " will be block next time.")
 client:set(ip, -1)
 client:expire(ip, block_ttl)
 return ngx.exit(ngx.OK)
 end
 end
else
 -- key does not exist
 client:set(ip, 1)
 -- ngx.say(ip, " view 1 times")
 client:expire(ip, ttl)
 return ngx.exit(ngx.OK)
end

Nginx 访问日志中记录毫秒级别的时间精度

Tue, 24 Jul 2018 18:31:42 +0800

Nginx 的 access log 可以记录毫秒级的时间戳，但是是以 EPOCH 开始的毫秒数，比如 1503544071.865, 另一个变量 $time_local 记录的是秒级别的时间格式，比如 24/Aug/2017:11:07:51 +0800，在业务量大的时候，我们需要记录毫秒精度的时间格式，比如 24/Aug/2017:11:07:51.865 +0800, 这个可以通过 Lua 实现。

首先需要在 nginx.conf 中定义一个变量，叫做 time_millis，并初始化为空。类似于使用 auto-ssl 获取证书的时候需要指定一个 fallback 的自签证书。

 map $host $time_millis {
 default '';
 }

如果不定义这个变量，在 log_format 中引用的时候会报错。

使用 Lua 获取毫秒数，并追加到 $time_local 的末尾。

 log_by_lua_block {
 millis = string.gsub(ngx.var.msec, "(%d+).(%d+)", "%2")
 ngx.var.time_millis = string.gsub(ngx.var.time_local, "(.+) (.+)", "%1." .. millis .. " %2")
 }

在 log_format 中将 $time_local 改为 $time_millis

http {
 map $host $time_millis {
 default '';
 }

 log_by_lua_block {
 millis = string.gsub(ngx.var.msec, "(%d+).(%d+)", "%2")
 ngx.var.time_millis = string.gsub(ngx.var.time_local, "(.+) (.+)", "%1." .. millis .. " %2")
 }

 log_format main '$remote_addr - $remote_user [$time_millis] "$request" '
 '$status $body_bytes_sent "$http_referer" '
 '"$http_user_agent" $msec "$http_x_forwarded_for" $host $request_time $upstream_response_time $scheme "$request_body"';
}