使用stream流构造树形结构
日常使用for循环构造树形结构,但是在遇到数据量较大的时候,使用stream流可以提高效率。所以在此展示如何使用stream流构造树形结构。
一、第一步先找出顶层节点(顶层节点的特点就是他的父节点为空,根据此条件先筛选出顶层节点。)
List<ExportOrganization> parentList = tempList.stream()
.filter(e -> StringUtils.isBlank(e.getPcompId())
|| "".equals(e.getPcompId()))
.collect(Collectors.toList());
二、第二步依次遍历全部节点构造树形结构(其中的重点是
if (!subList.isEmpty()) {
getSubList(subList, all);
}
当子节点不为空,需要递归遍历子节点再次去构造树形结构
private static void getSubList(final List<ExportOrganization> parentList, final List<ExportOrganization> all) {
for (ExportOrganization exportOrganization : parentList) {
List<ExportOrganization> subList = all.stream().filter(o -> o.getPcompId()
.equals(exportOrganization.getCompId())).collect(Collectors.toList());
exportOrganization.setSubList(subList);
if (!subList.isEmpty()) {
getSubList(subList, all);
}
}
}