数据库单表实现多级子树

摘要

数据库单表存储多级别数据，使用parent_id字段进行关联。

实体类

package com.heifan.demo.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import java.util.List;

import lombok.Data;
import lombok.EqualsAndHashCode;

/**
 * <p>
 * 树形结构测试表
 * </p>
 *
 * @author HiF
 * @since 2022-03-22
 */
@Data
@EqualsAndHashCode(callSuper = false)
public class TestTreeNode implements Serializable {

    private static final long serialVersionUID = 1L;

    /**
     * 主键
     */
    @TableId(value = "id", type = IdType.AUTO)
    private Integer id;

    /**
     * 父级主键（0：无父级）
     */
    private Integer parentId;

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

}

测试用例，写法不规范，大家在开发过程中一定要规范开发！

package com.heifan.demo.domain.vo;

import com.heifan.demo.entity.TestTreeNode;
import lombok.Data;

import java.util.List;

@Data
public class TestTreeVO {

    private Integer id;

    /**
     * 父级主键（0：无父级）
     */
    private Integer parentId;

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

    /**
     * 子节点
     */
    private List<TestTreeVO> childList;
}

功能实现

public Object treeById(Integer id) {
        List<TestTreeVO> testTreeVOS = testTreeNodeDao.selectAll();
        List<TestTreeVO> testTreeNodeList = testTreeVOS.stream().filter(m -> m.getId().equals(id)).map(
                (m) -> {
                    m.setChildList(getChildrens(m, testTreeVOS));
                    return m;
                }
        ).collect(Collectors.toList());
        String s = JsonTool.beanToJson(testTreeNodeList);
        log.info(s);
        return Result.success();
    }


    /**
     * 递归查询子节点
     *
     * @param root 根节点
     * @param all  所有节点
     * @return 根节点信息
     */
    private List<TestTreeVO> getChildrens(TestTreeVO root, List<TestTreeVO> all) {
        List<TestTreeVO> children = all.stream().filter(m -> Objects.equals(m.getParentId(), root.getId())).map(
                (m) -> {
                    m.setChildList(getChildrens(m, all));
                    return m;
                }
        ).collect(Collectors.toList());
        return children;
    }