redis-allocator 0.4.2__py3-none-any.whl → 0.4.3__py3-none-any.whl

redis_allocator/_version.py

@@ -1 +1 @@
-__version__ = '0.4.2'
+__version__ = '0.4.3'

redis_allocator/allocator.py

@@ -634,6 +634,7 @@ class RedisAllocator(RedisLockPool, Generic[U]):
         local tailKey      = pool_pointer_str(false)
         local function push_to_tail(itemName, expiry)  -- push the item to the free list
             local tail = redis.call("GET", tailKey)
+            local head = redis.call("GET", headKey)
             if not tail then
                 tail = ""
             end
@@ -643,11 +644,33 @@ class RedisAllocator(RedisLockPool, Generic[U]):
             else
                 local tailVal = redis.call("HGET", poolItemsKey, tail)
                 local prev, next, expiry = split_pool_value(tailVal)
-                assert(next == "", "tail is not the last item in the free list")
+                assert(next == "" or next == '#ALLOCATED', "tail is not the last item in the free list")
                 redis.call("HSET", poolItemsKey, tail, join_pool_value(prev, itemName, expiry))
             end
             redis.call("SET", tailKey, itemName)  -- set the tail point to the new item
         end
+        -- Ensure the new head node is well-formed (prev="") and update tail/head
+        local function set_item_head_nil(nextItemName)
+            if nextItemName == "" then
+                -- list becomes empty
+                redis.call("SET", headKey, "")
+                redis.call("SET", tailKey, "")
+                return
+            end
+
+            local nextVal = redis.call("HGET", poolItemsKey, nextItemName)
+            if not nextVal then
+                -- corrupted pointer, clear list
+                redis.call("SET", headKey, "")
+                redis.call("SET", tailKey, "")
+                return
+            end
+
+            local _prev, nextNext, nextExpiry = split_pool_value(nextVal)
+            if _prev ~= "" then
+                redis.call("HSET", poolItemsKey, nextItemName, join_pool_value("", nextNext, nextExpiry))
+            end
+        end
         local function pop_from_head()  -- pop the item from the free list
             local head = redis.call("GET", headKey)
             if not head or head == "" then  -- the free list is empty
@@ -660,29 +683,33 @@ class RedisAllocator(RedisLockPool, Generic[U]):
             if is_expiry_invalid(headExpiry) then  -- the item has expired
                 redis.call("HDEL", poolItemsKey, head)
                 redis.call("SET", headKey, headNext)
+                set_item_head_nil(headNext)
                 return pop_from_head()
-            end
-            if redis.call("EXISTS", key_str(head)) > 0 then  -- the item is locked
+            elseif redis.call("EXISTS", key_str(head)) > 0 then  -- the item is locked
                 redis.call("HSET", poolItemsKey, head, join_pool_value("#ALLOCATED", "#ALLOCATED", headExpiry))
                 redis.call("SET", headKey, headNext)
+                set_item_head_nil(headNext)
                 return pop_from_head()
-            end
-            local prev, next, expiry = split_pool_value(headVal)
-            if next == "" then  -- the item is the last in the free list
+            elseif headNext == "" then  -- the item is the last in the free list
                 redis.call("SET", headKey, "")
                 redis.call("SET", tailKey, "")
             else
-                local nextVal = redis.call("HGET", poolItemsKey, next)
+                local nextVal = redis.call("HGET", poolItemsKey, headNext)
                 local nextPrev, nextNext, nextExpiry = split_pool_value(nextVal)
-                redis.call("HSET", poolItemsKey, next, join_pool_value("", nextNext, nextExpiry))
-                redis.call("SET", headKey, next)
+                redis.call("HSET", poolItemsKey, headNext, join_pool_value("", nextNext, nextExpiry))
+                redis.call("SET", headKey, headNext)
+            end
+            redis.call("HSET", poolItemsKey, head, join_pool_value("#ALLOCATED", "#ALLOCATED", headExpiry))
+            -- If we removed the current head, update head pointer
+            local savedHead = redis.call("GET", headKey)
+            if savedHead == itemName then
+                redis.call("SET", headKey, next or "")
             end
-            redis.call("HSET", poolItemsKey, head, join_pool_value("#ALLOCATED", "#ALLOCATED", expiry))
             return head, headExpiry
         end
         local function set_item_allocated(itemName, val)
             if not val then
-                val = redis.call("HGET", poolItemsKey, itemName)
+                val = redis.call("HGET", pool_str(), itemName)
             end
             if val then
                 local prev, next, expiry = split_pool_value(val)
