|
|
@ -51,31 +51,29 @@ public class ProjectService implements IProjectService{ |
|
|
|
List<ProjectVo.QueryProject> projectList = projectDao.queryProjectList(param.getStartTime(), param.getEndTime(), userId); |
|
|
|
//查询子项目
|
|
|
|
if (CollectionUtil.isNotEmpty(projectList)) { |
|
|
|
List<Long> projectIdList = projectList.stream().map(ProjectVo.QueryProject::getId).collect(Collectors.toList()); |
|
|
|
List<ProjectVo.QueryProject> sonProjectList = projectDao.querySonProject(projectIdList); |
|
|
|
if (CollectionUtil.isNotEmpty(sonProjectList)) { |
|
|
|
for (ProjectVo.QueryProject queryProject : projectList) { |
|
|
|
List<ProjectVo.QueryProject> sonList = new ArrayList<>(); |
|
|
|
for (ProjectVo.QueryProject sonProject : sonProjectList) { |
|
|
|
if (queryProject.getId().equals(sonProject.getParentId())){ |
|
|
|
sonList.add(sonProject); |
|
|
|
} |
|
|
|
querySunProject(projectList); |
|
|
|
} |
|
|
|
return projectList; |
|
|
|
} |
|
|
|
|
|
|
|
private void querySunProject(List<ProjectVo.QueryProject> projectList) { |
|
|
|
List<Long> projectIdList = projectList.stream().map(ProjectVo.QueryProject::getId).collect(Collectors.toList()); |
|
|
|
//查询子级
|
|
|
|
List<ProjectVo.QueryProject> sonProjectList = projectDao.querySonProject(projectIdList); |
|
|
|
if (CollectionUtil.isNotEmpty(sonProjectList)) { |
|
|
|
//若存在下一级,递归查询
|
|
|
|
querySunProject(sonProjectList); |
|
|
|
//将子级分配给父级
|
|
|
|
for (ProjectVo.QueryProject queryProject : projectList) { |
|
|
|
List<ProjectVo.QueryProject> sonList = new ArrayList<>(); |
|
|
|
for (ProjectVo.QueryProject sonProject : sonProjectList) { |
|
|
|
if (queryProject.getId().equals(sonProject.getParentId())){ |
|
|
|
sonList.add(sonProject); |
|
|
|
} |
|
|
|
queryProject.setSonProjectList(sonList); |
|
|
|
} |
|
|
|
queryProject.setSonProjectList(sonList); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// if (CollectionUtil.isNotEmpty(projectList)){
|
|
|
|
// for (ProjectVo.QueryProject project : projectList) {
|
|
|
|
// List<ProjectVo.QueryProject> sonProjectList = projectDao.querySonProject(project.getId());
|
|
|
|
// if (CollectionUtil.isNotEmpty(sonProjectList)){
|
|
|
|
// project.setSonProjectList(sonProjectList);
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
return projectList; |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|