From 63f9bdd068ff2d475e6005386aa3d699b9288099 Mon Sep 17 00:00:00 2001
From: Nik Stuckenbrock <privat@nikstuckenbrock.de>
Date: Fri, 19 Jul 2024 15:08:43 +0200
Subject: [PATCH] Fix in memory example

---
 examples/in_memory/main.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

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)