使用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);
            }
        }
    }

你可能感兴趣的:(java基础,算法)