AmineTMIRI/backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add cache to options watchlist

Browse Source
This commit is contained in:
MuslemRahimi 2024-09-24 00:31:41 +02:00
parent 000219599c
commit a77ae584dc
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
app/main.py
View File

@ -1187,9 +1187,19 @@ async def fetch_option_data(option_id: str):
@app.post("/get-options-watchlist") @app.post("/get-options-watchlist")
async def get_options_watchlist(data: OptionsWatchList, api_key: str = Security(get_api_key)): async def get_options_watchlist(data: OptionsWatchList, api_key: str = Security(get_api_key)):
options_list_id = sorted(data.optionsIdList)
cache_key = f"options-watchlist-{options_list_id}"
cached_result = redis_client.get(cache_key)
if cached_result:
return StreamingResponse(
io.BytesIO(cached_result),
media_type="application/json",
headers={"Content-Encoding": "gzip"}
)
result = [] result = []
for option_id in data.optionsIdList: for option_id in options_list_id:
file_path = OPTIONS_WATCHLIST_DIR / f"{option_id}.json" file_path = OPTIONS_WATCHLIST_DIR / f"{option_id}.json"
if file_path.exists(): if file_path.exists():
@ -1204,6 +1214,9 @@ async def get_options_watchlist(data: OptionsWatchList, api_key: str = Security(
result.extend(option_activity) result.extend(option_activity)
compressed_data = gzip.compress(orjson.dumps(result)) compressed_data = gzip.compress(orjson.dumps(result))
redis_client.set(cache_key, compressed_data)
redis_client.expire(cache_key, 60 * 30) # Set cache expiration time to 1 day
return StreamingResponse( return StreamingResponse(
io.BytesIO(compressed_data), io.BytesIO(compressed_data),
media_type="application/json", media_type="application/json",