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

Simplify key_builder calling

Browse Source 
The keybuilder is either returning a string, or a coroutine or other
awaitable. If the latter, await on the return value to get the string.
This commit is contained in:
Martijn Pieters
2023-04-27 16:29:10 +01:00
parent 255f40117b
commit a52f6b1406
1 changed files with 11 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
fastapi_cache/decorator.py
View File

@@ -116,24 +116,17 @@ def cache(
key_builder = key_builder or FastAPICache.get_key_builder()
backend = FastAPICache.get_backend()
if inspect.iscoroutinefunction(key_builder):
cache_key = await key_builder(
func,
namespace,
request=request,
response=response,
args=args,
kwargs=copy_kwargs,
)
else:
cache_key = key_builder(
func,
namespace,
request=request,
response=response,
args=args,
kwargs=copy_kwargs,
)
cache_key = key_builder(
func,
namespace,
request=request,
response=response,
args=args,
kwargs=copy_kwargs,
)
if inspect.isawaitable(cache_key):
cache_key = await cache_key
try:
ttl, ret = await backend.get_with_ttl(cache_key)
except Exception: