|
@@ -1294,9 +1294,11 @@ public class UserServiceImpl extends BaseServiceImpl<UserMapper, User> implement
|
|
|
|
|
|
if (nodesAll.size() > 0) {
|
|
|
//判断是否有子级,赋值
|
|
|
- Map<Long, List<WbsTreeContractLazyVO>> groupedByParentId = nodesAll.stream().collect(Collectors.groupingBy(WbsTreeContractLazyVO::getParentId));
|
|
|
+ Map<Long, List<WbsTreeContractLazyVO>> groupedByParentId = nodesAll.stream()
|
|
|
+ .filter(vo -> vo.getParentId() != null)
|
|
|
+ .collect(Collectors.groupingBy(WbsTreeContractLazyVO::getParentId));
|
|
|
for (WbsTreeContractLazyVO vo : nodesAll) {
|
|
|
- if (vo.getParentId() == 0) {
|
|
|
+ if (vo.getParentId()!=null&&vo.getParentId() == 0) {
|
|
|
vo.setHasChildren(1);
|
|
|
}
|
|
|
List<WbsTreeContractLazyVO> childNodes = groupedByParentId.getOrDefault(vo.getId(), null);
|
|
@@ -1466,8 +1468,8 @@ public class UserServiceImpl extends BaseServiceImpl<UserMapper, User> implement
|
|
|
*/
|
|
|
public List<NodeVO> buildNodeTreeByStream(List<WbsTreeContractLazyVO> distinctNodesAll,
|
|
|
Map<Long, WbsTreeContractLazyVO> lowestNodesMap) {
|
|
|
- List<WbsTreeContractLazyVO> list = distinctNodesAll.stream().filter(f -> f.getParentId().equals(0L)).collect(Collectors.toList());
|
|
|
- Map<Long, List<WbsTreeContractLazyVO>> map = distinctNodesAll.stream().collect(Collectors.groupingBy(WbsTreeContractLazyVO::getParentId));
|
|
|
+ List<WbsTreeContractLazyVO> list = distinctNodesAll.stream().filter(f->f.getParentId()!=null).filter(f -> f.getParentId().equals(0L)).collect(Collectors.toList());
|
|
|
+ Map<Long, List<WbsTreeContractLazyVO>> map = distinctNodesAll.stream().filter(f->f.getParentId()!=null).collect(Collectors.groupingBy(WbsTreeContractLazyVO::getParentId));
|
|
|
return recursionFnNodeTree(list, map, lowestNodesMap);
|
|
|
}
|
|
|
|