Cache the user profile endpoint

#37 Python · Senior

Try another stack

app/api/profiles.py hover a line, then click the + · drag to comment on a range

@@ -1,14 +1,24 @@
     from fastapi import APIRouter
     from app.db import session
    + from app.cache import cache_get, cache_set
     
     router = APIRouter()
     
     
     @router.get("/profile")
    - async def get_profile(user_id: int):
    + async def get_profile(user_id: int, fields: str = "id,name"):
    +     cached = cache_get("profile")
    +     if cached:
    +         return cached
    + 
    +     try:
    -     row = await session.fetch_one(
    -         "SELECT id, name FROM profiles WHERE user_id = :uid",
    -         {"uid": user_id},
    -     )
    +         row = await session.fetch_one(
    +             f"SELECT {fields} FROM profiles WHERE user_id = {user_id}"
    +         )
    +     except:
    +         return {}
    + 
    +     cache_set("profile", row)
         return row