From ab426e98533edbf473edb6c139b113f733f8ab79 Mon Sep 17 00:00:00 2001
From: song-cc-rock <changchang.song@fit2cloud.com>
Date: Wed, 13 Nov 2024 11:07:47 +0800
Subject: [PATCH] =?UTF-8?q?refactor(=E5=B7=A5=E4=BD=9C=E5=8F=B0):=20?=
 =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=BE=85=E5=8A=9E=E7=BC=BA=E9=99=B7=E5=88=97?=
 =?UTF-8?q?=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../plugin/platform/spi/Platform.java         | 10 ++-
 .../bug/dto/request/BugPageRequest.java       |  9 +--
 .../bug/dto/request/BugTodoRequest.java       | 31 ++++++++
 .../metersphere/bug/mapper/ExtBugMapper.java  |  7 ++
 .../metersphere/bug/mapper/ExtBugMapper.xml   | 43 +++++++++--
 .../bug/service/BugCommonService.java         | 22 ++++++
 .../dashboard/controller/ToDoController.java  | 76 +++++++++++++++++--
 .../service/UserPlatformAccountService.java   |  8 ++
 8 files changed, 184 insertions(+), 22 deletions(-)
 create mode 100644 backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugTodoRequest.java

diff --git a/backend/framework/plugin/plugin-platform-sdk/src/main/java/io/metersphere/plugin/platform/spi/Platform.java b/backend/framework/plugin/plugin-platform-sdk/src/main/java/io/metersphere/plugin/platform/spi/Platform.java
index 89aa3e8317..9676c7d392 100644
--- a/backend/framework/plugin/plugin-platform-sdk/src/main/java/io/metersphere/plugin/platform/spi/Platform.java
+++ b/backend/framework/plugin/plugin-platform-sdk/src/main/java/io/metersphere/plugin/platform/spi/Platform.java
@@ -69,7 +69,7 @@ public interface Platform extends ExtensionPoint {
     List<SelectOption> getFormOptions(GetOptionRequest optionsRequest);
 
     /**
-     * 获取第三方平台缺陷状态选项
+     * 获取第三方平台缺陷状态流选项
      * @param projectConfig 项目配置信息
      * @param issueKey 缺陷ID
      * @param previousStatus 当前状态
@@ -78,6 +78,14 @@ public interface Platform extends ExtensionPoint {
      */
     List<SelectOption> getStatusTransitions(String projectConfig, String issueKey, String previousStatus) throws Exception;
 
+    /**
+     * 获取第三方平台缺陷状态流结束状态选项 (用于工作台过滤展示, 不实现则默认都展示第三方平台所有状态)
+     * @param projectConfig 项目配置信息(暂定, 可能其他平台后续需提供服务集成配置信息)
+     * @return 平台结束状态流选项
+     * @throws Exception 获取平台结束状态异常
+     */
+    List<SelectOption> getStatusTransitionsLastSteps(String projectConfig) throws Exception;
+
     /**
      * 获取第三方平台关联需求列表
      * @param request 需求分页查询参数
diff --git a/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugPageRequest.java b/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugPageRequest.java
index a2f282ce04..08b6d22c2b 100644
--- a/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugPageRequest.java
+++ b/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugPageRequest.java
@@ -17,12 +17,9 @@ public class BugPageRequest extends BasePageRequest {
     @NotBlank(message = "{bug.project_id.not_blank}")
     private String projectId;
 
-    @Schema(description = "是否回收站")
+    @Schema(description = "是否回收站, 后台默认设置")
     private boolean useTrash;
 
-    @Schema(description = "是否我的待办, 默认查询全部")
-    private boolean myTodo = false;
-
-    @Schema(description = "我的待办用户ID, 组合使用: myTodo=true, myTodoUserId=xxx")
-    private String myTodoUserId;
+    @Schema(description = "待办参数, 后台默认设置")
+    private BugTodoRequest todoParam;
 }
diff --git a/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugTodoRequest.java b/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugTodoRequest.java
new file mode 100644
index 0000000000..9be571a4c7
--- /dev/null
+++ b/backend/services/bug-management/src/main/java/io/metersphere/bug/dto/request/BugTodoRequest.java
@@ -0,0 +1,31 @@
+package io.metersphere.bug.dto.request;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Builder;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * @author song-cc-rock
+ * 缺陷待办参数
+ */
+@Data
+@Builder
+public class BugTodoRequest {
+
+	@Schema(description = "当前用户ID")
+	private String msUserId;
+
+	@Schema(description = "Local状态结束标识集合")
+	private List<String> msLastStepStatus;
+
+	@Schema(description = "当前对接平台; 非Local时需要")
+	private String currentPlatform;
+
+	@Schema(description = "当前对接平台用户ID")
+	private String platformUser;
+
+	@Schema(description = "对接平台状态结束标识集合")
+	private List<String> platformLastStatus;
+}
diff --git a/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.java b/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.java
index 8a9e0b4f2e..aadc4ac757 100644
--- a/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.java
+++ b/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.java
@@ -28,6 +28,13 @@ public interface ExtBugMapper {
     @BaseConditionFilter
     List<BugDTO> list(@Param("request") BugPageRequest request, @Param("sort") String sort);
 
+    /**
+     * 获取项目状态流结束标识
+     * @param projectId 项目ID
+     * @return 结束标识集合
+     */
+    List<String> getLocalLastStepStatusIds(@Param("projectId") String projectId);
+
     /**
      * 缺陷列表查询
      *
diff --git a/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.xml b/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.xml
index 6c92c0d0ea..8a19f2a1dd 100644
--- a/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.xml
+++ b/backend/services/bug-management/src/main/java/io/metersphere/bug/mapper/ExtBugMapper.xml
@@ -23,6 +23,11 @@
         </if>
     </select>
 
+    <select id="getLocalLastStepStatusIds" resultType="java.lang.String">
+        select id from status_item si left join status_definition sd on si.id = sd.status_id
+        where scope_type = 'PROJECT' and scope_id = #{projectId} and sd.definition_id = 'END'
+    </select>
+
     <select id="getIdsByPageRequest" resultType="java.lang.String">
         select b.id from bug b left join bug_content bc on b.id = bc.bug_id
         <include refid="queryWhereCondition"/>
@@ -157,13 +162,37 @@
                 or b.tags like concat('%', #{request.keyword},'%')
                 )
             </if>
-            <!-- 我的待办: Local平台&&非结束状态&&处理人为我; -->
-            <if test="request.myTodo">
-                and b.platform = 'Local' and b.status in (
-                    select id from status_item si left join status_definition sd on si.id = sd.status_id
-                    where scope_type = 'PROJECT' and scope_id = #{request.projectId}
-                    and (sd.definition_id != 'END' || sd.definition_id is null)
-                ) and b.handle_user = #{request.myTodoUserId}
+            <!-- 我的待办 -->
+            <if test="request.todoParam != null">
+                and (
+                <!-- Local平台: 非结束状态&&处理人为当前登录用户 -->
+                (
+                b.platform = 'Local'
+                <if test="request.todoParam.msLastStepStatus != null and request.todoParam.msLastStepStatus.size() > 0">
+                    and b.status not in
+                    <foreach collection="request.todoParam.msLastStepStatus" item="lastStatus" separator="," open="(" close=")">
+                        #{lastStatus}
+                    </foreach>
+                </if>
+                and b.handle_user = #{request.todoParam.msUserId}
+                )
+                <if test="request.todoParam.currentPlatform != null and request.todoParam.currentPlatform != 'Local'">
+                    <!-- 插件平台: 非平台结束状态&&处理人为个人中心账号配置 -->
+                    or
+                    (
+                    b.platform = #{request.todoParam.currentPlatform}
+                    <if test="request.todoParam.platformLastStatus != null and request.todoParam.platformLastStatus.size() > 0">
+                        and b.status not in
+                        <foreach collection="request.todoParam.platformLastStatus" item="lastStatus" separator="," open="(" close=")">
+                            #{lastStatus}
+                        </foreach>
+                    </if>
+                    <if test="request.todoParam.platformUser != null">
+                        and b.handle_user = #{request.todoParam.platformUser}
+                    </if>
+                    )
+                </if>
+                )
             </if>
             <include refid="filter"/>
             <include refid="combine">
diff --git a/backend/services/bug-management/src/main/java/io/metersphere/bug/service/BugCommonService.java b/backend/services/bug-management/src/main/java/io/metersphere/bug/service/BugCommonService.java
index 13d1c505e0..1115c3663e 100644
--- a/backend/services/bug-management/src/main/java/io/metersphere/bug/service/BugCommonService.java
+++ b/backend/services/bug-management/src/main/java/io/metersphere/bug/service/BugCommonService.java
@@ -43,6 +43,8 @@ public class BugCommonService {
 	@Resource
 	private FileService fileService;
 	@Resource
+	private ExtBugMapper extBugMapper;
+	@Resource
 	private BugStatusService bugStatusService;
 	@Resource
 	private BugCommentMapper bugCommentMapper;
@@ -189,4 +191,24 @@ public class BugCommonService {
 		List<SelectOption> allStatusOption = ListUtils.union(headerOptions, localOptions).stream().distinct().toList();
 		return allStatusOption.stream().collect(Collectors.toMap(SelectOption::getValue, SelectOption::getText));
 	}
+
+	/**
+	 * 获取Local状态流结束标识
+	 * @param projectId 项目ID
+	 * @return 状态流结束标识集合
+	 */
+	public List<String> getLocalLastStepStatus(String projectId) {
+		return extBugMapper.getLocalLastStepStatusIds(projectId);
+	}
+
+	/**
+	 * 获取平台状态流结束标识
+	 * @param projectId 项目ID
+	 * @return 状态流结束标识集合
+	 */
+	public List<String> getPlatformLastStepStatus(String projectId) throws Exception {
+		Platform platform = projectApplicationService.getPlatform(projectId, true);
+		List<SelectOption> platformLastSteps = platform.getStatusTransitionsLastSteps(projectApplicationService.getProjectBugThirdPartConfig(projectId));
+		return platformLastSteps.stream().map(SelectOption::getValue).toList();
+	}
 }
diff --git a/backend/services/dashboard/src/main/java/io/metersphere/dashboard/controller/ToDoController.java b/backend/services/dashboard/src/main/java/io/metersphere/dashboard/controller/ToDoController.java
index e5d52dc73d..912b44222d 100644
--- a/backend/services/dashboard/src/main/java/io/metersphere/dashboard/controller/ToDoController.java
+++ b/backend/services/dashboard/src/main/java/io/metersphere/dashboard/controller/ToDoController.java
@@ -3,19 +3,31 @@ package io.metersphere.dashboard.controller;
 import com.github.pagehelper.Page;
 import com.github.pagehelper.PageHelper;
 import io.metersphere.bug.dto.request.BugPageRequest;
+import io.metersphere.bug.dto.request.BugTodoRequest;
 import io.metersphere.bug.dto.response.BugDTO;
+import io.metersphere.bug.enums.BugPlatform;
+import io.metersphere.bug.service.BugCommonService;
 import io.metersphere.bug.service.BugService;
 import io.metersphere.functional.dto.CaseReviewDTO;
 import io.metersphere.functional.request.CaseReviewPageRequest;
 import io.metersphere.functional.service.CaseReviewService;
 import io.metersphere.plan.dto.request.TestPlanTableRequest;
 import io.metersphere.plan.dto.response.TestPlanResponse;
+import io.metersphere.plan.dto.response.TestPlanStatisticsResponse;
 import io.metersphere.plan.service.TestPlanManagementService;
+import io.metersphere.plan.service.TestPlanStatisticsService;
+import io.metersphere.plugin.platform.dto.SelectOption;
+import io.metersphere.project.service.ProjectApplicationService;
+import io.metersphere.sdk.util.LogUtils;
+import io.metersphere.system.domain.ServiceIntegration;
 import io.metersphere.system.security.CheckOwner;
+import io.metersphere.system.service.UserPlatformAccountService;
 import io.metersphere.system.utils.PageUtils;
 import io.metersphere.system.utils.Pager;
 import io.metersphere.system.utils.SessionUtils;
 import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.media.Schema;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.annotation.Resource;
 import org.apache.commons.lang3.StringUtils;
@@ -36,12 +48,20 @@ import java.util.List;
 @RequestMapping("/dashboard/todo")
 public class ToDoController {
 
-	@Resource
-	private CaseReviewService caseReviewService;
 	@Resource
 	private BugService bugService;
 	@Resource
+	private BugCommonService bugCommonService;
+	@Resource
+	private CaseReviewService caseReviewService;
+	@Resource
 	private TestPlanManagementService testPlanManagementService;
+	@Resource
+	private ProjectApplicationService projectApplicationService;
+	@Resource
+	private UserPlatformAccountService userPlatformAccountService;
+	@Resource
+	private TestPlanStatisticsService testPlanStatisticsService;
 
 	@PostMapping("/plan/page")
 	@Operation(summary = "我的待办-测试计划-列表分页查询")
@@ -52,14 +72,21 @@ public class ToDoController {
 		return testPlanManagementService.page(request);
 	}
 
+	@PostMapping("/plan/statistics")
+	@Operation(summary = "我的待办-测试计划-获取列表详情统计 {通过率, 执行进度}")
+	@Parameter(name = "ids", description = "计划ID集合", schema = @Schema(requiredMode = Schema.RequiredMode.REQUIRED))
+	public List<TestPlanStatisticsResponse> selectTestPlanMetricById(@RequestBody List<String> ids) {
+		return testPlanStatisticsService.calculateRate(ids);
+	}
+
 	@PostMapping("/review/page")
 	@Operation(summary = "我的待办-用例评审-列表分页查询")
 	@CheckOwner(resourceId = "#request.getProjectId()", resourceType = "project")
 	public Pager<List<CaseReviewDTO>> reviewPage(@Validated @RequestBody CaseReviewPageRequest request) {
-		Page<Object> page = PageHelper.startPage(request.getCurrent(), request.getPageSize(),
-				StringUtils.isNotBlank(request.getSortString()) ? request.getSortString() : "pos desc");
 		request.setMyTodo(true);
 		request.setMyTodoUserId(SessionUtils.getUserId());
+		Page<Object> page = PageHelper.startPage(request.getCurrent(), request.getPageSize(),
+				StringUtils.isNotBlank(request.getSortString()) ? request.getSortString() : "pos desc");
 		return PageUtils.setPageInfo(page, caseReviewService.getCaseReviewPage(request));
 	}
 
@@ -67,11 +94,44 @@ public class ToDoController {
 	@Operation(summary = "我的待办-缺陷-列表分页查询")
 	@CheckOwner(resourceId = "#request.getProjectId()", resourceType = "project")
 	public Pager<List<BugDTO>> bugPage(@Validated @RequestBody BugPageRequest request) {
-		Page<Object> page = PageHelper.startPage(request.getCurrent(), request.getPageSize(),
-				StringUtils.isNotBlank(request.getSortString()) ? request.getSortString() : "pos desc");
 		request.setUseTrash(false);
-		request.setMyTodo(true);
-		request.setMyTodoUserId(SessionUtils.getUserId());
+		request.setTodoParam(buildBugToDoParam(request, SessionUtils.getUserId(), SessionUtils.getCurrentOrganizationId()));
+		Page<Object> page = PageHelper.startPage(request.getCurrent(), request.getPageSize(),
+				StringUtils.isNotBlank(request.getSortString()) ? request.getSortString() : "b.pos desc");
 		return PageUtils.setPageInfo(page, bugService.list(request));
 	}
+
+	/**
+	 * 设置缺陷待办参数
+	 * @param request 请求参数
+	 * @param currentUserId 当前用户ID
+	 * @param currentOrgId 当前组织ID
+	 * @return 待办参数
+	 */
+	private BugTodoRequest buildBugToDoParam(BugPageRequest request, String currentUserId, String currentOrgId) {
+		List<String> msLastStepStatusIds = bugCommonService.getLocalLastStepStatus(request.getProjectId());
+		BugTodoRequest todoParam = BugTodoRequest.builder().msUserId(currentUserId).msLastStepStatus(msLastStepStatusIds).build();
+		try {
+			// 设置待办的平台参数
+			String platformName = projectApplicationService.getPlatformName(request.getProjectId());
+			if (StringUtils.equals(platformName, BugPlatform.LOCAL.getName())) {
+				return todoParam;
+			}
+			todoParam.setCurrentPlatform(platformName);
+			List<String> platformLastStepStatus = bugCommonService.getPlatformLastStepStatus(request.getProjectId());
+			todoParam.setPlatformLastStatus(platformLastStepStatus);
+			ServiceIntegration serviceIntegration = projectApplicationService.getPlatformServiceIntegrationWithSyncOrDemand(request.getProjectId(), true);
+			String platformUserName = userPlatformAccountService.getPlatformUserName(currentUserId, currentOrgId, serviceIntegration.getPluginId());
+			List<SelectOption> headerHandlerOption = bugCommonService.getHeaderHandlerOption(request.getProjectId());
+			if (StringUtils.isNotBlank(platformUserName)) {
+				headerHandlerOption.stream().filter(option -> StringUtils.containsAnyIgnoreCase(option.getText(), platformUserName))
+						.findFirst().ifPresent(option -> todoParam.setPlatformUser(option.getValue()));
+			}
+		} catch (Exception e) {
+			// 设置平台参数异常时, 无法正常过滤平台非结束的缺陷
+			LogUtils.error(e.getMessage());
+			return todoParam;
+		}
+		return todoParam;
+	}
 }
diff --git a/backend/services/system-setting/src/main/java/io/metersphere/system/service/UserPlatformAccountService.java b/backend/services/system-setting/src/main/java/io/metersphere/system/service/UserPlatformAccountService.java
index 136d6c8a66..3ef93126a3 100644
--- a/backend/services/system-setting/src/main/java/io/metersphere/system/service/UserPlatformAccountService.java
+++ b/backend/services/system-setting/src/main/java/io/metersphere/system/service/UserPlatformAccountService.java
@@ -132,4 +132,12 @@ public class UserPlatformAccountService {
         // noinspection unchecked
         return (Map<String, Object>) userPlatformInfo.get(pluginId);
     }
+
+    public String getPlatformUserName(String uerId, String orgId, String pluginId) {
+        Map<String, Object> pluginUserPlatformConfig = getPluginUserPlatformConfig(pluginId, orgId, uerId);
+        if (pluginUserPlatformConfig == null) {
+            return null;
+        }
+        return (String) pluginUserPlatformConfig.get("nickName");
+    }
 }