add cache

Author: sumaro2101

api_v1/users/views.py

@@ -1,5 +1,6 @@
 from fastapi import APIRouter, Depends
 from fastapi_users import FastAPIUsers
+from fastapi_cache.decorator import cache
 
 from config.models import User
 from api_v1.auth.backends import auth_backend
@@ -18,11 +19,11 @@
 
 
 @router.get(path='/test',
-            response_model=UserRead,
             dependencies=[Depends(active_user)],
             )
+@cache()
 async def test_end_point(user: User = Depends(active_user)):
-    return user
+    return dict(name=user.email)
 
 
 router.include_router(fastapi_users.get_users_router(UserRead, UserUpdate),

main.py

@@ -1,12 +1,18 @@
 from contextlib import asynccontextmanager
 from fastapi import FastAPI
+from fastapi_cache import FastAPICache
+from fastapi_cache.backends.redis import RedisBackend
+from fastapi_cache.decorator import cache
+
+from redis import asyncio as aioredis
 
 from api_v1 import register_routers
 from app_includes import (
     register_errors,
     register_middlewares,
     register_prometheus,
     )
+from config import settings
 
 
 def start_app() -> FastAPI:
@@ -23,7 +29,15 @@ def start_app() -> FastAPI:
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
+    redis = aioredis.from_url(settings.redis.redis_url)
+    FastAPICache.init(RedisBackend(redis), prefix='fastapi-cache')
     yield
 
 
 app = start_app()
+
+
+@app.get(path='/test')
+@cache()
+async def test_end_point():
+    return dict(hello='world')

pyproject.toml

@@ -20,8 +20,8 @@ alembic = "^1.14.0"
 pytest = "^8.3.3"
 prometheus-fastapi-instrumentator = "^7.0.0"
 fastapi-users = {extras = ["sqlalchemy"], version = "^14.0.0"}
-redis = "^5.2.1"
-aioredis = "^2.0.1"
+redis = {extras = ["redis"], version = "^5.2.1"}
+fastapi-cache2 = "^0.2.2"
 
 
 [tool.poetry.group.dev.dependencies]