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

陌宇君_1751

陌宇君_1751

2026-09-27

318人浏览

原创

直接用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 秒才开始缓慢恢复——这个行为是否符合业务预期,得看你的服务容忍度,而不是脚本能改出来的。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

redis

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2023.11.02

4149

19

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

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

2023.11.14

3635

11

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

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

2024.04.02

6432

6

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

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

2024.04.07

603

6

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

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

2024.04.07

716

6

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

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

2024.04.07

5922

6

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

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

2024.04.07

1140

6

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

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

2024.04.07

956

6

redis主要作用有哪些
redis主要作用有哪些

redis 的主要作用包括:1. 缓存数据,提高访问速度;2. 充当消息队列,实现消息传递;3. 存储各种数据类型,如字符串、散列和集合;4. 管理会话信息,确保可靠性和可用性;5. 限制请求速率,防止服务器超载等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.04.07

5358

6

热门下载

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

精品课程

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

共0课时 | 0人学习

进程与SOCKET
进程与SOCKET

共6课时 | 0.5万人学习