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

如何在Redis中使用Lua脚本实现漏桶限流?

陌宇君_1751

陌宇君_1751

发布时间:2026-09-27 10:07:20

|

318人浏览过

|

来源于php中文网

原创

直接用INCR+EXPIRE做漏桶会出错,因无法原子执行“检查水位→放行判断→更新水位→续期时间”整套逻辑,并发时多个请求可能同时读到低水位而超限放行;必须用Lua脚本封装惰性漏水计算并由客户端传入高精度时间戳。

如何在redis中使用lua脚本实现漏桶限流?

为什么直接用 INCR + EXPIRE 做漏桶会出错

漏桶的核心是“匀速漏水”+“突发流量缓冲”,但 Redis 单命令无法原子性地完成“检查当前水位→决定是否放行→更新水位→设置/续期过期时间”这一整套逻辑。如果拆成多个命令(比如先 GET 再 INCR 再 EXPIRE),并发时会出现竞态:两个请求同时读到水位为 0,都允许通过,实际就超限了。

必须用 Lua 脚本把整个判断和更新逻辑包在一个原子操作里,且脚本内不能依赖客户端时间(比如 redis.call('TIME') 返回的是服务器时间,但漏桶需要按固定速率“漏水”,得靠时间戳差值算漏了多少)。

EVAL 脚本里怎么模拟“漏水”过程

关键不是真等时间过去,而是每次请求来时,用当前时间减去上一次记录的时间,算出该漏掉多少单位(向下取整),再从当前水位里减掉——这叫“惰性漏水”。水位不能低于 0,也不能超过桶容量。

  • 脚本接收 5 个参数:KEYS[1](桶的 key)、ARGV[1](桶容量)、ARGV[2](漏水速率,单位:次/秒)、ARGV[3](当前请求时间戳,毫秒)、ARGV[4](本次请求尝试消耗的令牌数,通常为 1)
  • 用 redis.call('HMGET', KEYS[1], 'water', 'last_time') 一次性读出当前水位和上次更新时间
  • 若 last_time 为空,说明桶是新的,直接设水位为 math.min(ARGV[4], ARGV[1]),并存入当前时间
  • 否则计算已过去秒数:(tonumber(ARGV[3]) - tonumber(last_time)) / 1000,再算应漏水量:math.floor(elapsed * tonumber(ARGV[2]))
  • 新水位 = math.max(0, math.min(tonumber(water) - leak, ARGV[1])),然后判断 new_water >= ARGV[4] 决定是否放行

示例调用:

redis-cli --eval ./leaky_bucket.lua my:bucket , 10 0.5 1717023456789 1

为什么 ARGV[3] 必须由客户端传入时间戳

不能在 Lua 里调用 redis.call('TIME'),因为它的返回是秒级精度且不包含毫秒,误差太大;更不能用 os.time()(Redis 的 Lua 环境禁用该函数)。客户端必须用高精度时间(如 Date.now() 或 System.currentTimeMillis())生成毫秒时间戳,并传进来——这是保证漏水计算准确的唯一方式。

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

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

下载

另一个陷阱:如果客户端时间偏差大(比如快了 10 秒),会导致算出的 leak 过大,水位被清空,误拒合法请求。生产环境务必校准客户端与 Redis 服务器时间(NTP 同步)。

桶的存储结构选 HSET 还是 SET + EXPIRE

用 HSET 存 {water: "5", last_time: "1717023456789"} 更合理。虽然单个 SET + 过期也能存水位,但无法原子读写两个字段——你没法只更新 last_time 而不重置 water,也没法在判断时同时拿到两者。

注意点:

  • 不要给桶 key 设置固定过期时间(比如 EXPIRE my:bucket 3600),因为漏桶要长期存在,只是内部状态随时间变化
  • 如果长时间没请求,水位自然漏完变成 0,但 key 本身保留,下次来请求时继续基于 last_time 计算即可
  • 避免用 DEL 清桶——这会丢失 last_time,导致下一次请求误判为“新桶”,瞬间放行满容量

真正难处理的是“突发后持续低频”的场景:比如桶容量 10、速率 1/s,突然来 10 个请求打满,之后每秒只来 1 个。这时水位会在第 10 秒才开始缓慢恢复——这个行为是否符合业务预期,得看你的服务容忍度,而不是脚本能改出来的。

热门AI工具

更多
Laper
Laper Hot

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

VibeKnow
VibeKnow Hot

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

墨刀AI
墨刀AI Hot

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

WorkBuddy

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

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

DeepSeek

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

UP简历
UP简历 Hot

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

豆包大模型

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

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

相关专题

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

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

4149

2023.11.02

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

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

3635

2023.11.14

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

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

6432

2024.04.02

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

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

603

2024.04.07

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

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

716

2024.04.07

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

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

5922

2024.04.07

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

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

1140

2024.04.07

redis的8种数据类型有哪些
redis的8种数据类型有哪些

redis 提供 8 种数据类型:字符串(文本、数字、二进制)、哈希(键值对)、列表(有序集合)、集合(无序唯一元素)、有序集合(按分数排序)、地理空间(地理位置)、hyperloglog(估计大数据基数)和位图(位序列存储)。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

956

2024.04.07

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

80

2026.09.23

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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