Public/fastapi-cache
1
0
Fork 0
mirror of https://github.com/long2ice/fastapi-cache.git synced 2026-03-25 04:57:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #112 from schmocker/main

Browse Source 
add cache-control and etag to header of fist response
This commit is contained in:
long2ice
2023-01-15 12:27:56 +08:00
committed by GitHub
parent 62ef8bed37 334b829a80
commit 7a89f28b54
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
fastapi_cache/decorator.py
View File

@@ -132,9 +132,7 @@ def cache(
return coder.decode(ret)
ret = await ensure_async_func(*args, **kwargs)
try:
await backend.set(
cache_key, coder.encode(ret), expire or FastAPICache.get_expire()
)
await backend.set(cache_key, coder.encode(ret), expire)
except ConnectionError:
pass
return ret
@@ -154,11 +152,16 @@ def cache(
return coder.decode(ret)
ret = await ensure_async_func(*args, **kwargs)
encoded_ret = coder.encode(ret)
try:
await backend.set(cache_key, coder.encode(ret), expire or FastAPICache.get_expire())
await backend.set(cache_key, encoded_ret, expire)
except ConnectionError:
pass
response.headers["Cache-Control"] = f"max-age={expire}"
etag = f"W/{hash(encoded_ret)}"
response.headers["ETag"] = etag
return ret
return inner