登录
首页 >  数据库 >  Redis

Redis实现令牌桶限流算法,核心在于利用其数据结构和原子操作来控制请求的频率。以下是实现的基本思路和步骤:一、令牌桶算法简介令牌桶算法是一种常见的限流算法,它通过一个“令牌桶”来管理请求的速率。系统以固定速率向桶中添加令牌,每个请求需要消耗一定数量的令牌才能被处理。如果桶中没有足够的令牌,则请求会被拒绝或等待。桶容量:令牌桶的最大容量(例如每秒最多允许100个请求)。令牌生成速率:单位时间内生成

时间:2026-03-30 21:09:28 377浏览 收藏

本文深入解析了如何利用Redis高效、可靠地实现令牌桶限流算法,强调必须通过Lua脚本(EVAL)保障“检查-补充-消耗-更新过期时间”全流程的原子性,避免INCR+EXPIRE非原子操作引发的计数器残留风险;指出关键细节——必须使用PTTL和TIME(而非客户端时间)实现毫秒级精准令牌补充,防止时钟漂移导致策略失效;同时介绍了Redis 6.2+提供的CL.THROTTLE命令这一更简洁可靠的原生方案,并提醒其依赖redis-cell模块及单位、兼容性等实战注意事项,为高并发场景下的限流落地提供了兼具严谨性与可操作性的技术指南。

Redis怎样实现基于令牌桶的限流算法

INCR + EXPIRE 模拟令牌桶,但要注意原子性问题

Redis 本身没有原生令牌桶命令,得靠多个命令组合实现。最常见写法是先 INCR 计数器,再 EXPIRE 设置过期时间——但这不是原子操作,高并发下可能漏设过期,导致计数器永久残留。

正确做法是用 EVAL 执行 Lua 脚本,把“增、判、设过期”锁在一次请求里。比如:当前令牌数不足时,不增加;足够则减一并保持 TTL 不变。

  • 脚本里用 redis.call("INCR", KEYS[1]) 获取新值,再用 redis.call("TTL", KEYS[1]) 查剩余时间
  • 首次访问需同时设初始值和 TTL,推荐用 SET key 1 EX 60 NX(注意 NX 防覆盖)
  • 别依赖客户端时间判断窗口,所有逻辑必须在 Redis 端完成,否则时钟不同步会出错

PTTLTTL 的区别影响重置逻辑

令牌桶需要“定期补充”,但 Redis 不支持定时回调。实际做法是每次请求时,根据上次更新时间和当前时间差,算出应补多少令牌。这时必须用 PTTL(毫秒级),而不是 TTL(秒级),否则在 1 秒内多次请求会误判为“没过期”,导致补发失败。

例如限流窗口 100ms,补速率 10 token/s,那每 100ms 补 1 个。如果只查 TTL,两次请求间隔 50ms 但落在同一秒内,TTL 返回值不变,就无法触发补充。

  • PTTL 获取精确剩余毫秒数,再结合 TIME 或客户端传入时间戳推算流逝时间
  • 补令牌后要重设 PEXPIRE,不能只改 value,否则 TTL 会归零或错乱
  • 注意 Lua 脚本里 TIME 返回的是秒+微秒数组,要自己转成毫秒做计算

CL.THROTTLE(Redis 6.2+)更省事,但得确认服务端版本

Redis 6.2 加了 CL.THROTTLE 命令,底层就是令牌桶,直接返回 [是否允许, 当前余量, 下次允许时间戳, 总容量, 重试等待毫秒] 五个值。不用手写 Lua,也不用担心竞态。

但它不是默认开启的,有些云厂商 Redis 实例(比如阿里云早期版本)没开 redis-cell 模块,调用会报 (error) ERR unknown command `CL.THROTTLE`

  • 先执行 MODULE LIST 看输出里有没有 name:redis-cell
  • 参数顺序是 CL.THROTTLE key max_burst count per_msec,注意单位是毫秒,不是秒
  • 返回数组第 4 项是“被拒绝后还要等多少毫秒”,可直接塞给客户端做 Retry-After

客户端时间不准会导致令牌计算漂移

很多实现让客户端传“上次访问时间戳”进来,由 Lua 脚本算间隔。这很危险——手机 App、浏览器、NTP 不稳的服务器,时间可能快慢几秒。结果就是同一用户在不同设备上触发限流策略不一致,甚至出现“明明刚用过,却说还能用”的情况。

真正可靠的做法是:只存最后一次成功操作的 Redis 服务端时间(用 TIME 命令获取),所有时间差都在 Lua 里用两次 TIME 结果相减得出。

  • 别在客户端算“now - last_time”,而是在 Lua 里做 current_time - stored_time
  • 存储的时间戳必须是 TIME 返回的整数对(秒+微秒),不要转成字符串或 JS Date
  • 如果必须兼容老版本 Redis(LPUSH+LLEN+EXPIRE),也别信客户端时间

事情说清了就结束

今天带大家了解了的相关知识,希望对你有所帮助;关于数据库的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

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