klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 05:40:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
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):