From f20bb54896a9ac7b65c20b0cc7b0a744d1741155 Mon Sep 17 00:00:00 2001 From: wenjun <861923274@qq.com> Date: Thu, 1 Apr 2021 20:34:49 +0800 Subject: [PATCH] [Improvement][API] Improvement query all project list (#5169) (#5192) --- .../api/service/impl/ProjectServiceImpl.java | 9 ++------- .../dolphinscheduler/api/service/ProjectServiceTest.java | 3 +-- .../dolphinscheduler/dao/mapper/ProjectMapper.java | 6 ++++++ .../apache/dolphinscheduler/dao/mapper/ProjectMapper.xml | 4 ++++ 4 files changed, 13 insertions(+), 9 deletions(-) diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectServiceImpl.java index c2e85cf612..6b25993098 100644 --- a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectServiceImpl.java +++ b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectServiceImpl.java @@ -447,19 +447,14 @@ public class ProjectServiceImpl extends BaseServiceImpl implements ProjectServic } /** - * query all project list that have one or more process definitions. + * query all project list * * @return project list */ @Override public Map queryAllProjectList() { Map result = new HashMap<>(); - List projects = new ArrayList<>(); - - List projectIds = processDefinitionMapper.listProjectIds(); - if (CollectionUtils.isNotEmpty(projectIds)) { - projects = projectMapper.selectBatchIds(projectIds); - } + List projects = projectMapper.queryAllProject(); result.put(Constants.DATA_LIST, projects); putMsg(result, Status.SUCCESS); diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ProjectServiceTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ProjectServiceTest.java index 3cdf7142e6..7b70206df8 100644 --- a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ProjectServiceTest.java +++ b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ProjectServiceTest.java @@ -322,8 +322,7 @@ public class ProjectServiceTest { @Test public void testQueryAllProjectList() { - Mockito.when(processDefinitionMapper.listProjectIds()).thenReturn(getProjectIds()); - Mockito.when(projectMapper.selectBatchIds(getProjectIds())).thenReturn(getList()); + Mockito.when(projectMapper.queryAllProject()).thenReturn(getList()); Map result = projectService.queryAllProjectList(); logger.info(result.toString()); diff --git a/dolphinscheduler-dao/src/main/java/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.java b/dolphinscheduler-dao/src/main/java/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.java index 872d53fd92..b84272cc89 100644 --- a/dolphinscheduler-dao/src/main/java/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.java +++ b/dolphinscheduler-dao/src/main/java/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.java @@ -92,4 +92,10 @@ public interface ProjectMapper extends BaseMapper { */ ProjectUser queryProjectWithUserByProcessInstanceId(@Param("processInstanceId") int processInstanceId); + /** + * query all project + * @return projectList + */ + List queryAllProject(); + } diff --git a/dolphinscheduler-dao/src/main/resources/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.xml b/dolphinscheduler-dao/src/main/resources/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.xml index a6d72d0d2d..74844d41a8 100644 --- a/dolphinscheduler-dao/src/main/resources/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.xml +++ b/dolphinscheduler-dao/src/main/resources/org/apache/dolphinscheduler/dao/mapper/ProjectMapper.xml @@ -109,4 +109,8 @@ join t_ds_user u on dp.user_id = u.id where di.id = #{processInstanceId}; + +