java利用stream去构建树形结构数据

首先构建一个树形的实体类

@Data
public class SalaryNodeTreeVo {

    /*结点父类id*/
    private Integer parentId;

    /*结点名称*/
    private String nodeName;

    /*当前节点id*/
    private Integer nodeId;

    /*孩子结点list*/
    private List<SalaryNodeTreeVo> children;

}

然后利用stream去构建树（parentId为0的即为根节点）

    private List<SalaryNodeTreeVo> buildTree(List<SalaryNodeTreeVo> nodes) {
        Map<Integer, List<SalaryNodeTreeVo>> children = nodes.stream().filter(node -> node.getParentId() != 0)
                .collect(Collectors.groupingBy(node -> node.getParentId()));
        nodes.forEach(node -> node.setChildren(children.get(node.getNodeId())));
        return nodes.stream().filter(node -> node.getParentId() == 0).collect(Collectors.toList());
    }

这样返回的出参既是一个树形结构。