@@ -179,7 +179,7 @@ async def _ensure_semaphore_initialized(self) -> None:
179179 cls = type (self )
180180 assert cls .register_semaphore is not None # nosec
181181 await cls .register_semaphore ( # pylint: disable=not-callable
182- keys = [self .tokens_key , self .holders_key ],
182+ keys = [self .tokens_key , self .holders_set ],
183183 args = [self .capacity , ttl_seconds ],
184184 client = self .redis_client .redis ,
185185 )
@@ -260,7 +260,7 @@ async def _try_acquire() -> list[str] | None:
260260 cls = type (self )
261261 assert cls .acquire_script is not None # nosec
262262 result = await cls .acquire_script ( # pylint: disable=not-callable
263- keys = [self .holders_key , self .holder_key ],
263+ keys = [self .holders_set , self .holder_key ],
264264 args = [
265265 token ,
266266 self .instance_id ,
@@ -297,7 +297,7 @@ async def release(self) -> None:
297297 cls = type (self )
298298 assert cls .release_script is not None # nosec
299299 result = await cls .release_script ( # pylint: disable=not-callable
300- keys = [self .tokens_key , self .holders_key , self .holder_key ],
300+ keys = [self .tokens_key , self .holders_set , self .holder_key ],
301301 args = [self .instance_id ],
302302 client = self .redis_client .redis ,
303303 )
@@ -342,7 +342,7 @@ async def reacquire(self) -> None:
342342 cls = type (self )
343343 assert cls .renew_script is not None # nosec
344344 result = await cls .renew_script ( # pylint: disable=not-callable
345- keys = [self .holders_key , self .holder_key ],
345+ keys = [self .holders_set , self .holder_key ],
346346 args = [self .instance_id , ttl_seconds ],
347347 client = self .redis_client .redis ,
348348 )
@@ -383,7 +383,7 @@ async def is_acquired(self) -> bool:
383383 async def get_current_count (self ) -> int :
384384 """Get the current number of semaphore holders"""
385385 return await handle_redis_returns_union_types (
386- self .redis_client .redis .scard (self .holders_key )
386+ self .redis_client .redis .scard (self .holders_set )
387387 )
388388
389389 async def get_available_count (self ) -> int :
0 commit comments