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

ops/2024/9/23 20:14:41/

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表示请求被限流
elsereturn 1 -- 返回1表示请求被允许
end


http://www.ppmy.cn/ops/6017.html

相关文章

JavaScript-4.正则表达式、BOM

正则表达式 正则表达式包含在"/","/"中 开始与结束 ^ 字符串的开始 $ 字符串的结束 例: "^The":表示所有以"The"开始的字符串("There"、"The cat"等&#x…

AI网关助力边缘物联网

随着人工智能技术的迅猛发展,AI网关作为通往智能未来的桥梁,正逐渐被人们所认识和重视。AI网关是一种智能设备,它连接了各种智能终端设备,促进它们之间的交流和协同工作。在这个信息化、智能化时代,AI网关的作用愈发凸…

Java基于微信小程序的讲座预约系统的研究与实现,附源码

博主介绍:✌程序员徐师兄、7年大厂程序员经历。全网粉丝12w、csdn博客专家、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java技术领域和毕业项目实战✌ 🍅文末获取源码联系🍅 👇🏻 精彩专栏推荐订阅👇…

openjudge_2.5基本算法之搜索_1804:小游戏

题目 1804:小游戏 总时间限制: 1000ms 内存限制: 65536kB 描述 一天早上,你起床的时候想:“我编程序这么牛,为什么不能靠这个赚点小钱呢?”因此你决定编写一个小游戏。 游戏在一个分割成w * h个正方格子的矩形板上进行。如图所示…

分布式光纤测温解决方案

安科瑞电气股份有限公司 祁洁 15000363176 一、方案介绍 分布式光纤测温(DTS)集光电信号检测、计算机技术等为一体,具有实时监测、测温精度高、测量距离长、可精确定位、采用光纤作为传感器和传输介质,具有抗电磁干扰、本征防…

C/C++中设置随机数

前言 我们通常在写一个数据结构后,需要去测试其正确性和性能比较,那在平常手动输入数据的方式太鸡肋,并且不具有普遍性和随机性。基于这个原因,我们必须要掌握设置随机数,不但可以给我们提供更多的数据,还可…

HttpServletResponse HttpServletRequest

HttpServletResponse 和 HttpServletRequest 是 Java Servlet API 中的两个核心接口,它们分别代表了 HTTP 响应和 HTTP 请求。在基于 Java 的 Web 应用中,特别是使用 Servlet 技术时,这两个接口被广泛应用。 HttpServletRequest HttpServle…

6G技术,信道建模

目录 6G技术: 6G信道建模: 6G信号处理: 6G天线技术: 6G频谱共享: