From 44dbf976b522302ea3f9bbfd7da9f7abe7cb8c92 Mon Sep 17 00:00:00 2001
From: Arthur Lu <learthurgo@gmail.com>
Date: Mon, 17 Feb 2025 22:04:23 +0000
Subject: [PATCH] add time stamp (in ms) to get response
---
app/main.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/app/main.py b/app/main.py
index 109bbba..7f22e25 100644
--- a/app/main.py
+++ b/app/main.py
@@ -4,6 +4,7 @@ from cache.cache import BaselineCache
from cache.prefetch_cache import PrefetchCache
from cache.tiered_cache import TieredCache
from config import CACHE_STRATEGY, CACHE_LIMIT, L2_CACHE_LIMIT
+import time
app = FastAPI()
@@ -20,16 +21,17 @@ else:
@app.get("/user/{user_id}")
def fetch_user_profile(user_id: str):
"""Fetch user profile with caching"""
+ start = time.time()
cached_profile = cache.get(user_id)
if cached_profile:
- return {"user_id": user_id, "profile": cached_profile, "source": "cache"}
+ return {"user_id": user_id, "profile": cached_profile, "source": "cache", "time_ms": (time.time() - start) * 1000}
profile = get_user_profile(user_id)
if profile is None:
raise HTTPException(status_code=404, detail="User not found")
cache.put(user_id, profile) # Store in cache
- return {"user_id": user_id, "profile": profile, "source": "database"}
+ return {"user_id": user_id, "profile": profile, "source": "database", "time_ms": (time.time() - start) * 1000}
@app.post("/update_user/")
def modify_user_profile(user_id: str, name: str, followers: int, bio: str, posts: str):