分享一个使用stream流将list集合转为tree的方法

最新推荐文章于 2024-04-02 14:09:42 发布

听风执酒

最新推荐文章于 2024-04-02 14:09:42 发布

阅读量319

点赞数

文章标签： list 数据结构

本文链接：https://blog.csdn.net/weixin_44763213/article/details/131538363

版权

该代码实现了一个将包含id和pid的DemoData对象列表转换为树形结构的功能。首先，它创建了两个映射，一个用于存储单个对象，另一个用于按pid分组。然后，遍历分组映射，设置父节点的子节点并移除已处理的项。最后，返回顶级节点的列表，按sort属性排序。

摘要由CSDN通过智能技术生成

对象需要有id,pid


 public static List<DemoData> transToTree(List<DemoData> sources) {
        if (CollectionUtils.isEmpty(sources)) {
            return Collections.emptyList();
        }
        Map<Integer, DemoData> sourceMap = sources.stream().collect(Collectors.toMap(DemoData::getId, e -> e));
        Map<Integer, List<DemoData>> pIdToChildrenListMap = sources.stream().collect(Collectors.groupingBy(DemoData::getPid));
        List<Integer> willBeRemovedIdList = new LinkedList<>();
        for (Map.Entry<Integer, List<DemoData>> entry : pIdToChildrenListMap.entrySet()) {
            DemoData demoData = sourceMap.get(entry.getKey());
            if (demoData == null) {
                continue;
            }
            demoData.setChildren(entry.getValue().stream().sorted(Comparator.comparing(DemoData::getSort)).collect(Collectors.toList()));
            willBeRemovedIdList.add(entry.getKey());
        }
        willBeRemovedIdList.forEach(pIdToChildrenListMap::remove);
        // 获取顶级
        return pIdToChildrenListMap.values().stream().flatMap(Collection::stream).sorted(Comparator.comparing(DemoData::getSort)).collect(Collectors.toList());
    }