github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

fix: celery事物 数据库未保存

pull/12119/head
feng 2023-11-14 19:30:15 +08:00 committed by Bryan
parent 7c51d90a3d
commit 8667943443
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
apps/ops/signal_handlers.py
View File

@ -1,4 +1,5 @@
import ast import ast
import time
from celery import signals from celery import signals
from django.core.cache import cache from django.core.cache import cache
@ -82,9 +83,16 @@ def before_task_publish(body=None, **kwargs):
@signals.task_prerun.connect @signals.task_prerun.connect
def on_celery_task_pre_run(task_id='', kwargs=None, **others): def on_celery_task_pre_run(task_id='', kwargs=None, **others):
count = 0
qs = CeleryTaskExecution.objects.filter(id=task_id)
while not qs.exists() and count < 5:
count += 1
time.sleep(1)
qs = CeleryTaskExecution.objects.filter(id=task_id)
# 更新状态 # 更新状态
CeleryTaskExecution.objects.filter(id=task_id) \ qs.update(state='RUNNING', date_start=timezone.now())
.update(state='RUNNING', date_start=timezone.now())
# 关闭之前的数据库连接 # 关闭之前的数据库连接
close_old_connections() close_old_connections()