分布式限流——Redis + Lua实现滑动窗口算法

Zset（有序集合）在Redis中用来实现滑动窗口限流的主要思路是利用其自动排序和可过期成员的特点：

1. 初始化及数据结构选择：

   • 为需要限流的接口或服务创建一个唯一的键（key）对应一个Zset。
   • Zset中的每个成员通常是请求的唯一标识符（如UUID或其他唯一字符串），用于区分不同的请求。
   • Zset的score字段用来存储每个请求的时间戳，由于Redis中的score支持浮点数，通常会存储Unix时间戳（秒级或毫秒级精度）。

2. 添加请求记录：

   • 当有新的请求到来时，将当前时间戳作为score，添加到Zset中，同时成员可以是任意唯一标识符，或者是省略，仅保留score的有序排列。

3. 检查窗口内的请求数量：

   • 根据限流策略（比如每分钟100次），计算出当前时间戳对应的窗口开始时间（当前时间减去窗口长度）。
   • 使用Zset的ZCARD命令，查找score在窗口范围内的元素数量。
   • 如果数量超过设定的阈值，则拒绝新请求。

4. 移除过期请求记录：

   • 可以结合Zset的过期功能（TTL）来自动清理超时的请求记录，也可以在每次处理请求时手动清理窗口开始时间之前的所有记录，这样能确保Zset只包含当前窗口内的请求。

5. 原子操作与并发控制：

   • 在高并发场景下，为了保证限流逻辑的正确执行，可以通过编写Lua脚本来实现一系列操作的原子性执行，避免因并发问题造成的计数不准确。

通过以上步骤，Zset能够有效地维护一个时间窗口内的请求数量，从而达到限流的目的，确保服务在高峰期不会因为过多请求而导致性能瓶颈。

以下是一个基于Redis Zset实现滑动窗口限流的Lua脚本示例，假设我们希望限制在过去的10秒内某个服务的请求数量不超过100个：

-- KEYS[1] 是服务的唯一标识符
-- ARGV[1] 是窗口大小（秒）
-- ARGV[2] 是限流阈值（请求次数）
-- ARGV[3] 是当前请求的时间戳（毫秒级）

-- 获取窗口开始时间戳
local windowStart = tonumber(ARGV[3]) - tonumber(ARGV[1]) * 1000

-- 移除窗口开始时间之前的请求记录
redis.call('ZREMRANGEBYSCORE', KEYS[1], 0, windowStart)

-- 添加当前请求记录
redis.call('ZADD', KEYS[1], ARGV[3], '')

-- 获取窗口内的请求数量
local requestCount = redis.call('ZCARD', KEYS[1])

-- 判断是否超过限流阈值
if requestCount > tonumber(ARGV[2]) then
    -- 如果超过阈值，删除刚添加的请求记录（模拟请求被拒绝）
    redis.call('ZREM', KEYS[1], '')
    return 0 -- 返回0表示请求被限流
else
    return 1 -- 返回1表示请求被允许
end

原文地址：https://blog.csdn.net/QGhurt/article/details/137889474

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！