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

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堆积。
版权声明:除非特别标注,否则均为本站原创文章,转载时请以链接形式注明文章出处。
还木有评论哦,快来抢沙发吧~