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

redis_allocator/_version.py

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

redis_allocator/allocator.py

@@ -304,6 +304,11 @@ class RedisAllocatorPolicy(ABC, Generic[U]):
         """
         pass
 
+    @abstractmethod
+    def refresh_pool_all(self, allocator: 'RedisAllocator[U]'):
+        """Refresh the allocation pool."""
+        pass
+
     def check_health_once(self, r_obj: RedisAllocatorObject[U], duration: int = 3600) -> bool:
         """Check the health of the object."""
         with contextlib.closing(r_obj.open()):
@@ -405,6 +410,11 @@ class DefaultRedisAllocatorPolicy(RedisAllocatorPolicy[U]):
         self._update_lock_key = f"{allocator._pool_str()}|policy_update_lock"
         atexit.register(lambda: self.finalize(self._allocator()))
 
+    def refresh_pool_all(self, allocator: 'RedisAllocator[U]'):
+        allocator.clear()
+        for _ in range(self.updater.params):
+            self.refresh_pool(allocator)
+
     def malloc(self, allocator: 'RedisAllocator[U]', timeout: Timeout = 120,
                obj: Optional[U] = None, params: Optional[dict] = None,
                cache_timeout: Timeout = 3600) -> Optional[RedisAllocatorObject[U]]:
@@ -786,9 +796,9 @@ class RedisAllocator(RedisLockPool, Generic[U]):
                     push_to_tail(itemName, expiry)
                 end
             else
-                if locked then
-                    set_item_allocated(itemName, value)
-                end
+                -- if locked then
+                --     set_item_allocated(itemName, value)
+                -- end
             end
         end
 
@@ -824,6 +834,18 @@ class RedisAllocator(RedisLockPool, Generic[U]):
         end
         '''
 
+    @cached_property
+    def _clear_script(self):
+        return self.redis.register_script(f'''
+        {self._lua_required_string}
+        redis.call("DEL", poolItemsKey)
+        redis.call("SET", headKey, "")
+        redis.call("SET", tailKey, "")
+        ''')
+
+    def clear(self):
+        self._clear_script()
+
     @cached_property
     def _extend_script(self):
         """Cached Lua script to add or update keys in the pool.
@@ -966,11 +988,6 @@ class RedisAllocator(RedisLockPool, Generic[U]):
         """
         return f'{self.prefix}|{self.suffix}-cache'
 
-    def clear(self):
-        """Clear all resources from the allocation pool and cache."""
-        super().clear()
-        self.redis.delete(self._cache_str)
-
     def _soft_bind_name(self, name: str) -> str:
         """Get the Redis key for a soft binding.
 

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

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

{redis_allocator-0.4.6.dist-info → redis_allocator-0.5.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
 redis_allocator/__init__.py,sha256=TVjUm-8YEu_MQD_PkfeIKiVknpCJBrUY9cWN1LlaZcU,1016
-redis_allocator/_version.py,sha256=Kz7woA4tU2VuvwrlsLhgdzMyOrCqVahApsovYLgj_0M,22
-redis_allocator/allocator.py,sha256=8ReS-cWbAjuhMhlCbfr-FI6Q-pQO__VQWEk5I8O_MPM,52930
+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.4.6.dist-info/licenses/LICENSE,sha256=Wt4X1rHpffQfEiyWcDUx8BMLjXxfPqaiYZ7Lgsj7L4c,1068
+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.4.6.dist-info/METADATA,sha256=HvlGuWcNB30DLHHzgIRBgw6T3cUiycNHNCMefk4mWq0,21727
-redis_allocator-0.4.6.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-redis_allocator-0.4.6.dist-info/top_level.txt,sha256=0hXzU7sK5FCeSolTEYxThOt3HOybnwaXv1FLRJvHVgI,22
-redis_allocator-0.4.6.dist-info/RECORD,,
+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,,