fastapi 如何实现简单的基于 IP 的频率限制-Python教程-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
<h1>简单内存存储：{ip: [timestamp1, timestamp2, ...]}</h1><p>ip_request_times: Dict[str, List[float]] = {}</p><h1>限流配置</h1><p>MAX_REQUESTS = 10
TIME_WINDOW_SEC = 60</p><p>def rate_limit_dependency(request: Request):
client_ip = request.client.host
now = time.time()</p><pre class="brush:python;toolbar:false;"># 清理过期时间戳
if client_ip in ip_request_times:
    ip_request_times[client_ip] = [
        t for t in ip_request_times[client_ip] if now - t = 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"}