refactor: update cache decorator to use create_with method for TTLMapAsyncStorage

Author: zoola969

cachium/_decorators.py

@@ -74,7 +74,7 @@ def cache(
         >>> import asyncio
         >>> from cachium import cache
         >>> from cachium.storages.ttl_map import TTLMapAsyncStorage
-        >>> @cache(storage=lambda: TTLMapAsyncStorage())
+        >>> @cache(storage=TTLMapAsyncStorage.create_with())
         ... async def add_async(a: int, b: int) -> int:
         ...     return a + b
         >>> asyncio.run(add_async(1, 2))

cachium/storages/ttl_map.py

@@ -153,7 +153,7 @@ def create_with(
         max_size: int | None = 1000,
         ttl: timedelta | None = timedelta(minutes=1),
     ) -> Callable[..., Self]:
-        """Return a callable that creates a new instance of `TTLMapAsyncStorage` with the specified parameters."""
+        """Return a callable that creates a new instance of `TTLMapStorage` with the specified parameters."""
         return lambda: cls(max_size=max_size, ttl=ttl)
 
     @override

docs/guides/quickstart.md

@@ -32,7 +32,7 @@ from py_cashier import cache
 from py_cashier.storages.ttl_map import TTLMapAsyncStorage
 
 # Configure TTL and max size for async storage
-@cache(storage=lambda: TTLMapAsyncStorage(max_size=512, ttl=timedelta(seconds=30)))
+@cache(storage=TTLMapAsyncStorage.create_with(max_size=512, ttl=timedelta(seconds=30)))
 async def add_async(a: int, b: int) -> int:
     # Simulate I/O
     await asyncio.sleep(0.1)

tests/test__decorator.py

@@ -58,7 +58,7 @@ async def test_cache_async(a: int, b: int, expected: int) -> None:
     """Test that the cache decorator works correctly for asynchronous functions."""
     calls: dict[tuple[int, int], int] = defaultdict(int)
 
-    @cache(storage=lambda: TTLMapAsyncStorage(max_size=1000, ttl=timedelta(seconds=1)))
+    @cache(storage=TTLMapAsyncStorage.create_with(max_size=1000, ttl=timedelta(seconds=1)))
     async def func(x: int, y: int) -> int:
         nonlocal calls
         calls[(x, y)] += 1
@@ -91,7 +91,7 @@ async def test_cache_dog_piling_async(a: int, b: int, expected: int, tasks_count
     """
     calls = 0
 
-    @cache(storage=lambda: TTLMapAsyncStorage(max_size=1000, ttl=timedelta(seconds=1)))
+    @cache(storage=TTLMapAsyncStorage.create_with(max_size=1000, ttl=timedelta(seconds=1)))
     async def func(x: int, y: int) -> int:
         await asyncio.sleep(0.1)
         nonlocal calls
@@ -192,7 +192,7 @@ async def test_cache_failing_func_async(a: int, b: int, tasks_count: int) -> Non
     """Test that failing functions are not cached in asynchronous context."""
     calls = 0
 
-    @cache(storage=lambda: TTLMapAsyncStorage(max_size=1000, ttl=timedelta(seconds=1)))
+    @cache(storage=TTLMapAsyncStorage.create_with(max_size=1000, ttl=timedelta(seconds=1)))
     async def func(x: int, y: int) -> int:
         await asyncio.sleep(0.1)
         nonlocal calls
@@ -227,7 +227,7 @@ def _f() -> None:
     ("func", "storage"),
     [
         (_af, TTLMapStorage.create_with(max_size=1000, ttl=timedelta(seconds=1))),
-        (_f, lambda: TTLMapAsyncStorage(max_size=1000, ttl=timedelta(seconds=1))),
+        (_f, TTLMapAsyncStorage.create_with(max_size=1000, ttl=timedelta(seconds=1))),
     ],
 )
 def test__decorator__invalid_storage(func: Callable[..., Any], storage: PStorage | PAsyncStorage) -> None: