Python应用如何实现基于Redis的滑动窗口限流逻辑【技巧】

admin 百科 10
Python中用Redis的ZSET实现滑动窗口限流,通过ZREMRANGEBYSCORE清理过期请求、ZADD插入新请求、ZCARD统计数量,并用Lua脚本保证原子性;支持按用户/IP/接口等多维度key设计,辅以降级与监控策略。

Python应用如何实现基于Redis的滑动窗口限流逻辑【技巧】-第1张图片-佛山资讯网

Python应用中用Redis实现滑动窗口限流,核心是利用Redis的ZSET(有序集合)按时间戳排序存储请求记录,并通过ZREMRANGEBYSCORE自动清理过期项,再用ZCARD统计当前窗口内请求数。不依赖外部库也能高效完成,关键是设计好时间粒度和key结构。

用ZSET模拟时间滑动窗口

ZSET的score设为毫秒级时间戳,member可设为唯一标识(如用户ID+时间戳或随机UUID),这样既能去重又能按时间排序。每次请求到来时:

  • 先用ZREMRANGEBYSCORE key -inf (current_timestamp - window_ms)剔除窗口外的旧记录
  • 再用ZADD key current_timestamp member插入新请求
  • 最后用ZCARD key获取当前窗口内请求数,与阈值比对

例如:1分钟内最多100次请求,window_ms = 60000,所有score小于time.time()*1000 - 60000的成员都会被自动清理。

避免单Key热点,支持多维度限流

直接用全局key会成为瓶颈,应按限流维度构造key,比如:

立即学习“Python免费学习笔记(深入)”;

  • 按用户限流:f"rate_limit:uid:{user_id}"
  • 按IP限流:f"rate_limit:ip:{ip_address}"
  • 按接口+用户组合:f"rate_limit:api:{endpoint}:uid:{user_id}"

注意key生命周期无需手动设置过期——只要窗口内无新请求,ZSET自然变空;也可给key加个较长的EXPIRE(如1小时),防止极端情况下残留空key堆积。

标签: python redis win 热点 red

发布评论 0条评论)

还木有评论哦,快来抢沙发吧~