perf: db connection close if needt

This commit is contained in:
ibuler
2025-05-14 14:29:14 +08:00
committed by 老广
parent a72e6456d9
commit e9ff988d8c
5 changed files with 7 additions and 6 deletions

View File

@@ -115,11 +115,10 @@ def on_celery_task_pre_run(task_id='', kwargs=None, **others):
@signals.task_postrun.connect
def on_celery_task_post_run(task_id='', state='', **kwargs):
close_old_connections()
CeleryTaskExecution.objects.filter(id=task_id).update(
state=state, date_finished=timezone.now(), is_finished=True
)
close_old_connections()
@signals.after_task_publish.connect