Fix cache key builder.

This commit is contained in:
long2ice
2021-01-06 10:34:30 +08:00
parent 3134e5f67c
commit a42fdaf632
5 changed files with 152 additions and 127 deletions

View File

@@ -30,9 +30,10 @@ def cache(
coder = coder or FastAPICache.get_coder()
expire = expire or FastAPICache.get_expire()
key_builder = key_builder or FastAPICache.get_key_builder()
request = kwargs.get("request")
request = kwargs.pop("request", None)
response = kwargs.pop("response", None)
backend = FastAPICache.get_backend()
cache_key = key_builder(func, namespace, *args, **kwargs)
cache_key = key_builder(func, namespace, args=args, kwargs=kwargs)
ttl, ret = await backend.get_with_ttl(cache_key)
if not request:
if ret is not None:

View File

@@ -7,10 +7,10 @@ from starlette.responses import Response
def default_key_builder(
func,
namespace: Optional[str] = "",
request: Request = None,
response: Response = None,
*args,
**kwargs,
request: Optional[Request] = None,
response: Optional[Response] = None,
args: Optional[tuple] = None,
kwargs: Optional[dict] = None,
):
from fastapi_cache import FastAPICache