Redis 与 Lua 脚本（Part 2）

Redis与Lua脚本Part2

这篇是 Redis 学习笔记 Part 2，重点放在 Lua 脚本能力和在 Redis/OpenResty 场景中的实际使用。

Lua 基础快速梳理

参考文档：The Programming Language Lua

最近需要阅读和编写一些 Lua 脚本，先把最关键的语法点集中整理一遍。

1. 变量与作用域

Lua 默认变量是全局变量，只有显式写 local 才是局部变量。

2. 数据类型

Lua 共有 8 种类型：

nil：空值。给变量赋 nil 可以理解为删除该键。
boolean：true / false。
number：Lua 5.1 中是双精度浮点数。
string：不可变字符串。
function：函数是一等公民。
userdata：保存 C 数据结构。
thread：协程。
table：Lua 唯一的复合数据结构。

Lua 的核心是 table。它既能当数组，也能当字典。

数组示例（注意索引从 1 开始）：

1
-- 隐式索引
2
local list = { "Misc", "Web", "Pwn" }
3

4
-- 显式索引
5
local list2 = {
6
  [1] = "Misc",
7
  [2] = "Web",
8
  [3] = "Pwn"
9
}
10

11
print(list[1]) -- Misc
12
print(list[0]) -- nil

长度获取：

1
local list = {
2
  [1] = "Chinese",
3
  [2] = "English",
4
  [3] = "Math"
5
}
6

7
print(#list) -- 3

字典示例：

1
local user = {
2
  name = "bx", -- 等价于 ["name"] = "bx"
3
  ["age"] = 18,
4
  [10001] = "id"
5
}
6

7
print(user.name)
8
print(user["name"])
9
print(user["age"])
10
print(user[10001])

遍历时常用 ipairs 和 pairs：

函数	适用对象	顺序	遇到 `nil` 的行为
`ipairs`	连续数字索引数组	从 1 递增	遇到第一个 `nil` 停止
`pairs`	全部键值对	不保证顺序	不因 `nil` 提前停止

ipairs 示例：

1
local t = {10, 20, 30}
2

3
for i, v in ipairs(t) do
4
  print(i, v)
5
end

pairs 示例：

1
local t = {
2
  name = "bx",
3
  age = 18
4
}
5

6
for k, v in pairs(t) do
7
  print(k, v)
8
end

3. 其他常用点

字符串拼接使用 ..，不是 +。

1
local key_prefix = "admin:"
2
local id = 1001
3

4
local key = key_prefix .. id
5
print("Key:", key)
6

7
local message = string.format("User ID: %d", id)
8
print("Message:", message)

Lua 没有 switch，只有 if / elseif / else。

1
local score = 0
2

3
-- 在 Lua 里，0 也是真值
4
if score then
5
  print("0 在 Lua 中为 true")
6
end
7

8
if score > 90 then
9
  print("A")
10
elseif score > 60 then
11
  print("B")
12
else
13
  print("C")
14
end

Lua 真值规则：只有 false 和 nil 为假，其他值（包括 0、""、{}）都为真。

元表（Metatable）

元表可以改写 table 的默认行为，常见元方法有：

__index：访问不存在字段时触发。
__newindex：给不存在字段赋值时触发。
__add / __sub：运算符重载。
__call：让 table 像函数一样被调用。
__tostring：自定义打印输出。

Redis 中的 Lua 脚本

1. 原子性

在 Redis 里执行 Lua 脚本时，Redis 会把整段脚本作为一个原子操作执行。脚本运行期间不会插入其他命令，因此非常适合实现“读-改-写”一体化逻辑。

2. `redis.call` 与 `redis.pcall`

方法	出错时行为
`redis.call`	抛异常并终止脚本
`redis.pcall`	返回错误对象，不直接中断

3. `EVAL` 与 `numkeys`

Python 端常见调用方式：

1
r.eval(script, numkeys, *keys_and_args)

参数拆分规则：

1
r.eval(script, numkeys, key1, key2, ..., arg1, arg2, ...)

numkeys 的意义是告诉 Redis：后续参数中前多少个属于 Key。

前 numkeys 个参数进入 KEYS[]
剩余参数进入 ARGV[]

所以写脚本时要明确区分 KEYS[] 与 ARGV[]。

4. 固定窗口限流示例

规则：用户 user:1001 在 10 秒内最多请求 3 次。

1
import redis
2
import time
3

4
r = redis.Redis(host='localhost', port=6379, decode_responses=True)
5

6

7
def example_rate_limiter():
8
    user_id = "user:1001"
9
    limit = 3
10
    window_seconds = 10
11

12
    rate_limit_script = """
13
    local key = KEYS[1]
14
    local limit = tonumber(ARGV[1])
15
    local window = tonumber(ARGV[2])
16

17
    local current = redis.call("INCR", key)
18

19
    if tonumber(current) == 1 then
20
        redis.call("EXPIRE", key, window)
21
    end
22

23
    if tonumber(current) > limit then
24
        return 0 -- Blocked
25
    else
26
        return 1 -- Allowed
27
    end
28
    """
29

30
    print(f"Simulating requests for {user_id} (limit={limit}, window={window_seconds}s)")
31

32
    for i in range(1, 6):
33
        allowed = r.eval(rate_limit_script, 1, f"ratelimit:{user_id}", limit, window_seconds)
34
        status = "Allowed" if allowed == 1 else "Blocked"
35
        print(f"Request {i}: {status}")
36
        time.sleep(0.5)
37

38

39
if __name__ == "__main__":
40
    example_rate_limiter()

输出示例：

🔍Simulating requests for user:1001 (limit=3, window=10s)
Request 1: Allowed
Request 2: Allowed
Request 3: Allowed
Request 4: Blocked
Request 5: Blocked

OpenResty 与 Lua

OpenResty 本质是 NGINX + LuaJIT。

它把 Lua 脚本能力嵌入 NGINX 请求处理链路，让我们可以在网关层做鉴权、限流、路由和安全逻辑。

小结

Lua 本身不复杂，但 table、真值规则、元表这三部分很关键。放到 Redis 场景后，核心是理解原子执行模型、参数分离（KEYS/ARGV）和错误处理方式（call/pcall）。

下一步可以继续扩展：滑动窗口限流、令牌桶限流，以及在 OpenResty 中落地成网关策略。