@@ -709,6 +736,22 @@ class RedisAllocator(RedisLockPool, Generic[U]):
                         redis.call("SET", tailKey, prev or "")
                     end
                     redis.call("HSET", poolItemsKey, itemName, join_pool_value("#ALLOCATED", "#ALLOCATED", expiry))
+                    -- If we removed the current head, update head pointer
+                    local savedHead = redis.call("GET", headKey)
+                    if savedHead == itemName then
+                        redis.call("SET", headKey, next or "")
+                    end
+                end
+            end
+            -- Invariant fix: ensure current head's prev pointer is always "".
+            local currentHead = redis.call("GET", headKey)
+            if currentHead and currentHead ~= "" then
+                local currentHeadVal = redis.call("HGET", poolItemsKey, currentHead)
+                if currentHeadVal then
+                    local curPrev, curNext, curExpiry = split_pool_value(currentHeadVal)
+                    if curPrev ~= "" then
+                        redis.call("HSET", poolItemsKey, currentHead, join_pool_value("", curNext, curExpiry))
+                    end
                 end
             end
         end
@@ -719,8 +762,9 @@ class RedisAllocator(RedisLockPool, Generic[U]):
             assert(value, "value should not be nil")
             local prev, next, expiry = split_pool_value(value)
             if is_expiry_invalid(expiry) then  -- Check if the item has expired
-                set_item_allocated(itemName)
+                set_item_allocated(itemName, value)
                 redis.call("HDEL", poolItemsKey, itemName)
+                return
             end
             local locked = redis.call("EXISTS", key_str(itemName)) > 0
             if prev == "#ALLOCATED" then
@@ -729,7 +773,7 @@ class RedisAllocator(RedisLockPool, Generic[U]):
                 end
             else
                 if locked then
-                    set_item_allocated(itemName)
+                    set_item_allocated(itemName, value)
                 end
             end
         end

{redis_allocator-0.4.2.dist-info → redis_allocator-0.4.3.dist-info}/METADATA

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

{redis_allocator-0.4.2.dist-info → redis_allocator-0.4.3.dist-info}/RECORD

@@ -1,15 +1,15 @@
 redis_allocator/__init__.py,sha256=TVjUm-8YEu_MQD_PkfeIKiVknpCJBrUY9cWN1LlaZcU,1016
-redis_allocator/_version.py,sha256=zrcTwxP3SA3F1Js78afrfeaZeJr8w2hLtFtEZoUqUmg,22
-redis_allocator/allocator.py,sha256=RL1opmUfi5OFDmCe43O_aK--st0Wfi55t4rK7YzT2S4,48889
+redis_allocator/_version.py,sha256=GI8jYEksCyU_GLvu0eW6pPFjDcmTuWNQVavfvCX2KuQ,22
+redis_allocator/allocator.py,sha256=WZ3UkdAIq8x-hywA_8jyQcPLRu17G-QgqRCdFjfo6tg,51029
 redis_allocator/lock.py,sha256=fqf6WUWHKYenEArWopMIF6kWEnDfADC-bZvnQImsQVo,27400
 redis_allocator/task_queue.py,sha256=8DjNr2uxhzCsHatV_CHOeGh7_K9pqQZFApSbe2blRO0,14989
-redis_allocator-0.4.2.dist-info/licenses/LICENSE,sha256=Wt4X1rHpffQfEiyWcDUx8BMLjXxfPqaiYZ7Lgsj7L4c,1068
+redis_allocator-0.4.3.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=_aoSDkzyjiUyzw6WoO2bMV1Ow9ymLOOMSJrp7TENU70,22139
 tests/test_lock.py,sha256=MDMRNN46VhWqkHUIhYOMEDgZkFFCW_WjwRLTOjkFF-Q,46952
 tests/test_task_queue.py,sha256=Fh5naikFajfOvL6GngEy_TPfOYCYZolZfVwtR6T4dTY,31710
-redis_allocator-0.4.2.dist-info/METADATA,sha256=2i_PpL_MD-TCX1yVddl9YObBPtatBrRQQNYImBAeB5M,21727
-redis_allocator-0.4.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-redis_allocator-0.4.2.dist-info/top_level.txt,sha256=0hXzU7sK5FCeSolTEYxThOt3HOybnwaXv1FLRJvHVgI,22
-redis_allocator-0.4.2.dist-info/RECORD,,
+redis_allocator-0.4.3.dist-info/METADATA,sha256=yHNCfaUB9TQQSYMkWWdnhmuI72tT5ESvod2SyCyuFNU,21727
+redis_allocator-0.4.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+redis_allocator-0.4.3.dist-info/top_level.txt,sha256=0hXzU7sK5FCeSolTEYxThOt3HOybnwaXv1FLRJvHVgI,22
+redis_allocator-0.4.3.dist-info/RECORD,,