redis-allocator 0.5.1__py3-none-any.whl → 0.6.0__py3-none-any.whl

redis_allocator/_version.py

@@ -1 +1 @@
-__version__ = '0.5.1'
+__version__ = '0.6.0'

redis_allocator/allocator.py

@@ -17,17 +17,16 @@ Key features:
 4. Support for an updater to refresh the pool's keys periodically
 5. Policy-based control of allocation behavior through RedisAllocatorPolicy
 """
+import random
 import atexit
 import logging
 import weakref
 import contextlib
 from abc import ABC, abstractmethod
-from typing import Any, Callable
+from typing import Any, Callable, Optional, TypeVar, Generic, Sequence, Iterable
 from functools import cached_property
 from threading import current_thread
 from concurrent.futures import ThreadPoolExecutor
-from typing import (Optional, TypeVar, Generic,
-                    Sequence, Iterable)
 from redis import StrictRedis as Redis
 from cachetools import cached, TTLCache
 from .lock import RedisLockPool, Timeout
@@ -412,8 +411,7 @@ class DefaultRedisAllocatorPolicy(RedisAllocatorPolicy[U]):
 
     def refresh_pool_all(self, allocator: 'RedisAllocator[U]'):
         allocator.clear()
-        for _ in range(self.updater.params):
-            self.refresh_pool(allocator)
+        self.refresh_pool(allocator, n=len(self.updater.params))
 
     def malloc(self, allocator: 'RedisAllocator[U]', timeout: Timeout = 120,
                obj: Optional[U] = None, params: Optional[dict] = None,
@@ -469,7 +467,7 @@ class DefaultRedisAllocatorPolicy(RedisAllocatorPolicy[U]):
             # If we got here, we acquired the lock, so we can update the pool
             self.refresh_pool(allocator)
 
-    def refresh_pool(self, allocator: 'RedisAllocator[U]'):
+    def refresh_pool(self, allocator: 'RedisAllocator[U]', shuffle=True, n=1):
         """Refresh the allocation pool using the updater.
 
         Args:
@@ -478,12 +476,17 @@ class DefaultRedisAllocatorPolicy(RedisAllocatorPolicy[U]):
         if self.updater is None:
             return
 
-        keys = self.updater()
+        keys = []
+        for _ in range(n):
+            keys.extend(self.updater())
 
         if len(keys) == 0:
             logger.warning("No keys to update to the pool")
             return
 
+        if shuffle:
+            random.shuffle(keys)
+
         # Update the pool based on the number of keys
         if len(self.updater) == 1:
             allocator.assign(keys, timeout=self.expiry_duration)

redis_allocator/lock.py

@@ -62,7 +62,6 @@ class BaseLock(ABC):
         Returns:
             The current status of the key.
         """
-        pass
 
     @abstractmethod
     def update(self, key: str, value='1', timeout: Timeout = 120):
@@ -73,7 +72,6 @@ class BaseLock(ABC):
             value: The value to set for the key.
             timeout: The lock timeout in seconds.
         """
-        pass
 
     @abstractmethod
     def lock(self, key: str, value: str = '1', timeout: Timeout = 120) -> bool:
@@ -87,7 +85,6 @@ class BaseLock(ABC):
         Returns:
             True if the ownership of the key is successfully acquired, False otherwise.
         """
-        pass
 
     @abstractmethod
     def is_locked(self, key: str) -> bool:
@@ -99,7 +96,6 @@ class BaseLock(ABC):
         Returns:
             True if the key is locked, False otherwise.
         """
-        pass
 
     @abstractmethod
     def lock_value(self, key: str) -> Optional[str]:
@@ -111,7 +107,6 @@ class BaseLock(ABC):
         Returns:
             The value of the key if the key is locked, None otherwise.
         """
-        pass
 
     @abstractmethod
     def rlock(self, key: str, value: str = '1', timeout=120) -> bool:
@@ -127,7 +122,6 @@ class BaseLock(ABC):
         Returns:
             True if the ownership of the key is successfully acquired, False otherwise.
         """
-        pass
 
     @abstractmethod
     def unlock(self, key: str) -> bool:
@@ -139,7 +133,6 @@ class BaseLock(ABC):
         Returns:
             True if the key is successfully released, False if the key is not locked.
         """
-        pass
 
     @abstractmethod
     def _conditional_setdel(self, op: str, key: str, value: float, set_value: Optional[float] = None,
@@ -157,7 +150,6 @@ class BaseLock(ABC):
         Returns:
             Whether the operation was successful.
         """
-        pass
 
     def setgt(self, key: str, value: float, set_value: Optional[float] = None, ex: Optional[int] = None) -> bool:
         """Sets a new value when the comparison value is greater than the current value."""
@@ -229,32 +221,26 @@ class BaseLockPool(BaseLock, metaclass=ABCMeta):
     @abstractmethod
     def extend(self, keys: Optional[Sequence[str]] = None):
         """Extend the pool with the specified keys."""
-        pass
 
     @abstractmethod
     def shrink(self, keys: Sequence[str]):
         """Shrink the pool by removing the specified keys."""
-        pass
 
     @abstractmethod
     def assign(self, keys: Optional[Sequence[str]] = None):
         """Assign keys to the pool, replacing any existing keys."""
-        pass
 
     @abstractmethod
     def clear(self):
         """Empty the pool."""
-        pass
 
     @abstractmethod
     def keys(self) -> Iterable[str]:
         """Get the keys in the pool."""
