perf: 支持在线会话暂停操作 (#11146)

* perf: 支持在线会话暂停操作

* perf: 优化代码

---------

Co-authored-by: Eric <xplzv@126.com>

apps/terminal/api/session/task.py

@@ -3,17 +3,20 @@
 import logging
 
 from rest_framework import status
+from rest_framework.decorators import action
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.views import APIView, Response
 
 from common.api import JMSBulkModelViewSet
+from common.const.http import POST
 from common.utils import get_object_or_none
 from orgs.utils import tmp_to_root_org
 from terminal import serializers
+from terminal.const import TaskNameType
 from terminal.models import Session, Task
 from terminal.utils import is_session_approver
 
-__all__ = ['TaskViewSet', 'KillSessionAPI', 'KillSessionForTicketAPI']
+__all__ = ['TaskViewSet', 'KillSessionAPI', 'KillSessionForTicketAPI', ]
 logger = logging.getLogger(__file__)
 
 
@@ -21,9 +24,44 @@ class TaskViewSet(JMSBulkModelViewSet):
     queryset = Task.objects.all()
     serializer_class = serializers.TaskSerializer
     filterset_fields = ('is_finished',)
+    serializer_classes = {
+        'create_toggle_task': serializers.LockTaskSessionSerializer,
+        'handle_ticket_task': serializers.LockTaskSessionSerializer,
+        'default': serializers.TaskSerializer
+    }
+    rbac_perms = {
+        "create_toggle_task": "terminal.terminate_session",
+    }
+
+    @action(methods=[POST], detail=False, url_path='toggle-lock-session')
+    def create_toggle_task(self, request, *args, **kwargs):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        session_id = serializer.validated_data['session_id']
+        task_name = serializer.validated_data['task_name']
+        session_ids = [session_id, ]
+        validated_session = create_sessions_tasks(session_ids, request.user, task_name=task_name)
+        return Response({"ok": validated_session})
+
+    @action(methods=[POST], detail=False, permission_classes=[IsAuthenticated, ],
+            url_path='toggle-lock-session-for-ticket', )
+    def handle_ticket_task(self, request, *args, **kwargs):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        session_id = serializer.validated_data['session_id']
+        task_name = serializer.validated_data['task_name']
+        session_ids = [session_id, ]
+        user_id = request.user.id
+        for session_id in session_ids:
+            if not is_session_approver(session_id, user_id):
+                return Response({}, status=status.HTTP_403_FORBIDDEN)
+        with tmp_to_root_org():
+            validated_session = create_sessions_tasks(session_ids, request.user, task_name=task_name)
+
+        return Response({"ok": validated_session})
 
 
-def kill_sessions(session_ids, user):
+def create_sessions_tasks(session_ids, user, task_name=TaskNameType.kill_session):
     validated_session = []
 
     for session_id in session_ids:
@@ -31,9 +69,10 @@ def kill_sessions(session_ids, user):
         if session and not session.is_finished:
             validated_session.append(session_id)
             Task.objects.create(
-                name="kill_session", args=session.id, terminal=session.terminal,
+                name=task_name, args=session.id, terminal=session.terminal,
                 kwargs={
-                    'terminated_by': str(user)
+                    'terminated_by': str(user),
+                    'created_by': str(user)
                 }
             )
     return validated_session
@@ -47,7 +86,7 @@ class KillSessionAPI(APIView):
 
     def post(self, request, *args, **kwargs):
         session_ids = request.data
-        validated_session = kill_sessions(session_ids, request.user)
+        validated_session = create_sessions_tasks(session_ids, request.user)
         return Response({"ok": validated_session})
 
 
@@ -63,6 +102,6 @@ class KillSessionForTicketAPI(APIView):
                 return Response({}, status=status.HTTP_403_FORBIDDEN)
 
         with tmp_to_root_org():
-            validated_session = kill_sessions(session_ids, request.user)
+            validated_session = create_sessions_tasks(session_ids, request.user)
 
         return Response({"ok": validated_session})