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

如何在Redis中用Lua脚本实现令牌桶限流算法?

落浩酱_9656

落浩酱_9656

发布时间:2026-10-02 12:22:19

|

150人浏览过

|

来源于php中文网

原创

因为INCR与EXPIRE非原子操作,中间中断会导致key永久残留或超发;必须用Lua脚本将令牌计算、校验、扣减、更新时间戳等步骤原子执行。

如何在redis中用lua脚本实现令牌桶限流算法?

为什么直接用 INCR + EXPIRE 不能可靠实现令牌桶?

因为 Redis 的原子性只保证单命令,INCR 和 EXPIRE 分开执行时可能在中间被中断(比如服务重启、网络断开),导致令牌数已增加但过期时间未设置,桶“永远不漏”。更糟的是,令牌桶需要动态计算「当前可用令牌数 = 上次填充量 + 新增令牌 - 消耗量」,涉及读-改-写,必须用 Lua 脚本保证整个逻辑原子执行。

EVAL 脚本里怎么算「当前令牌数」和「是否放行」?

核心是记录两个关键值:上次更新时间戳(last_time)和当前令牌数(tokens)。每次请求进来,先用当前时间减去 last_time,按速率算出应新增的令牌(向下取整),再与最大容量取最小值,最后扣减本次请求所需令牌。脚本返回 1 表示放行,0 表示拒绝。

示例脚本(限流键为 rate:uid:123,最大容量 10,每秒补充 2 个):

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

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

下载
local key = KEYS[1]
local capacity = tonumber(ARGV[1])
local fill_rate = tonumber(ARGV[2])
local now = tonumber(ARGV[3])
local tokens_needed = tonumber(ARGV[4])
<p>local bucket = redis.call("HMGET", key, "tokens", "last_time")
local tokens = tonumber(bucket[1]) or capacity
local last_time = tonumber(bucket[2]) or now</p><p>-- 计算应补充的令牌数(注意:单位是秒,需转为毫秒做差)
local delta = math.floor((now - last_time) * fill_rate)
tokens = math.min(capacity, tokens + delta)
local allowed = tokens >= tokens_needed
if allowed then
tokens = tokens - tokens_needed
end</p><p>-- 总是更新 last_time 为当前时间(不是 now + delta)
redis.call("HMSET", key, "tokens", tokens, "last_time", now)
redis.call("EXPIRE", key, 60) -- 防误存,设个兜底过期时间
return allowed and 1 or 0

调用时传参顺序和时间戳单位容易错在哪?

Redis EVAL 的 ARGV 是字符串数组,Lua 里必须显式 tonumber();更重要的是,now 必须用毫秒时间戳(如 Node.js 的 Date.now()),而 fill_rate 单位是「令牌/秒」,所以计算 delta 时要除以 1000 —— 很多人漏掉这个换算,导致补得太快或太慢。

  • 错误写法:delta = (now - last_time) * fill_rate(若 now 是毫秒,结果会大 1000 倍)
  • 正确写法:delta = math.floor((now - last_time) / 1000 * fill_rate)
  • EXPIRE 时间建议设为远大于业务周期的值(比如 60 秒),避免桶提前消失,但又不能设成永不过期

为什么不用 CL.THROTTLE?它和 Lua 实现有啥实质区别?

CL.THROTTLE 是 Redis 6.2+ 内置命令,本质也是 Lua 实现,但它返回的是结构化信息(剩余令牌、重试等待秒数等),且默认使用「滑动窗口」逻辑而非严格令牌桶。如果你需要精确控制填充节奏(比如每 500ms 补 1 个)、或依赖自定义字段(如绑定用户角色动态调整容量),还是得自己写 Lua;另外,CL.THROTTLE 不支持自定义过期策略,对长期空闲的 key 无法自动清理。

真正难的不是写对脚本,而是压测时发现「高并发下令牌数偶尔突降为负」——这通常是因为客户端传入的 now 时间不同步,或脚本里没处理 tokens 小于 0 的边界(应强制设为 0)。别省那句 tokens = math.max(0, tokens)。

热门AI工具

更多
UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

PixTV
PixTV Hot

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

立刻MV
立刻MV Hot

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

豆包大模型

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

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

WorkBuddy

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

音述AI
音述AI Hot

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

DeepSeek

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

相关专题

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

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

5036

2023.08.14

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

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

4229

2023.11.02

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

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

3715

2023.11.14

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

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

6692

2024.04.02

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

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

623

2024.04.07

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

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

736

2024.04.07

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

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

6222

2024.04.07

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

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

1140

2024.04.07

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

0

2026.09.30

热门下载

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

精品课程

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

共0课时 | 0人学习

进程与SOCKET
进程与SOCKET

共6课时 | 0.5万人学习

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

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