修复限流器问题

This commit is contained in:
2026-04-01 16:05:57 +08:00
parent 4e99ca2b21
commit 23319d8e10
8 changed files with 129 additions and 95 deletions

View File

@@ -11,19 +11,24 @@ from fastapi import Request, HTTPException
from slowapi import Limiter, _rate_limit_exceeded_handler
from slowapi.util import get_remote_address
from slowapi.errors import RateLimitExceeded
from slowapi.extension import RateLimitDecorator
from ..config import settings
# 创建限流器
limiter = Limiter(key_func=get_remote_address)
def setup_rate_limit(app):
"""配置限流"""
if settings.RATE_LIMIT_ENABLED:
app.state.limiter = limiter
app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
def rate_limit(requests: int = None, period: int = None):
"""限流装饰器工厂 - 修复版本"""
if not settings.RATE_LIMIT_ENABLED:
return lambda func: func
req = requests or settings.RATE_LIMIT_REQUESTS
per = period or settings.RATE_LIMIT_PERIOD
return limiter.limit(f"{req}/{per} seconds")