lambda实现List转Tree

最新推荐文章于 2024-02-21 15:20:13 发布

原创最新推荐文章于 2024-02-21 15:20:13 发布 · 475 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

本文详细介绍了如何利用Java的Lambda表达式将List数据结构转换为树形结构，适用于数据组织和操作。通过实例代码展示转换过程，帮助开发者理解Lambda在构建树形结构中的应用。

    /**
     * 返回获取父级
     *
     * @param deptTrees
     * @return
     */
    public List<DeptTree> getParent(List<DeptTree> deptTrees) {
        List<DeptTree> collect = deptTrees
                .stream().filter(m -> StrUtil.isNotBlank(m.getPid()) && m.getLevel() != null && m.getLevel() == 1).map((m) -> {
                            m.setChildren(getChildren(m, deptTrees));
                            return m;
                        }
                ).collect(toList());
        // 如果没有父级，就是子级数据
        if (collect.isEmpty()) {
            return deptTrees;
        }
        return collect;
    }


    /**
     * 递归查询子节点
     *
     * @param root 根节点
     * @param all  所有节点
     * @return 根节点信息
     */
    private List<DeptTree> getChildren(DeptTree root, List<DeptTree> all) {
        return all.stream().filter(m -> Objects.equals(m.getPid(), root.getDeptId())).map(
                m -> {
                    m.setChildren(getChildren(m, all));
                    return m;
                }
        ).collect(toList());
    }


import com.fasterxml.jackson.annotation.JsonInclude;
import io.swagger.v3.oas.annotations.Hidden;
import lombok.Getter;
import lombok.Setter;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.List;

@Getter
@Setter
@Accessors(chain = true)
public class DeptTree implements Serializable {

    /**
     * 父级ID
     */
    private String pid;

    /**
     * 子级
     */
    @JsonInclude(JsonInclude.Include.NON_NULL) // 若被注解的字段值为 null，则序列化时忽略该字段。
    private List<DeptTree> children;
    /**
     * 部门id
     */
    private String deptId;
    /**
     * 部门名称
     */
    private String deptName;


    @Hidden
    private Integer level;

}