-        pass
 
     @abstractmethod
     def _get_key_lock_status(self, keys: Iterable[str]) -> Iterable[bool]:
         """Get the lock status of the specified keys."""
-        pass
 
     def values_lock_status(self) -> Iterable[bool]:
         """Get the lock status of all keys in the pool."""

{redis_allocator-0.5.1.dist-info → redis_allocator-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: redis-allocator
-Version: 0.5.1
+Version: 0.6.0
 Summary: Redis-based resource allocation system.
 Home-page: https://github.com/invoker-bot/RedisAllocator-python
 Author: Invoker Bot

redis_allocator-0.6.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+redis_allocator/__init__.py,sha256=TVjUm-8YEu_MQD_PkfeIKiVknpCJBrUY9cWN1LlaZcU,1016
+redis_allocator/_version.py,sha256=CBY3jsC-9HCm7eZ6CKD-sYLCejqOJ1pYWPQM4LGIXcI,22
+redis_allocator/allocator.py,sha256=DMB7roghIMlbPPMSE8dMgfX-u7R0hKEzx1vuIlmxi0A,53495
+redis_allocator/lock.py,sha256=pq1nVPJ4JT5vATF5-5rNZ4eplsBNeHXSxo0-aj-cxsE,27218
+redis_allocator/task_queue.py,sha256=8DjNr2uxhzCsHatV_CHOeGh7_K9pqQZFApSbe2blRO0,14989
+redis_allocator-0.6.0.dist-info/licenses/LICENSE,sha256=Wt4X1rHpffQfEiyWcDUx8BMLjXxfPqaiYZ7Lgsj7L4c,1068
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/conftest.py,sha256=Ts82uylQSzP_GcaN0E02o3xcFdjw20cXNzh3RAdYKW4,3967
+tests/test_allocator.py,sha256=TalMWgLhJw2NjwDMX5HJvIhDDmTGtSBvJjteOxCZatc,23365
+tests/test_lock.py,sha256=MDMRNN46VhWqkHUIhYOMEDgZkFFCW_WjwRLTOjkFF-Q,46952
+tests/test_task_queue.py,sha256=Fh5naikFajfOvL6GngEy_TPfOYCYZolZfVwtR6T4dTY,31710
+redis_allocator-0.6.0.dist-info/METADATA,sha256=xe0yY79O-I4DnojK2Nt2l8NI_8mcXl6NEMuWzyYhOEY,21727
+redis_allocator-0.6.0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+redis_allocator-0.6.0.dist-info/top_level.txt,sha256=0hXzU7sK5FCeSolTEYxThOt3HOybnwaXv1FLRJvHVgI,22
+redis_allocator-0.6.0.dist-info/RECORD,,

{redis_allocator-0.5.1.dist-info → redis_allocator-0.6.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.0)
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

tests/test_allocator.py

@@ -434,7 +434,7 @@ class TestRedisAllocator:
             allocator_with_policy.gc()  # some times should be called twice to remove the expired items
             redis_client.register_script("print('-------------------------------')")()
             print(self.get_redis_pool_state(allocator_with_policy, redis_client))
-            allocator_with_policy.policy.refresh_pool(allocator_with_policy)
+            allocator_with_policy.policy.refresh_pool(allocator_with_policy, shuffle=False)
             assert len(allocator_with_policy) == 4
             allocator_with_policy.gc()
             assert self.get_redis_pool_state(allocator_with_policy, redis_client) == self.generate_pool_state(

redis_allocator-0.5.1.dist-info/RECORD

@@ -1,15 +0,0 @@
-redis_allocator/__init__.py,sha256=TVjUm-8YEu_MQD_PkfeIKiVknpCJBrUY9cWN1LlaZcU,1016
-redis_allocator/_version.py,sha256=s8Yq9Om1yBxrMA7xYQ5Y13Paeuxnq99NxhyjuPlnH6A,22
-redis_allocator/allocator.py,sha256=TTjr8jcQpJt96RJkssRU60z0m1aUsqBlEud87W4DCkY,53414
-redis_allocator/lock.py,sha256=fqf6WUWHKYenEArWopMIF6kWEnDfADC-bZvnQImsQVo,27400
-redis_allocator/task_queue.py,sha256=8DjNr2uxhzCsHatV_CHOeGh7_K9pqQZFApSbe2blRO0,14989
-redis_allocator-0.5.1.dist-info/licenses/LICENSE,sha256=Wt4X1rHpffQfEiyWcDUx8BMLjXxfPqaiYZ7Lgsj7L4c,1068
-tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/conftest.py,sha256=Ts82uylQSzP_GcaN0E02o3xcFdjw20cXNzh3RAdYKW4,3967
-tests/test_allocator.py,sha256=zOBdaOhlpLsv63J4QZ9vOhL1k1gu_vO68x4CpPDa9_4,23350
-tests/test_lock.py,sha256=MDMRNN46VhWqkHUIhYOMEDgZkFFCW_WjwRLTOjkFF-Q,46952
-tests/test_task_queue.py,sha256=Fh5naikFajfOvL6GngEy_TPfOYCYZolZfVwtR6T4dTY,31710
-redis_allocator-0.5.1.dist-info/METADATA,sha256=yI2vLDxF_5JULEbkUZ5XkVsIriC8qEwtLjbGDmp6Neg,21727
-redis_allocator-0.5.1.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-redis_allocator-0.5.1.dist-info/top_level.txt,sha256=0hXzU7sK5FCeSolTEYxThOt3HOybnwaXv1FLRJvHVgI,22
-redis_allocator-0.5.1.dist-info/RECORD,,