Merge pull request #26 from monica-veloso/moni/fix-call-with-lock

Moni/fix call with lock

Author: monica-veloso

src/yapcache/__about__.py

@@ -1 +1 @@
-__version__ = "0.1.21"
+__version__ = "0.1.22"

src/yapcache/__init__.py

@@ -55,7 +55,8 @@ def decorator(fn: Callable[P, Awaitable[R]]):
         async def wrapper(*args, **kwargs):
             key = cache_key(*args, **kwargs)
 
-            async def _call_with_lock():
+            async def _call_with_lock(*args, **kwargs):
+                key = cache_key(*args, **kwargs)
                 async with lock(key + ":lock"):
                     found = await cache.get(
                         key
@@ -80,7 +81,7 @@ async def _call_with_lock():
             if isinstance(found, CacheItem):
                 status = CacheStatus.HIT
                 if found.is_stale and key not in update_tasks:
-                    task = asyncio.create_task(_call_with_lock())
+                    task = asyncio.create_task(_call_with_lock(*args, **kwargs))
                     update_tasks[key] = task  # TODO: acho que tem problema
                     task.add_done_callback(lambda _: update_tasks.pop(key))
                     status = CacheStatus.STALE