后端返回一棵树

最新推荐文章于 2024-07-22 14:15:06 发布

不来也不去zz

最新推荐文章于 2024-07-22 14:15:06 发布

阅读量70

点赞数

分类专栏： work 文章标签： java

本文链接：https://blog.csdn.net/weixin_43723036/article/details/127841087

版权

work 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

查询一个list通过stream流和递归生成一颗有层级的树

 @Override
    public List<IndustryCode> getIndustryCodeTree() {
        List<IndustryCode> industryCodeList = industryCodeMapper.getIndustryCodeList();
        return industryCodeList.stream()
                .filter(IndustryCode -> IndustryCode.getRank()==1)
                .peek(IndustryCode -> IndustryCode.setChildren(getChildren(IndustryCode,industryCodeList)))
                .collect(Collectors.toList());
    }

    //递归查找 行业代码的下级
    private List<IndustryCode> getChildren(IndustryCode root, List<IndustryCode> all) {
        return all.stream()
                .filter(industryCode -> industryCode.getParentId().equals(root.getCode()))
                .peek(industryCode -> industryCode.setChildren(getChildren(industryCode, all)))
                .collect(Collectors.toList());