修复限流器问题
This commit is contained in:
@@ -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")
|
||||
Reference in New Issue
Block a user