/**
* 查询目录列表(多层级结构的树形数据)
*/
@Override
public List<NodeVO> findCatalogueList() {
List<NodeVO> rootNodeList = xxxMapper.findList();
List<NodeVO> resultList = new ArrayList<>();
// 先找到所有的一级节点
rootNodeList.forEach(node->{
int index = 1;
index += rootNodeList.stream().filter(rootNode -> node.getParentId().equals(rootNode.getId())).count();
if(index == 1){ //如果该parentId无相等的id,则index仍然为1,将一级节点加入最后结果List中
resultList.add(node);
}
});
// 递归给一级节点设置子节点
resultList.forEach(node->node.setChildNodeList(childHandle(node.getId(), rootNodeList)));
return resultList;
}
//递归获取子节点
private List<NodeVO> childHandle(Integer id, List<NodeVO> list) {
List<NodeVO> childList = new ArrayList<>();
// 遍历所有节点,如果该节点的父类id等于参数id,则作为孩子节点加入该节点中
list.forEach(node->{
if (node.getParentId().equals(id)) {
childList.add(node);
}
});
// 把孩子节点的孩子节点循环一遍
childList.forEach(node->node.setChildNodeList(childHandle(node.getId(), list)));
if (childList.size() == 0) { return null; }
return childList;
}