登录
首页 >  Golang >  Go教程

GolangRedis滑动窗口限流实现教程

时间:2026-03-21 19:00:42 206浏览 收藏

本文深入剖析了在Golang中基于Redis实现滑动窗口限流的关键难点与最佳实践,直击INCR+EXPIRE非原子操作引发的key永久残留、窗口错乱、多实例时间不同步等生产级陷阱;强调必须通过Lua脚本原子执行计数、过期设置与过期数据清理,并统一采用Redis服务端时间戳(而非本地time.Now)确保窗口逻辑严谨可靠;同时对比ZSET与List方案的适用场景,给出精简可用的Lua脚本范例及Go调用要点,涵盖NTP时钟同步、内存防膨胀、集群主节点约束等实战细节,为高并发系统提供稳定、精准、可落地的限流保障。

如何在Golang中实现基于Redis的滑动窗口限流 Go语言分布式限流算法

滑动窗口限流为什么不能只用 Redis INCR + EXPIRE

因为 INCREXPIRE 是两个独立命令,中间可能被中断(如 Redis 连接断开、服务崩溃),导致 key 永久存在或永久丢失,窗口边界错乱。真实场景下,必须保证「计数 + 过期时间设置」原子性。

  • 推荐用 EVAL 执行 Lua 脚本,Redis 保证脚本内操作原子执行
  • 不要在 Go 层做“先查再判再增”逻辑——竞态无法避免,尤其多实例部署时
  • 注意 Lua 脚本里不能用 redis.call("EXPIRE", ...) 对已存在的 key 重复设过期:Redis 6.2+ 才支持 EXPIRE 返回值判断,老版本得用 PEXPIREAT 算毫秒级绝对过期时间

Go 里调用 Lua 实现滑动窗口的最小可行写法

核心是把窗口切分成多个时间桶(比如 1 秒 1 桶),用一个有序集合(ZSET)存每个请求的时间戳,再用 ZREMRANGEBYSCORE 清理过期桶。但更轻量的做法是直接用 list + LTRIM,适合精度要求不高的场景(如 1 分钟窗口,每秒 1 桶)。

  • LPUSH + LTRIM 维护一个带时间戳的 list,长度限制为窗口总桶数 × 每桶最大请求数
  • 但 list 不支持按时间范围剔除,所以更稳的是 ZSET:member 用唯一 ID(如 req_id:timestamp),score 存 Unix 时间戳
  • Go 中用 redis.Client.Eval 调用脚本,传入 keywindow_secondsmax_requests 三个参数
  • 脚本返回 1 表示放行,0 表示拒绝;别忽略返回值类型转换:int64 而不是 bool
local key = KEYS[1]
local window = tonumber(ARGV[1])
local max = tonumber(ARGV[2])
local now = tonumber(ARGV[3])
local zset = redis.call("ZRANGEBYSCORE", key, 0, now - window)
local count = #zset
if count >= max then
  return 0
end
redis.call("ZADD", key, now, ARGV[4])
redis.call("EXPIRE", key, window + 1)
return 1

多实例部署时,时间不同步会让滑动窗口失效

如果各机器系统时间差超过窗口粒度(比如窗口是 1 秒,但 A 机比 B 机快 1.2 秒),B 机认为刚进来的请求已过期,A 机却还在计数,结果就是实际 QPS 超出预期。

  • 强制所有服务同步 NTP,误差控制在 100ms 内(ntpq -p 检查 offset)
  • 不要用本地 time.Now().Unix() 当 score,改用 Redis 的 TIME 命令返回服务端时间(Lua 里用 redis.call("TIME") 获取)
  • 如果业务对精度要求不高,可降级为固定窗口(INCR + EXPIRE),但要注意它会放大峰值流量(窗口切换瞬间允许双倍请求)

Redis 内存增长不可控?得清理过期 zset 成员

ZADD 不会自动删旧数据,靠脚本里的 ZRANGEBYSCORE 只读不删,长期运行会导致 key 膨胀。必须主动清理。

  • 在 Lua 脚本末尾加 redis.call("ZREMRANGEBYSCORE", key, 0, now - window),和判断逻辑共用同一时间点
  • 别用 ZREMRANGEBYRANK —— rank 不代表时间顺序,zset 按 score 排序,rank 是位置索引
  • 如果窗口很长(如 24 小时)、QPS 很高,单次 ZREMRANGEBYSCORE 可能阻塞 Redis,考虑分批删(但 Lua 不支持循环 sleep,得在 Go 层拆成多次 EVAL)

时间戳必须统一来源,窗口逻辑才可靠。哪怕 Redis 集群有从节点,也只跟主节点交互,别让客户端自己拼时间。

好了,本文到此结束,带大家了解了《GolangRedis滑动窗口限流实现教程》,希望本文对你有所帮助!关注golang学习网公众号,给大家分享更多Golang知识!

资料下载
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>