|
@@ -156,6 +156,8 @@ public class ExcelTabServiceImpl extends BaseServiceImpl<ExcelTabMapper, ExcelTa
|
|
|
List<Long> ids = records.stream().map(ExcelTabVO::getId).collect(Collectors.toList());
|
|
|
List<ExcelTabVO> excelTabs = this.baseMapper.getByIds(ids);
|
|
|
Map<Long, ExcelTabVO> map = excelTabs.stream().collect(toMap(ExcelTabVO::getId, v -> v));
|
|
|
+ List<ExcelTabVO> projectInfoList = baseMapper.selectRootExcelTab(new Page<>(), null, ids);
|
|
|
+ Map<Long, List<ExcelTabVO>> collect = projectInfoList.stream().collect(Collectors.groupingBy(ExcelTabVO::getId));
|
|
|
records.forEach(v -> {
|
|
|
ProjectInfoVO1 projectInfoVO1 = new ProjectInfoVO1();
|
|
|
projectInfoVO1.setProjectId(v.getProjectId());
|
|
@@ -163,7 +165,18 @@ public class ExcelTabServiceImpl extends BaseServiceImpl<ExcelTabMapper, ExcelTa
|
|
|
if (map.get(v.getId()) != null) {
|
|
|
BeanUtil.copyProperties(map.get(v.getId()), v);
|
|
|
}
|
|
|
- v.setProjectInfoList(Collections.singletonList(projectInfoVO1));
|
|
|
+ List<ExcelTabVO> excelTabVOS = collect.get(v.getId());
|
|
|
+ if (excelTabVOS != null) {
|
|
|
+ List<ProjectInfoVO1> list = excelTabVOS.stream().map(vo -> {
|
|
|
+ ProjectInfoVO1 projectInfoVO2 = new ProjectInfoVO1();
|
|
|
+ projectInfoVO2.setProjectId(vo.getProjectId());
|
|
|
+ projectInfoVO2.setProjectName(vo.getProjectName());
|
|
|
+ return projectInfoVO2;
|
|
|
+ }).collect(Collectors.toList());
|
|
|
+ v.setProjectInfoList(list);
|
|
|
+ } else {
|
|
|
+ v.setProjectInfoList(Collections.singletonList(projectInfoVO1));
|
|
|
+ }
|
|
|
});
|
|
|
}
|
|
|
return page.setRecords(records);
|