Public/fastapi-cache
1
0
Fork 0
mirror of https://github.com/long2ice/fastapi-cache.git synced 2026-03-25 04:57:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix redis example

Browse Source
This commit is contained in:
Nik Stuckenbrock
2024-07-19 15:09:39 +02:00
parent 63f9bdd068
commit 552d54c1ca
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
examples/redis/main.py
View File

@@ -1,5 +1,7 @@
# pyright: reportGeneralTypeIssues=false
from contextlib import asynccontextmanager
import time
from typing import AsyncIterator
import pendulum
import uvicorn
@@ -17,7 +19,15 @@ from starlette.responses import JSONResponse, Response
import redis.asyncio as redis
from redis.asyncio.connection import ConnectionPool
app = FastAPI()
@asynccontextmanager
async def lifespan(_: FastAPI) -> AsyncIterator[None]:
pool = ConnectionPool.from_url(url="redis://redis")
r = redis.Redis(connection_pool=pool)
FastAPICache.init(RedisBackend(r), prefix="fastapi-cache")
yield
app = FastAPI(lifespan=lifespan)
app.mount(
path="/static",
@@ -80,12 +90,5 @@ async def cache_response_obj():
return JSONResponse({"a": 1})
@app.on_event("startup")
async def startup():
pool = ConnectionPool.from_url(url="redis://redis")
r = redis.Redis(connection_pool=pool)
FastAPICache.init(RedisBackend(r), prefix="fastapi-cache")
if __name__ == "__main__":
uvicorn.run("main:app", reload=True)