perf: 终断批量快捷命令执行的任务

This commit is contained in:
wangruidong
2024-02-19 14:47:32 +08:00
committed by Bryan
parent f592f19b08
commit 135fb7c6f9
4 changed files with 42 additions and 1 deletions

View File

@@ -16,7 +16,7 @@ from common.const.http import POST
from common.permissions import IsValidUser
from ops.const import Types
from ops.models import Job, JobExecution
from ops.serializers.job import JobSerializer, JobExecutionSerializer, FileSerializer
from ops.serializers.job import JobSerializer, JobExecutionSerializer, FileSerializer, JobTaskStopSerializer
__all__ = [
'JobViewSet', 'JobExecutionViewSet', 'JobRunVariableHelpAPIView',
@@ -187,6 +187,17 @@ class JobExecutionViewSet(OrgBulkModelViewSet):
queryset = queryset.filter(creator=self.request.user)
return queryset
@action(methods=[POST], detail=False, serializer_class=JobTaskStopSerializer, permission_classes=[IsValidUser, ],
url_path='stop')
def stop(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
if not serializer.is_valid():
return Response({'error': serializer.errors}, status=400)
task_id = serializer.validated_data['task_id']
instance = get_object_or_404(JobExecution, task_id=task_id, creator=request.user)
instance.stop()
return Response({'task_id': task_id}, status=200)
class JobAssetDetail(APIView):
rbac_perms = {