讲师中心 微信公众号

工具推荐

视频效率加速
AniShort
AniShort

AI短剧协同制作平台

Laper
Laper

剧本策划到成片协作

立刻MV
立刻MV

低成本歌曲 MV、视频生成

updream

一键生成高质量推广成片

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

落晨大大_4655

落晨大大_4655

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

|

970人浏览过

|

来源于php中文网

原创

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

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

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

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

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

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

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

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

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

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

Biopython
Biopython

全面的分子生物学工具集。适用于序列操作、文件解析(FASTA/GenBank/PDB)、系统发育分析,以及通过 Bio.Entrez 以编程方式访问 NCBI/

下载
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_eval 的 usec_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是一款面向对话、写作、编程和推理场景的AI大模型工具。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1911

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

5324

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1849

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

32257

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3807

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3827

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1263

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

616

2023.08.10

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

80

2026.11.06

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn