feat(api): added rate limiting

- Need update on testing env
This commit is contained in:
2024-01-24 21:09:15 -05:00
parent dd0bcbc984
commit ec6e39bd7c
6 changed files with 201 additions and 12 deletions

View File

@@ -1,6 +1,9 @@
from fastapi import APIRouter
router = APIRouter(prefix="/control", tags=["control"])
router = APIRouter(
prefix="/control",
tags=["control"],
)
@router.get("/ping")

View File

@@ -1,8 +1,20 @@
from contextlib import asynccontextmanager
import redis.asyncio as redis
from fastapi import FastAPI
from fastapi_limiter import FastAPILimiter
from src.neo_neo_todo.control import control
app = FastAPI()
@asynccontextmanager
async def lifespan(_: FastAPI):
redis_connection = redis.from_url("redis://localhost:6379", encoding="utf8")
await FastAPILimiter.init(redis_connection)
yield
await FastAPILimiter.close()
app = FastAPI(lifespan=lifespan)
app.include_router(control.router)