Java使用 Stream 流和 Lambda 组装复杂父子树形结构

@Data
public class Department {
    /**
     * 主键Id
     */
    private Integer id;
    /**
     * 该部门的父部门Id
     */
    private Integer parentDeptId;
    /**
     * 真正部门Id
     */
    private Integer deptId;
    /**
     * 部门的名称
     */
    private String name;
    /**
     * 部门在结构中所处的层级
     */
    private Integer level;
    /**
     * 状态是否启用
     */
    private Integer status;
}

@Data
public class DepartmentVO implements Serializable {
    /**
     * 子节点集合，封装自己
     */
    private List<DepartmentVO> childrenList;
    /**
     * 部门Id
     */
    protected Integer deptId;
    /**
     * 父部门Id
     */
    protected Integer parentDeptId;
    /**
     * 部门名称
     */
    protected String name;
}

    @Override
    public List<DepartmentVO> departmentStructure(String id){
            //step1：这里 map 只是简单转换了返回的对象属性（返回需要的类型），本质还是所有部门数据
            List<DepartmentVO> departmentVOList = this.getDepartmentListById(id).stream()
                    .map(e -> e.copyProperties(DepartmentVO.class))
                    .collect(Collectors.toList());
            //step2：利用父节点分组，所有部门的父 Id 进行分组，把所有的子节点 List 集合都找出来并一层层分好组
            Map<Integer, List<DepartmentVO>> departmentListMap = departmentVOList.stream()
                    .collect(Collectors.groupingBy(DepartmentVO::getParentDeptId));
            //step3：关键一步，关联上子部门，将子部门的 List 集合经过遍历一层层地放置好，最终会得到完整的部门父子关系 List 集合
            departmentVOList.forEach(e -> e.setChildrenList(departmentListMap.get(e.getDeptId())));
            //step4：过滤出顶级部门，即所有的子部门数据都归属于一个顶级父部门 Id
            List<DepartmentVO> resultList = departmentVOList.stream()
                    .filter(e -> Constants.TOP_DEPARTMENT_NUM.equals(e.getParentDeptId()))
                    .collect(Collectors.toList());
        return Optional.of(resultList).orElse(null);
    }

@Data
public class Area {
    /**
     * 地区id
     */
    public Long id;
    /**
     * 父Id
     */
    public Long parentId;
    /**
     * 地区名称
     */
    public String name;
    /**
     * 所属省Id
     */
    public Long provinceId;
    /**
     * 所属地级市Id
     */
    public Long cityId;
    /**
     * 所处层级
     */
    public Integer level;
}

@Data
public class AreaVO {
    /**
     * 子节点 list 集合
     */
    private List<AreaVO> childrenAreaVOList;
    /**
     * 区域id
     */
    public Long id;
    /**
     * 地区名称
     */
    public String name;
    /**
     * 所处层级
     */
    public Integer level;
    /**
     * 父Id
     */
    public Long parentId;
    /**
     * 所属省Id
     */
    public Long provinceId;
    /**
     * 所属地级市Id
     */
    public Long cityId;
}

    @Override
    public List<AreaVO> getAreaStructure() {
        //第一步，从数据库中查出所有数据，按照排序条件进行排序，本质上还是这个所有数据的 List 集合
        List<AreaVO> areaVOList = this.findAll(Sort.by("id").descending()).stream()
                //注：这里使用 map 映射了需要返回的 VO，即相同的属性字段就会转换
                .map(e -> e.copyProperties(AreaVO.class)).collect(Collectors.toList());
        if (CollectionUtils.isNotEmpty(areaVOList)){
            //第二步，根据父Id 字段进行分组，即所有数据都会按照第一层至最后一层都按照父子关系进行分组；注意，是对所有数据分组
            Map<Long, List<AreaVO>> areaVoListMap = areaVOList.parallelStream().collect(Collectors.groupingBy(AreaVO::getParentId));
            //第三步，也是最关键的一步，将所有子数据 List 集合经过遍历后都一层层地放置好，最终会得到一个包含父子关系的完整List
            areaVOList.forEach(e -> e.setChildrenAreaVOList(areaVoListMap.get(e.getId())));
            //第四步，过滤出符合顶层父Id的所有数据，即所有数据都归属于一个顶层父Id
            List<AreaVO> resultList = areaVOList.stream()
                    .filter(e -> Constants.COUNTRY_CHINA_TOP_NUM.equals(e.getParentId()))
                    .collect(Collectors.toList());
            return Optional.of(resultList).orElse(null);
        }
        return new ArrayList<>();
    }