Go语言Redis如何做分布式限流_Golang Lua脚本滑动窗口实现-Golang-PHP中文网

不能直接用incr+expire做滑动窗口，因为二者非原子操作，incr成功但expire失败会导致key永久存在，并发下易漏计数或重复计数；滑动窗口需动态维护时间范围内请求记录，必须用lua脚本封装清理、添加、统计三步并借助zset和redis.call("time")保证原子性与时间一致性。

go语言redis如何做分布式限流_golang lua脚本滑动窗口实现

为什么不能直接用 `INCR` + `EXPIRE` 做滑动窗口

因为这两条命令不是原子的，INCR 成功但 EXPIRE 失败时，key 会永久存在；并发场景下还可能漏计数或重复计数。Redis 6.0+ 虽支持 INCRBYEX，但它只解决“带过期的递增”，不支持按时间窗口剔除旧请求——滑动窗口需要动态维护一个时间范围内的请求记录，必须靠 Lua 脚本保证原子性。

`EVAL` 执行 Lua 脚本实现滑动窗口限流的核心逻辑

脚本要完成三件事：清理过期时间戳、添加当前请求时间、判断窗口内请求数是否超限。关键点在于用 ZSET 存储时间戳（score = 时间戳，member = 请求唯一标识，如 UUID 或毫秒级时间戳），利用 ZREMRANGEBYSCORE 清理过期项，再用 ZCARD 统计剩余数量。

脚本入参：KEYS[1] 是限流 key（如 "rate:login:192.168.1.1"），ARGV[1] 是窗口大小（单位：秒），ARGV[2] 是最大请求数
当前时间必须用 redis.call("TIME") 获取服务端时间，避免客户端时钟偏差
插入 member 时推荐用 redis.call("ZADD", KEYS[1], now, now)（用时间戳作 member），省去生成 UUID 的开销，也方便调试
注意 ZREMRANGEBYSCORE 的边界：应使用 (now - window)（左开区间），防止刚好卡在边界的时间点被误删

local key = KEYS[1]
local window = tonumber(ARGV[1])
local max = tonumber(ARGV[2])
local now = tonumber(redis.call("TIME")[1])
<p>redis.call("ZREMRANGEBYSCORE", key, 0, "(" .. (now - window))
redis.call("ZADD", key, now, now)
redis.call("EXPIRE", key, window + 1)</p><p>local count = redis.call("ZCARD", key)
if count > max then
return 0
else
return 1
end</p>

Go 客户端调用 `EVAL` 时的常见坑

用 github.com/go-redis/redis/v9 时，别直接拼接 Lua 脚本字符串传给 Eval 方法——脚本内容变更后 Redis 不会自动重新加载，且无法复用 SHA1 缓存。正确做法是先用 ScriptLoad 加载，后续用 EvalSha 执行。

Golang Stay Updated

提供最新的 Golang 新闻、社群和值得关注的人。适用于寻找 Go 学习资源、发现新库等。

下载

ScriptLoad 返回的 SHA1 值需缓存（全局变量或 sync.Once），否则每次调用都发一次 SCRIPT LOAD 请求，增加延迟
如果脚本里用了 redis.call("TIME")，就不能用 EVALSHA 在只读副本上执行（Redis 禁止在 replica 上调用 TIME），生产环境务必确保连接的是 master
Go 中传参要用 redis.Args{}.Add(...).Add(...) 或切片，别漏掉 KEYS 数量（哪怕只有 1 个 key）——Eval 第三个参数是 []string，第一个元素是 key，后面才是 ARGV
返回值是 int64，需显式转为 bool 判断：allowed, _ := ret.Val().(int64); if allowed == 1 { ... }