讲师中心 微信公众号
AI工具推荐 视频效率加速

如何在Redis中实现高并发限流器_利用Lua脚本编写令牌桶算法

梦静大大_1510

梦静大大_1510

发布时间:2026-06-27 06:48:37

|

772人浏览过

|

来源于php中文网

原创

必须用Lua脚本实现令牌桶,因其在Redis单线程中执行可保证“读-判-写”原子性,避免客户端多请求并发导致的竞态超卖;所有操作须封装于单次EVAL/EVALSHA调用中。

如何在redis中实现高并发限流器_利用lua脚本编写令牌桶算法

为什么必须用 Lua 脚本实现令牌桶,而不是客户端计算

Redis 单线程执行 Lua 脚本的原子性,是高并发下限流不超卖的核心保障。如果在客户端算好时间、查 key、判断、再 set,中间必然存在竞态——多个请求同时读到剩余令牌为 1,全部放行。Lua 在服务端一次性完成「读令牌数 + 判断 + 更新 + 返回结果」,彻底规避这个问题。

常见错误现象:INCR 配合 EXPIRE 手动模拟漏桶,但没加锁或没用 Lua,压测时 QPS 轻松突破阈值 2–3 倍。

  • 所有操作必须封装在一个 EVAL 或 EVALSHA 调用中
  • 不要在 Lua 里调用 redis.call("TIME") —— 它返回的是 Redis 服务器本地时间(秒级),精度不够;改用 redis.call("TIME") 返回两个整数(秒+微秒),或更推荐直接用 ARGV[3] 传入客户端毫秒时间戳(由调用方保证时钟一致)
  • 脚本内避免使用 for 循环做大量计算,会阻塞 Redis 主线程

令牌桶 Lua 脚本的关键参数与逻辑拆解

一个生产可用的令牌桶脚本至少要接收 4 个参数:KEYS[1](桶名)、ARGV[1](最大容量)、ARGV[2](每秒补充令牌数)、ARGV[3](当前毫秒时间戳)。脚本需完成:读旧状态 → 计算应新增令牌 → 截断上限 → 判断是否够用 → 更新并返回结果。

示例核心逻辑(精简版):

local key = KEYS[1]
local capacity = tonumber(ARGV[1])
local rate = tonumber(ARGV[2])
local now_ms = tonumber(ARGV[3])
<p>local bucket = redis.call("HMGET", key, "last_ms", "tokens")
local last_ms = tonumber(bucket[1]) or now_ms
local tokens = tonumber(bucket[2]) or capacity</p><p>-- 计算已过去时间(秒),换算应添加令牌数
local delta_sec = math.max(0, (now_ms - last_ms) / 1000.0)
local new_tokens = math.min(capacity, tokens + delta_sec * rate)</p><p>-- 判断是否允许请求
local allowed = (new_tokens >= 1) and 1 or 0
if allowed == 1 then
new_tokens = new_tokens - 1
end</p><p>-- 写回:更新令牌数和最后时间
redis.call("HMSET", key, "tokens", new_tokens, "last_ms", now_ms)
redis.call("PEXPIRE", key, 60000) -- 自动过期,避免冷 key 残留
return { allowed, math.floor(new_tokens) }</p>

注意:PEXPIRE 时间设为远大于预期桶生命周期(如 60 秒),不是“每秒重置”,而是“长期有效但有兜底”;否则频繁过期/重建 key 会引发性能抖动。

Redis Skill - 高性能缓存管理
Redis Skill - 高性能缓存管理

Redis 缓存和数据结构管理技能。通过自然语言操作 Redis,支持 String、Hash、List、Set、ZSet、Stream 等数据结构操作。当用户提到 Redis、缓存、消息队列、会话存储时使用此技能。

下载

Java 中调用该脚本的典型写法(Lettuce 客户端)

别把 Lua 脚本硬编码在 Java 字符串里——维护困难且无法复用 SHA。应预加载脚本并缓存其 SHA 值,后续用 EVALSHA 调用,降低网络和解析开销。

  • 首次加载:用 redisClient.eval(script, ScriptOutputType.MULTI) 获取 SHA,存入静态变量或配置中心
  • 日常调用:用 redisClient.evalSha(sha, ScriptOutputType.MULTI, keys, args)
  • args 必须是字符串数组,数值要 String.valueOf(...) 转换,否则 Lettuce 会序列化失败
  • 务必设置超时:TimeoutOptions.enabled(Duration.ofMillis(50)),防止 Lua 卡死拖垮整个连接池

