refactor: replace FileLock with utility functions for locking mechanism

Author: liuruibin

apps/common/event/__init__.py

@@ -8,11 +8,9 @@
 """
 from django.utils.translation import gettext as _
 
-from common.lock.impl.file_lock import FileLock
 from .listener_manage import *
 from ..db.sql_execute import update_execute
 
-lock = FileLock()
 update_document_status_sql = """
                              UPDATE "public"."document"
                              SET status ="replace"("replace"("replace"(status, '1', '3'), '0', '3'), '4', '3')
@@ -23,7 +21,7 @@
 def run():
     from models_provider.models import Model, Status
 
-    if lock.try_lock('event_init', 30 * 30):
+    if try_lock('event_init', 30 * 30):
         try:
             # 修改Model状态为ERROR
             QuerySet(Model).filter(
@@ -34,4 +32,4 @@ def run():
             # 更新文档状态
             update_execute(update_document_status_sql, [])
         finally:
-            lock.un_lock('event_init')
+            un_lock('event_init')

apps/common/job/clean_chat_job.py

@@ -9,14 +9,13 @@
 from django_apscheduler.jobstores import DjangoJobStore
 from application.models import Application, Chat, ChatRecord
 from django.db.models import Q, Max
-from common.lock.impl.file_lock import FileLock
+from common.utils.lock import try_lock, un_lock
 from common.utils.logger import maxkb_logger
 
 from knowledge.models import File
 
 scheduler = BackgroundScheduler()
 scheduler.add_jobstore(DjangoJobStore(), "default")
-lock = FileLock()
 
 
 def clean_chat_log_job():
@@ -71,12 +70,12 @@ def clean_chat_log_job():
 
 
 def run():
-    if lock.try_lock('clean_chat_log_job', 30 * 30):
+    if try_lock('clean_chat_log_job', 30 * 30):
         try:
             scheduler.start()
             existing_job = scheduler.get_job(job_id='clean_chat_log')
             if existing_job is not None:
                 existing_job.remove()
             scheduler.add_job(clean_chat_log_job, 'cron', hour='0', minute='5', id='clean_chat_log')
         finally:
-            lock.un_lock('clean_chat_log_job')
+            un_lock('clean_chat_log_job')

apps/common/job/clean_debug_file_job.py

@@ -8,13 +8,12 @@
 from django.utils import timezone
 from django_apscheduler.jobstores import DjangoJobStore
 
-from common.lock.impl.file_lock import FileLock
+from common.utils.lock import un_lock, try_lock
 from common.utils.logger import maxkb_logger
 from knowledge.models import File
 
 scheduler = BackgroundScheduler()
 scheduler.add_jobstore(DjangoJobStore(), "default")
-lock = FileLock()
 
 
 def clean_debug_file():
@@ -27,12 +26,12 @@ def clean_debug_file():
 
 
 def run():
-    if lock.try_lock('clean_debug_file', 30 * 30):
+    if try_lock('clean_debug_file', 30 * 30):
         try:
             scheduler.start()
             clean_debug_file_job = scheduler.get_job(job_id='clean_debug_file')
             if clean_debug_file_job is not None:
                 clean_debug_file_job.remove()
             scheduler.add_job(clean_debug_file, 'cron', hour='2', minute='0', second='0', id='clean_debug_file')
         finally:
-            lock.un_lock('clean_debug_file')
+            un_lock('clean_debug_file')