mirror of
https://github.com/ltcptgeneral/cs239-caching.git
synced 2025-04-04 05:23:26 +00:00
50b5ea0acd
add simulated 10ms delay to database, updated test script naming to better convey the experiment method
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
from fastapi import FastAPI, HTTPException
|
|
from database import get_user_ids, get_user_profile, update_user_profile, get_user_friend_ids
|
|
from cache.cache import BaselineCache
|
|
from cache.prefetch_cache import PrefetchCache
|
|
from cache.tiered_cache import TieredCache
|
|
from cache.eviction_seive import SeiveCache
|
|
from config import CACHE_STRATEGY, CACHE_LIMIT, L2_CACHE_LIMIT
|
|
import time
|
|
|
|
app = FastAPI()
|
|
|
|
# Initialize cache based on strategy from config.yaml or environment variable
|
|
if CACHE_STRATEGY == "Baseline":
|
|
print("Using baseline cache strategy")
|
|
cache = BaselineCache(limit=CACHE_LIMIT)
|
|
elif CACHE_STRATEGY == "Prefetch":
|
|
print("Using prefetch cache strategy")
|
|
cache = PrefetchCache(limit=CACHE_LIMIT)
|
|
elif CACHE_STRATEGY == "Tiered":
|
|
print("Using tiered cache strategy")
|
|
cache = TieredCache(limit=CACHE_LIMIT, l2_limit=L2_CACHE_LIMIT)
|
|
elif CACHE_STRATEGY == "Seive":
|
|
print("Using seive cache strategy")
|
|
cache = SeiveCache(limit=CACHE_LIMIT)
|
|
else:
|
|
raise ValueError(f"Invalid CACHE_STRATEGY: {CACHE_STRATEGY}")
|
|
|
|
@app.get("/users")
|
|
def fetch_user_ids():
|
|
return {"ids": get_user_ids()}
|
|
|
|
@app.get("/users_and_friends")
|
|
def fetch_user_and_friends():
|
|
return get_user_friend_ids()
|
|
|
|
@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", "time_ms": (time.time() - start) * 1000}
|
|
|
|
profile = get_user_profile(user_id)
|
|
time.sleep(10 / 1000) # simulate 10 ms db delay
|
|
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", "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, friends: list[str]):
|
|
"""Update user profile and refresh cache"""
|
|
update_user_profile(user_id, name, followers, bio, posts, friends)
|
|
cache.invalidate(user_id) # Invalidate old cache
|
|
return {"message": "User profile updated successfully"}
|