diff --git a/examples/in_memory/main.py b/examples/in_memory/main.py index ac2c5dc..c43dce3 100644 --- a/examples/in_memory/main.py +++ b/examples/in_memory/main.py @@ -1,5 +1,6 @@ # pyright: reportGeneralTypeIssues=false -from typing import Dict, Optional +from contextlib import asynccontextmanager +from typing import AsyncIterator, Dict, Optional import pendulum import uvicorn @@ -11,7 +12,13 @@ from pydantic import BaseModel from starlette.requests import Request from starlette.responses import JSONResponse, Response -app = FastAPI() +@asynccontextmanager +async def lifespan(_: FastAPI) -> AsyncIterator[None]: + FastAPICache.init(InMemoryBackend()) + yield + + +app = FastAPI(lifespan=lifespan) ret = 0 @@ -119,10 +126,5 @@ def namespaced_injection( } -@app.on_event("startup") -async def startup(): - FastAPICache.init(InMemoryBackend()) - - if __name__ == "__main__": uvicorn.run("main:app", reload=True)