ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-01-31 02:46:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support a more granular distributed lock wait

Browse Source
This commit is contained in:
allegroai 2024-06-20 17:57:54 +03:00
parent 9c210bb4fa
commit fcfc1e8998
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apiserver/sync.py
View File

@ -21,7 +21,7 @@ def distributed_lock(name: str, timeout: int, max_wait: int = 0):
max_wait = max_wait or timeout * 2 max_wait = max_wait or timeout * 2
pid = os.getpid() pid = os.getpid()
while _redis.set(lock_name, value=pid, ex=timeout, nx=True) is None: while _redis.set(lock_name, value=pid, ex=timeout, nx=True) is None:
sleep(1) sleep(0.1)
if time.time() - start > max_wait: if time.time() - start > max_wait:
holder = _redis.get(lock_name) holder = _redis.get(lock_name)
raise Exception(f"Could not acquire {name} lock for {max_wait} seconds. The lock is hold by {holder}") raise Exception(f"Could not acquire {name} lock for {max_wait} seconds. The lock is hold by {holder}")