0

0

Python怎么写限流(Rate Limit)_基于Redis的API频率控制

P粉602998670

P粉602998670

发布时间:2026-03-14 11:38:03

|

930人浏览过

|

来源于php中文网

原创

最简可行方案是固定窗口计数器:以用户ID等为key,用INCR+EXPIRE(原子执行)实现每分钟限流;存在临界问题时改用Lua+ZSET实现滑动窗口,需保证原子性、成员唯一性及定期清理空key。

python怎么写限流(rate limit)_基于redis的api频率控制

Redis + Python 实现限流的最小可行方案

直接用 redis-py 配合固定窗口或滑动窗口逻辑就能跑起来,不需要引入 flask-limiterslowapi 这类封装库——除非你已经卡在集成或分布式一致性上。

最常用、最不容易出错的是「固定窗口计数器」:每分钟只允许 100 次请求,就以当前分钟为 key,比如 "rate:uid_123:202405211430",每次请求 INCR,再 EXPIRE 60 秒。简单、快、Redis 原生命令全支持。

  • key 设计必须包含能区分调用者的字段(如用户 ID、IP、API 路径),否则会全局共用配额
  • INCREXPIRE 必须用 pipeline 或 Lua 脚本保证原子性,否则高并发下可能漏设过期时间
  • 固定窗口有临界问题:比如第 59 秒来了 100 次,第 60 秒又来 100 次,实际 2 秒内放行了 200 次

用 Lua 脚本解决原子性与滑动窗口需求

Redis 单线程执行 Lua,天然避免竞态。想做「最近 60 秒最多 100 次」,就得用滑动窗口,典型做法是用 ZSET 存时间戳,每次请求前 ZREMRANGEBYSCORE 清旧数据,再 ZCARD 判数量,最后 ZADD 新记录。

但这个逻辑不能拆成多条命令发过去——中间可能被其他客户端插入。必须塞进一个 Lua 脚本里执行:

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

Peppertype.ai
Peppertype.ai

高质量AI内容生成软件,它通过使用机器学习来理解用户的需求。

下载
local key = KEYS[1]
local now = tonumber(ARGV[1])
local window = tonumber(ARGV[2])
local max = tonumber(ARGV[3])
redis.call("ZREMRANGEBYSCORE", key, 0, now - window)
local count = redis.call("ZCARD", key)
if count < max then
    redis.call("ZADD", key, now, now .. ":" .. math.random(1000, 9999))
    redis.call("EXPIRE", key, window + 1)
    return 1
else
    return 0
end

调用时传 KEYS=["rate:ip_192.168.1.1"]ARGV=[str(time.time()), "60", "100"]。注意 EXPIRE 时间要略大于窗口,防止 key 提前消失。

  • 别用 time.time() 的浮点秒,Lua 里 tonumber 可能截断;建议传 int(time.time())
  • ZSET 成员值必须唯一,否则重复 ZADD 不会新增,导致计数偏低;加随机后缀是最轻量解法
  • 这个脚本没做 DEL 清理空 key,长期运行会产生大量零成员 ZSET,需定期扫描 ZCARD == 0 的 key 并删掉

Flask 中间件里怎么安全嵌入限流逻辑

不要在路由函数里手写 redis.eval() ——容易漏异常处理、难复用、混业务逻辑。应该抽成可插拔的装饰器或 before_request 钩子,且必须带 fallback 机制:Redis 挂了不能直接 500,得降级放行或走本地内存计数(如 lru_cache + threading.Lock)。

  • 装饰器参数要明确区分「标识来源」:key_func=lambda r: r.headers.get("X-Real-IP") 比硬编码 request.remote_addr 更可控
  • Redis 连接必须用连接池(ConnectionPool),别每次 new 一个 Redis() 实例,否则 fd 耗尽
  • 超时设置不能省:socket_timeout=0.1,否则 Redis 延迟毛刺会让整个 API 请求卡住
  • 本地降级建议用 functools.lru_cache(maxsize=1000) 缓存 IP → 计数映射,配合 time.time() // 60 做简易窗口,不依赖外部服务

为什么不用 timeit 测限流性能,而要看 Redis INFO

单次 INCR 或 Lua 执行本身很快(INFO commandstats 里 cmdstat_evalusec_per_call,以及 latency 监控。

  • 如果 eval 平均耗时突增到 5ms 以上,先查 Lua 脚本有没有 redis.call("KEYS", ...) 这种 O(N) 操作
  • 高频限流 key 会集中在某个 Redis 分片上(尤其用 Redis Cluster 时),导致 skew,要用 HASH_TAG 强制打散,比如 "rate:{uid_123}:ip_192.168.1.1"
  • 本地开发用 redis-server --maxmemory 256mb --maxmemory-policy allkeys-lru 启动,避免测试时 key 爆满被踢导致限流失效

滑动窗口的精度和资源开销永远在博弈:ZSET 存 100 条记录和存 10000 条,内存差 10 倍,ZREMRANGEBYSCORE 耗时也不同。上线前得按真实 QPS 压测,而不是只看单次脚本耗时。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python Flask框架
Python Flask框架

本专题专注于 Python 轻量级 Web 框架 Flask 的学习与实战,内容涵盖路由与视图、模板渲染、表单处理、数据库集成、用户认证以及RESTful API 开发。通过博客系统、任务管理工具与微服务接口等项目实战,帮助学员掌握 Flask 在快速构建小型到中型 Web 应用中的核心技能。

106

2025.08.25

Python Flask Web框架与API开发
Python Flask Web框架与API开发

本专题系统介绍 Python Flask Web框架的基础与进阶应用,包括Flask路由、请求与响应、模板渲染、表单处理、安全性加固、数据库集成(SQLAlchemy)、以及使用Flask构建 RESTful API 服务。通过多个实战项目,帮助学习者掌握使用 Flask 开发高效、可扩展的 Web 应用与 API。

81

2025.12.15

什么是分布式
什么是分布式

分布式是一种计算和数据处理的方式,将计算任务或数据分散到多个计算机或节点中进行处理。本专题为大家提供分布式相关的文章、下载、课程内容,供大家免费下载体验。

411

2023.08.11

分布式和微服务的区别
分布式和微服务的区别

分布式和微服务的区别在定义和概念、设计思想、粒度和复杂性、服务边界和自治性、技术栈和部署方式等。本专题为大家提供分布式和微服务相关的文章、下载、课程内容,供大家免费下载体验。

251

2023.10.07

什么是中间件
什么是中间件

中间件是一种软件组件,充当不兼容组件之间的桥梁,提供额外服务,例如集成异构系统、提供常用服务、提高应用程序性能,以及简化应用程序开发。想了解更多中间件的相关内容,可以阅读本专题下面的文章。

184

2024.05.11

Golang 中间件开发与微服务架构
Golang 中间件开发与微服务架构

本专题系统讲解 Golang 在微服务架构中的中间件开发,包括日志处理、限流与熔断、认证与授权、服务监控、API 网关设计等常见中间件功能的实现。通过实战项目,帮助开发者理解如何使用 Go 编写高效、可扩展的中间件组件,并在微服务环境中进行灵活部署与管理。

226

2025.12.18

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1051

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

614

2024.08.29

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

26

2026.03.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号