sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Only remove the process lock from a process that created it

Browse source
This commit is contained in:
sabaimran 2024-05-08 10:14:52 -07:00
parent a50deb2762
commit 70d0ee4310
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/khoj/database/adapters/__init__.py
View file

@ -439,8 +439,8 @@ class ProcessLockAdapters:
return True
@staticmethod
def remove_process_lock(process_name: str):
return ProcessLock.objects.filter(name=process_name).delete()
def remove_process_lock(process_lock: ProcessLock):
return process_lock.delete()
@staticmethod
def run_with_lock(func: Callable, operation: ProcessLock.Operation, max_duration_in_seconds: int = 600, **kwargs):
@ -450,9 +450,10 @@ class ProcessLockAdapters:
return
success = False
process_lock = None
try:
# Set process lock
ProcessLockAdapters.set_process_lock(operation, max_duration_in_seconds)
process_lock = ProcessLockAdapters.set_process_lock(operation, max_duration_in_seconds)
logger.info(f"🔐 Locked {operation} to execute {func}")
# Execute Function
@ -467,8 +468,13 @@ class ProcessLockAdapters:
success = False
finally:
# Remove Process Lock
ProcessLockAdapters.remove_process_lock(operation)
logger.info(f"🔓 Unlocked {operation} process after executing {func} {'Succeeded' if success else 'Failed'}")
if process_lock:
ProcessLockAdapters.remove_process_lock(process_lock)
logger.info(
f"🔓 Unlocked {operation} process after executing {func} {'Succeeded' if success else 'Failed'}"
)
else:
logger.info(f"Skip removing {operation} process lock as it was not set")
def run_with_process_lock(*args, **kwargs):