0

0

fastapi 如何实现简单的基于 IP 的频率限制

冷漠man

冷漠man

发布时间:2026-01-17 21:32:02

|

146人浏览过

|

来源于php中文网

原创

FastAPI需借助依赖项与缓存实现IP限流:简易版用内存字典记录IP时间戳并清理过期项;增强版加asyncio.Lock防并发冲突;生产环境应换为Redis的sorted set,配合可信IP校验确保安全。

fastapi 如何实现简单的基于 ip 的频率限制

FastAPI 本身不内置 IP 频率限制功能,但可通过中间件或依赖项 + 缓存(如内存缓存或 Redis)轻松实现。下面介绍一种轻量、可直接运行的基于内存的 IP 限流方案,适合开发或低流量场景;也附上扩展到 Redis 的关键思路。

使用 FastAPI 依赖项 + 内存缓存(简易版)

核心思路:提取客户端 IP,按 IP 统计请求次数,设定时间窗口(如 60 秒内最多 10 次),超限则返回 429。

注意:客户端 IP 需从 request.client.host 获取,但在反向代理(如 Nginx)后需改用 X-Forwarded-For 头,本例默认直连;生产环境请务必校验可信代理头。

示例代码:

from fastapi import Depends, FastAPI, Request, HTTPException
from starlette.status import HTTP_429_TOO_MANY_REQUESTS
from typing import Dict, List, Optional
import time

简单内存存储:{ip: [timestamp1, timestamp2, ...]}

ip_request_times: Dict[str, List[float]] = {}

限流配置

MAX_REQUESTS = 10 TIME_WINDOW_SEC = 60

def rate_limit_dependency(request: Request): client_ip = request.client.host now = time.time()

# 清理过期时间戳
if client_ip in ip_request_times:
    ip_request_times[client_ip] = [
        t for t in ip_request_times[client_ip] if now - t < TIME_WINDOW_SEC
    ]
else:
    ip_request_times[client_ip] = []

# 检查是否超限
if len(ip_request_times[client_ip]) >= MAX_REQUESTS:
    raise HTTPException(
        status_code=HTTP_429_TOO_MANY_REQUESTS,
        detail="Too many requests from this IP"
    )

# 记录本次请求时间
ip_request_times[client_ip].append(now)
return True

app = FastAPI()

@app.get("/public") def public_endpoint(rate_limited: bool = Depends(rate_limit_dependency)): return {"message": "OK"}

用 asyncio.Lock 避免并发写冲突(推荐增强版)

上面的内存方案在多进程/多线程下不安全,且高并发时可能因竞态导致误放行。FastAPI 常运行于异步环境,可用 asyncio.Lock 保护共享字典操作:

微尔企业网站管理系统1.75 build build 090709
微尔企业网站管理系统1.75 build build 090709

系统功能介绍 1 包含企业网站所必备的功能:企业信息、产品管理、人才招聘、新闻资讯、企业图片、以及视频下载等模块2 由于是从CMS系统的基础上开发而成,因此相对于一些其他的企业网站管理系统,本系统具备更强的可扩展能力,可以胜任从小型工作室到大中型企业网上门户等各种不同规模网站的需求。3 后台管理与模板完全分离,并具备非常灵活的标签技术,可以实现无限制个性化的界面定制4 操作简单,利用已经制作好的模

下载
  • 为每个 IP 分配独立锁(避免全局锁瓶颈)
  • defaultdict + Lock 管理时间戳列表
  • 清理和插入逻辑加锁,确保原子性

实际项目中建议封装成可复用的依赖类,支持不同路由配置不同限流规则(如 /login 更严格)。

生产环境:替换为 Redis(推荐)

内存方案无法跨进程/实例共享,上线必须用 Redis。核心逻辑不变,只是把 ip_request_times 换成 Redis 的 sorted set 或 key+list:

  • ZREMRANGEBYSCORE 清理过期时间戳
  • ZCARD 判断当前请求数
  • ZADD 插入新时间戳(score = 时间戳)
  • Key 格式建议:flood:{ip}:{endpoint},便于按端点隔离

可借助 aioredisredis-py(v4.2+ 支持 async)实现异步操作,避免阻塞事件循环。

补充:获取真实客户端 IP 的健壮写法

若部署在 Nginx、Cloudflare 后,request.client.host 是代理 IP。应优先读取 X-Forwarded-For,但需验证来源可信(只信任你自己的反向代理):

  • 设置 trusted_hosts=["127.0.0.1", "10.0.0.0/8"]
  • request.headers.get("X-Forwarded-For", "").split(",")[0].strip() 取最左 IP
  • 务必配合 Starlette's TrustedHostMiddleware 或手动校验,防止伪造

不加校验直接信任 X-Forwarded-For 会导致限流完全失效。

相关专题

更多
nginx 重启
nginx 重启

nginx重启对于网站的运维来说是非常重要的,根据不同的需求,可以选择简单重启、平滑重启或定时重启等方式。本专题为大家提供nginx重启的相关的文章、下载、课程内容,供大家免费下载体验。

229

2023.07.27

nginx 配置详解
nginx 配置详解

Nginx的配置是指设置和调整Nginx服务器的行为和功能的过程。通过配置文件,可以定义虚拟主机、HTTP请求处理、反向代理、缓存和负载均衡等功能。Nginx的配置语法简洁而强大,允许管理员根据自己的需要进行灵活的调整。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

498

2023.08.04

nginx配置详解
nginx配置详解

NGINX与其他服务类似,因为它具有以特定格式编写的基于文本的配置文件。本专题为大家提供nginx配置相关的文章,大家可以免费学习。

498

2023.08.04

tomcat和nginx有哪些区别
tomcat和nginx有哪些区别

tomcat和nginx的区别:1、应用领域;2、性能;3、功能;4、配置;5、安全性;6、扩展性;7、部署复杂性;8、社区支持;9、成本;10、日志管理。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

227

2024.02.23

nginx报404怎么解决
nginx报404怎么解决

当访问 nginx 网页服务器时遇到 404 错误,表明服务器无法找到请求资源,可以通过以下步骤解决:1. 检查文件是否存在且路径正确;2. 检查文件权限并更改为 644 或 755;3. 检查 nginx 配置,确保根目录设置正确、没有冲突配置等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

333

2024.07.09

Nginx报404错误解决方法
Nginx报404错误解决方法

解决方法:只需要加上这段配置:try_files $uri $uri/ /index.html;即可。想了解更多Nginx的相关内容,可以阅读本专题下面的文章。

3508

2024.08.07

nginx部署php项目教程汇总
nginx部署php项目教程汇总

本专题整合了nginx部署php项目教程汇总,阅读专题下面的文章了解更多详细内容。

14

2026.01.13

nginx配置文件详细教程
nginx配置文件详细教程

本专题整合了nginx配置文件相关教程详细汇总,阅读专题下面的文章了解更多详细内容。

19

2026.01.13

高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

43

2026.01.16

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
进程与SOCKET
进程与SOCKET

共6课时 | 0.3万人学习

Redis+MySQL数据库面试教程
Redis+MySQL数据库面试教程

共72课时 | 6.4万人学习

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

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