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

梦静大大_1510

梦静大大_1510

2026-06-27

772人浏览

原创

必须用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% 以上的误判率。这事没法靠代码绕过,得从基础设施层面卡死。

相关文章

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

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

下载

相关标签:

redis redis eval

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

相关专题

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

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

2023.08.14

5756

4

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

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

2023.11.02

4469

19

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

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

2023.11.14

4015

11

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

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

2024.04.02

7392

6

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

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

2024.04.07

643

6

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

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

2024.04.07

756

6

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

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

2024.04.07

7242

6

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

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

2024.04.07

1160

6

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

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

2024.04.07

1096

6

热门下载

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

精品课程

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

共0课时 | 0人学习