java8 集合构造树结构

癸酉金鸡

已于 2023-06-01 09:45:22 修改

阅读量474

点赞数

文章标签： java 开发语言

于 2023-06-01 09:21:26 首次发布

本文链接：https://blog.youkuaiyun.com/chinasi2012/article/details/130981484

版权

该代码片段展示了一个公共静态方法，用于根据给定的ID映射、父ID映射和节点映射函数，从列表中构建树形结构。方法首先将列表转换为Map，然后遍历并根据父ID关系组织节点。如果找不到父ID对应的节点，则跳过。最后，返回构建好的树形结构列表。isValid方法用于检查给定值是否有效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

公共方法

public static <T,V> List<T> buildTree(List<T> list, Function<T, V> idMapper, Function<T, V> pIdMapper,
                                  Function<T, List<T>> getNodeMapper, BiConsumer<T, List<T>> setNodeMapper){
    Map<V,T> vtMap = list.stream().collect(Collectors.toMap((idMapper),(value->value)));

    list.clear();
    for (Map.Entry<V, T> entry : vtMap.entrySet()) {
        T t = entry.getValue();
        V pId = pIdMapper.apply(t);

        if(isValid(pId)){
            T t1 = vtMap.get(pId);
            if(t1 == null){//找不到该父id对应的数据
                continue;
            }
            List<T> nodeList = getNodeMapper.apply(t1);
            if(nodeList == null){
                nodeList = new ArrayList();
                setNodeMapper.accept(t1, nodeList);
            }
            nodeList.add(t);
        } else {
            list.add(t);
        }
    }
    return list;
}

public static <V> boolean isValid(V value){
    boolean flag = false;
    if(null != value && value instanceof String && StringUtils.isNotNull(value)){
        flag = true;
    }
    if(null != value && value instanceof Integer && 0 != ((Integer) value).intValue()){
        flag = true;
    }
    return flag;
}

调用：

dtoList = commTree(dtoList, x->x.getId(), x->x.getParentId(), x->x.getNode(), KnowlCategoryDto::setNode);

注意：给属性赋值只能使用::而无法使用->