/**
* 给树形结构中特定字段赋值的方法
* @return 返回的是一个赋值完成的树形结构
*/
public List<TreePojoDTO> getList() {
//1.先取出树形结构的所有数据
List<TreePojoDTO> TreePojoDTOs = treePojoDTOMapper.selectTree("对应的查询条件");
//包含特定值的map
Map<String, String> finalHashMap = new HashMap<>();
//2.从集合中过滤出最高等级标识 -1为没有父级(即为最高等级)
List<TreePojoDTO> list = TreePojoDTOs.stream().filter(
TreePojoDTO -> "-1".equals(TreePojoDTO.getPTreeId())).
//.map 返回一个流
map((treePojoDTO)->{
//3.设置最外层 dto分数 可子项值
treePojoDTO.setChildrenList(getChildRateRisk(treePojoDTO,TreePojoDTOs, finalHashMap));
//4.设置最外层 dto 特定字段
treePojoDTO.setSpecificValue(finalHashMap.get(treePojoDTO.getTreeId()));
return treePojoDTO;
//sorted 根据指定字段进行排序 固定前端展示用·
}).sorted(Comparator.comparing(TreePojoDTO::getTreeId)).collect(Collectors.toList());
return list;
}
private List<TreePojoDTO> getChildRateRisk(TreePojoDTO rrpRateRiskEle, List<TreePojoDTO> rateRiskEleList,Map<String,String> HashMap) {
List<TreePojoDTO> children = rateRiskEleList.stream().filter(
tree -> tree.getPTreeId().equals(rrpRateRiskEle.getTreeId())
).map(tree -> {
tree.setSpecificValue(HashMap.get(tree.getTreeId()));
tree.setChildrenList(getChildRateRisk(tree,rateRiskEleList,HashMap));
return tree;
}).sorted(Comparator.comparing(TreePojoDTO::getTreeId)).collect(Collectors.toList());
return children;
}