容易踩的坑:HMGET 返回 nil 时,Lua 的 tonumber(nil) 是 nil,后续计算会报错。必须显式判空,如 tonumber(bucket[1]) or now_ms。

压测时令牌数异常归零或突增的根本原因

这不是脚本 bug,大概率是客户端时间不同步或调用方传入的 ARGV[3] 时间戳混乱。比如 Docker 容器未启用 host.time 共享、K8s 节点时钟漂移、或前端 JS 取 Date.now() 后经多层代理延迟送达。

解决方案很直接:

  • 强制所有服务端节点开启 NTP 同步,并监控时钟偏移(ntpq -p)
  • 限流脚本中弃用客户端传的时间,改用 redis.call("TIME")(返回秒+微秒数组),再换算成毫秒:(time_arr[1] * 1000 + math.floor(time_arr[2] / 1000))
  • 若必须用客户端时间(如需对齐业务日志时间线),增加校验:若传入时间比 Redis 当前时间早 >5s 或晚 >1s,直接拒绝并返回错误码

真正难调试的是跨机房部署时的时钟误差——哪怕只有 100ms 偏移,在 1000 QPS 下就可能造成 10% 以上的误判率。这事没法靠代码绕过,得从基础设施层面卡死。

热门AI工具

更多
VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

相关专题

更多
页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

5716

2023.08.14

常用的数据库软件
常用的数据库软件

常用的数据库软件有MySQL、Oracle、SQL Server、PostgreSQL、MongoDB、Redis、Cassandra、Hadoop、Spark和Amazon DynamoDB。更多关于数据库软件的内容详情请看本专题下面的文章。php中文网欢迎大家前来学习。

4469

2023.11.02

内存数据库有哪些
内存数据库有哪些

内存数据库有Redis、Memcached、Apache Ignite、VoltDB、TimesTen、H2 Database、Aerospike、Oracle TimesTen In-Memory Database、SAP HANA和ache Cassandra。更多关于内存数据库相关问题,详情请看本专题下面的文章。php中文网欢迎大家前来学习。

3995

2023.11.14

mongodb和redis哪个读取速度快
mongodb和redis哪个读取速度快

redis 的读取速度比 mongodb 更快。原因包括:1. redis 使用简单的键值存储,而 mongodb 存储 json 格式的数据,需要解析和反序列化。2. redis 使用哈希表快速查找数据,而 mongodb 使用 b-tree 索引。因此,redis 在需要高性能读取操作的应用程序中是一个更好的选择。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

7372

2024.04.02

redis怎么做缓存服务器
redis怎么做缓存服务器

redis 作为缓存服务器的答案:redis 是一款开源、高性能、分布式的键值存储,可作为缓存服务器使用。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

643

2024.04.07

redis怎么解决数据一致性
redis怎么解决数据一致性

redis 提供了两种一致性模型,以维护副本数据一致性:强一致性 (sync) 确保写操作仅在复制到所有从节点后才完成;最终一致性 (async) 则在主节点上写操作后认为已完成,牺牲一致性换取性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

756

2024.04.07

mysql和redis怎么保证双写一致性
mysql和redis怎么保证双写一致性

确保 mysql 和 redis 双写一致性的技术包括:1、事务性更新:同时更新 mysql 和 redis,保证一致性;2、主从复制:mysql 主服务器更改同步到 redis 从服务器;3、基于事件的更新:mysql 记录更改并发送到 redis等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

7182

2024.04.07

redis缓存一般存些什么数据
redis缓存一般存些什么数据

redis缓存中存储的数据类型包括:字符串、哈希、列表、集合、有序集合、位图、地理空间数据和hyperloglog。这些数据类型适用于存储各种数据,从简单信息到复杂对象和地理位置。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1160

2024.04.07

Linux 新手先学哪些命令
Linux 新手先学哪些命令

Linux新手命令专题汇集终端基础、文件目录操作、文本查看、软件安装和权限管理等入门文章,帮助按实际任务练习常用命令并建立命令行操作习惯。

0

2026.10.15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
phpEnv手册
phpEnv手册

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn