分布式限流——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个:

Lua 复制代码
-- 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
相关推荐
野蛮人6号36 分钟前
黑马点评系列问题之基础篇p7 06初识redis无法在虚拟机查到图形化界面存进去的键
redis·黑马点评
幼稚园的山代王1 小时前
RabbitMQ 4.1.1初体验-队列和交换机
分布式·rabbitmq·ruby
小新学习屋1 小时前
Spark从入门到熟悉(篇三)
大数据·分布式·spark
大春儿的试验田3 小时前
高并发收藏功能设计:Redis异步同步与定时补偿机制详解
java·数据库·redis·学习·缓存
hqxstudying3 小时前
Redis为什么是单线程
java·redis
沉着的码农5 小时前
【设计模式】基于责任链模式的参数校验
java·spring boot·分布式
Fireworkitte5 小时前
Redis 源码 tar 包安装 Redis 哨兵模式(Sentinel)
数据库·redis·sentinel
西岭千秋雪_6 小时前
Redis性能优化
数据库·redis·笔记·学习·缓存·性能优化
Uluoyu7 小时前
redisSearch docker安装
运维·redis·docker·容器
KK溜了溜了16 小时前
JAVA-springboot 整合Redis
java·spring boot·redis