perf: 任务列表响应速度优化 #7489

gcloud/apigw/views/get_task_count.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+"""
+Tencent is pleased to support the open source community by making 蓝鲸智云PaaS平台社区版 (BlueKing PaaS Community
+Edition) available.
+Copyright (C) 2017 THL A29 Limited, a Tencent company. All rights reserved.
+Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+http://opensource.org/licenses/MIT
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+"""
+from apigw_manager.apigw.decorators import apigw_require
+from blueapps.account.decorators import login_exempt
+from django.db.models import Value
+from django.views.decorators.http import require_GET
+
+from gcloud import err_code
+from gcloud.apigw.decorators import mark_request_whether_is_trust, project_inject, return_json_response, timezone_inject
+from gcloud.apigw.forms import GetTaskListForm
+from gcloud.iam_auth.intercept import iam_intercept
+from gcloud.iam_auth.view_interceptors.apigw import ProjectViewInterceptor
+from gcloud.taskflow3.models import TaskFlowInstance
+
+
+@login_exempt
+@require_GET
+@apigw_require
+@return_json_response
+@mark_request_whether_is_trust
+@project_inject
+@timezone_inject
+@iam_intercept(ProjectViewInterceptor())
+def get_task_count(request, project_id):
+    project = request.project
+    params_validator = GetTaskListForm(data=request.GET)
+    if not params_validator.is_valid():
+        return {"result": False, "data": "", "message": params_validator.errors, "code": err_code.VALIDATION_ERROR.code}
+    param_mappings = {
+        "keyword": "pipeline_instance__name__contains",
+        "is_started": "pipeline_instance__is_started",
+        "is_finished": "pipeline_instance__is_finished",
+        "executor": "pipeline_instance__executor",
+        "create_method": "create_method",
+        "template_id": "template_id",
+    }
+    filter_kwargs = dict(is_deleted=Value(0), project_id=project.id)
+    for param, filter_key in param_mappings.items():
+        if param in request.GET:
+            filter_kwargs[filter_key] = params_validator.cleaned_data[param]
+
+    tasks = TaskFlowInstance.objects.select_related("pipeline_instance").filter(**filter_kwargs)
+
+    try:
+        count = tasks.count()
+    except Exception as e:
+        return {"result": False, "data": "", "message": e, "code": err_code.INVALID_OPERATION.code}
+
+    response = {
+        "result": True,
+        "data": {"count": count},
+        "code": err_code.SUCCESS.code,
+    }
+    return response

gcloud/apigw/views/utils.py

@@ -180,7 +180,10 @@ def paginate_list_data(request, queryset):
         limit = int(request.GET.get("limit", 100))
         # limit 最大数量为200
         limit = 200 if limit > 200 else limit
-        count = -1
+        if request.GET.get("no_count"):
+            count = -1
+        else:
+            count = queryset.count()
 
         if offset < 0 or limit < 0:
             raise Exception("offset and limit must be greater or equal to 0.")