github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-22 20:09:54 +00:00
Code Issues Projects Releases Wiki Activity

perf: 修复事务中任务执行

Browse Source
This commit is contained in:
ibuler
2023-09-21 14:28:34 +08:00
committed by 老广
parent b023ca0c69
commit cdbe5d31e9
7 changed files with 13 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
apps/terminal/api/applet/host.py
View File

@@ -1,6 +1,5 @@
import uuid
from django.db import transaction
from rest_framework import status
from rest_framework import viewsets
from rest_framework.decorators import action
@@ -58,17 +57,13 @@ class AppletHostDeploymentViewSet(viewsets.ModelViewSet):
('applets', 'terminal.view_AppletHostDeployment'),
)
@staticmethod
def start_deploy(instance):
task = run_applet_host_deployment.apply_async((instance.id,), task_id=str(instance.id))
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
instance = serializer.save()
task = run_applet_host_deployment.delay(instance.id)
instance.save_task(instance.id)
transaction.on_commit(lambda: self.start_deploy(instance))
return Response({'task': str(instance.id)}, status=201)
return Response({'task': str(task.id)}, status=201)
@action(methods=['post'], detail=False)
def applets(self, request, *args, **kwargs):
@@ -84,10 +79,5 @@ class AppletHostDeploymentViewSet(viewsets.ModelViewSet):
applet_host_deployment_ids = [str(obj.id) for obj in applet_host_deployments]
task_id = str(uuid.uuid4())
model.objects.filter(id__in=applet_host_deployment_ids).update(task=task_id)
transaction.on_commit(lambda: self.start_install_applet(applet_host_deployment_ids, applet_id, task_id))
run_applet_host_deployment_install_applet.delay(applet_host_deployment_ids, applet_id)
return Response({'task': task_id}, status=201)
@staticmethod
def start_install_applet(applet_host_deployment_ids, applet_id, task_id):
run_applet_host_deployment_install_applet.apply_async((applet_host_deployment_ids, applet_id),
task_id=str(task_id))