fix(rate-limit): apply on router to avoid TypeError on ws endpoint

This commit is contained in:
MingxuanGame
2025-08-28 16:41:33 +00:00
parent 4324d7ed2b
commit fdb08fe31f
5 changed files with 23 additions and 14 deletions

10
main.py
View File

@@ -37,12 +37,11 @@ from app.service.osu_rx_statistics import create_rx_statistics
from app.service.redis_message_system import redis_message_system
from app.utils import bg_tasks, utcnow
from fastapi import Depends, FastAPI, HTTPException, Request
from fastapi import FastAPI, HTTPException, Request
from fastapi.exceptions import RequestValidationError
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
from fastapi_limiter import FastAPILimiter
from fastapi_limiter.depends import RateLimiter
import sentry_sdk
@@ -152,13 +151,6 @@ app = FastAPI(
lifespan=lifespan,
description=desc,
)
if settings.enable_rate_limit:
app.router.dependencies.extend(
[
Depends(RateLimiter(times=1200, minutes=1)),
Depends(RateLimiter(times=200, seconds=1)),
]
)
app.include_router(api_v2_router)