Skip to content

Commit 3e3576e

Browse files
committed
cleanup
1 parent c34f003 commit 3e3576e

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

packages/service-library/src/servicelib/redis/_semaphore.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)