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

如何在Redis中实现分布式限流令牌桶算法_通过Lua脚本控制令牌速率

星丽同学_8275

星丽同学_8275

发布时间:2026-06-28 08:45:36

|

306人浏览过

|

来源于php中文网

原创

必须用Lua脚本实现原子操作,因INCR+EXPIRE多命令存在竞态导致超发或key永久残留;Lua在Redis单线程内执行,确保令牌生成、校验、扣减三步一气呵成,彻底消除并发安全问题。

如何在redis中实现分布式限流令牌桶算法_通过lua脚本控制令牌速率

为什么直接用 INCR + EXPIRE 无法可靠实现令牌桶

因为 INCR 和 EXPIRE 是两个独立命令,中间存在竞态窗口:如果客户端执行 INCR 后崩溃或网络中断,EXPIRE 就不会被执行,导致 key 永久存在,后续所有请求都被拒绝。更糟的是,多个并发请求可能同时读到旧的令牌数、各自计算后写回,造成“超发”。Lua 脚本在 Redis 中原子执行,能彻底规避这个问题。

Lua 脚本必须包含的三个核心逻辑

一个可用的令牌桶 Lua 脚本不能只做“减令牌”,它得完整模拟桶行为:初始化桶、按时间补发令牌、判断是否允许请求、更新剩余令牌。以下逻辑缺一不可:

  • 用 redis.call("GET", KEYS[1]) 读当前令牌数,若为空则用 redis.call("SET", KEYS[1], ARGV[1], "EX", ARGV[2]) 初始化为容量,并记录过期时间(注意:不能用 EXPIRE 单独设,否则非原子)
  • 用 redis.call("TIME") 获取服务端时间戳,结合上次更新时间(存于 value 中,格式如 "{tokens}|{last_time}")计算应新增令牌数:min(capacity, current_tokens + rate * (now - last_time))
  • 判断 current_tokens >= 1:是则减 1 并写回新状态;否则返回 0 —— 这个返回值必须是整数(0 或 1),方便客户端直接判断

调用时 KEY 和 ARGV 的实际含义与陷阱

脚本本身不关心业务维度,全靠传入的 KEYS 和 ARGV 绑定上下文。常见错误是把用户 ID 当 KEY 直接拼进去,结果出现哈希槽不一致报错(CROSSSLOT Keys in request don't hash to the same slot)。正确做法:

redis-tools
redis-tools

按类别查找Redis命令、检测服务器连接、监控键数量和内存占用,支持离线速查表。

下载
  • KEYS[1] 必须是单个字符串,例如 "rate_limit:uid_123" —— 如果用集群,确保该 key 哈希后落在同一 slot(可加 {uid_123} 大括号强制哈希标签)
  • ARGV[1] 是桶容量(如 "100"),ARGV[2] 是 TTL 秒数(如 "60"),ARGV[3] 是每秒填充速率(如 "2")—— 注意:速率单位必须和 TTL 单位匹配,若 TTL 是 60 秒,速率就该是“60 秒内总补充量”,而非每秒数(否则浮点计算易出错)
  • 不要在 Lua 里做 os.time() 或 math.random():Redis 限制部分 Lua 函数,且 os.time() 返回的是 Redis 进程所在机器时间,未必和客户端一致;一律用 redis.call("TIME")

实际部署前必须验证的两个边界场景

脚本上线后最常出问题的不是主流程,而是边界情况。务必手动测试:

  • 桶已空时连续请求:预期返回 0,且 GET 查看 value 应保持 "0|{timestamp}" 格式,时间戳持续更新(证明补令牌逻辑仍在跑)
  • 首次请求时 key 不存在:脚本应自动初始化为满桶(如 "100|{time}"),而不是报错或返回 nil —— 若返回 nil,说明 GET 后没做空值判断就直接 string.match,会触发 Lua 错误
  • 注意 Redis 版本兼容性:低于 2.6 的 Redis 不支持 Lua;4.0+ 才支持 redis.call("TIME") 返回数组,旧版需用 redis.call("TIME")[1] 取秒数

真正难的不是写对脚本,而是让每个业务方传进来的 KEYS 和 ARGV 都符合约定,且集群环境下不跨 slot —— 这往往需要在客户端 SDK 层统一做 key 规范化处理,而不是靠运维事后补救。

热门AI工具

更多
PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

DeepSeek

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

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

豆包大模型

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

立刻MV
立刻MV Hot

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

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

WorkBuddy

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

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

音述AI
音述AI Hot

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

相关专题

更多
什么是分布式
什么是分布式

分布式是一种计算和数据处理的方式,将计算任务或数据分散到多个计算机或节点中进行处理。本专题为大家提供分布式相关的文章、下载、课程内容,供大家免费下载体验。

2033

2023.08.11

分布式和微服务的区别
分布式和微服务的区别

分布式和微服务的区别在定义和概念、设计思想、粒度和复杂性、服务边界和自治性、技术栈和部署方式等。本专题为大家提供分布式和微服务相关的文章、下载、课程内容,供大家免费下载体验。

3134

2023.10.07

页面置换算法
页面置换算法

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

5956

2023.08.14

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

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

4529

2023.11.02

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

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

4095

2023.11.14

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

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

7592

2024.04.02

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

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

643

2024.04.07

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

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

756

2024.04.07

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

80

2026.10.16

热门下载

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

精品课程

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

共0课时 | 0人学习

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

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