总阅读量:次

⭐⭐⭐ Spring Boot 项目实战 ⭐⭐⭐ Spring Cloud 项目实战
《Dubbo 实现原理与源码解析 —— 精品合集》 《Netty 实现原理与源码解析 —— 精品合集》
《Spring 实现原理与源码解析 —— 精品合集》 《MyBatis 实现原理与源码解析 —— 精品合集》
《Spring MVC 实现原理与源码解析 —— 精品合集》 《数据库实体设计合集》
《Spring Boot 实现原理与源码解析 —— 精品合集》 《Java 面试题 + Java 学习指南》

摘要: 原创出处 blog.csdn.net/a18505947362/article/details/122458089 「丰丶」欢迎转载，保留摘要，谢谢！

🙂🙂🙂关注**微信公众号：【芋道源码】**有福利：

  1. RocketMQ / MyCAT / Sharding-JDBC 所有源码分析文章列表
  2. RocketMQ / MyCAT / Sharding-JDBC 中文注释源码 GitHub 地址
  3. 您对于源码的疑问每条留言都将得到认真回复。甚至不知道如何读源码也可以请教噢。
  4. 新的源码解析文章实时收到通知。每周更新一篇左右。
  5. 认真的源码交流微信群。

记录

在Java后台利用递归思路进行构建树形结构数据，返回给前端，能以下拉菜单等形式进行展示。

为了简化代码，引入Lombok的Jar包，可省略实体类set()、get()方法。

<dependency>
     <groupId>org.projectlombok</groupId>
     <artifactId>lombok</artifactId>
     <version>1.16.12</version>
</dependency>

一、树节点数据类

import lombok.Getter;
import lombok.Setter;
 
import java.util.List;
 
/**
 *  TreeNode 树节点 （定义每一个节点的信息，即每一个节点对应一条数据信息）
 *
 *  @author LBF
 *  @date 2022/1/10 16:46
 */
@Setter
@Getter
public class TreeNode {
 
    /** 节点ID */
    private Integer id;
 
    /** 父节点ID：顶级节点为0 */
    private Integer parentId;
 
    /** 节点名称 */
    private String label;
 
    /** 子节点 */
    private List<TreeNode> children;
 
    public TreeNode(Integer id, Integer parentId, String label) {
        this.id = id;
        this.parentId = parentId;
        this.label = label;
    }
}

二、构建树形类

理解思路（个人）：

1、首先获取所有的根节点（顶级节点），即根节点的parentId = 0。

2、根据每一个根节点，与所有节点集合（数据）进行判断，当前节点是否为其下的子节点。

3、若是，则递归调用构建树形；若不是，则表明该节点不属于其下子节点。

4、应继续循环判断节点父子关系，直到所有节点与根节点判断完毕。

import java.util.ArrayList;
import java.util.List;
 
/**
 *  BuildTree 构建树形结构
 *
 *  @author LBF
 *  @date 2022/1/10 17:06
 */
public class TreeBuild {
    
    // 保存参与构建树形的所有数据（通常数据库查询结果）
    public List<TreeNode> nodeList = new ArrayList<>();
 
    /**
     *  构造方法
     *  @param nodeList 将数据集合赋值给nodeList，即所有数据作为所有节点。
     */
    public TreeBuild(List<TreeNode> nodeList){
        this.nodeList = nodeList;
    }
 
    /**
     *   获取需构建的所有根节点（顶级节点） "0"
     *   @return 所有根节点List集合
     */
    public List<TreeNode> getRootNode(){
        // 保存所有根节点（所有根节点的数据）
        List<TreeNode> rootNodeList = new ArrayList<>();
        // treeNode：查询出的每一条数据（节点）
        for (TreeNode treeNode : nodeList){
            // 判断当前节点是否为根节点，此处注意：若parentId类型是String，则要采用equals()方法判断。
            if (0 == treeNode.getParentId()) {
                // 是，添加
                rootNodeList.add(treeNode);
            }
        }
        return rootNodeList;
    }
 
    /**
     *  根据每一个顶级节点（根节点）进行构建树形结构
     *  @return  构建整棵树
     */
    public List<TreeNode> buildTree(){
        // treeNodes：保存一个顶级节点所构建出来的完整树形
        List<TreeNode> treeNodes = new ArrayList<TreeNode>();
        // getRootNode()：获取所有的根节点
        for (TreeNode treeRootNode : getRootNode()) {
            // 将顶级节点进行构建子树
            treeRootNode = buildChildTree(treeRootNode);
            // 完成一个顶级节点所构建的树形，增加进来
            treeNodes.add(treeRootNode);
        }
        return treeNodes;
    }
 
    /**
     *  递归-----构建子树形结构
     *  @param  pNode 根节点（顶级节点）
     *  @return 整棵树
     */
    public TreeNode buildChildTree(TreeNode pNode){
        List<TreeNode> childTree = new ArrayList<TreeNode>();
        // nodeList：所有节点集合（所有数据）
        for (TreeNode treeNode : nodeList) {
            // 判断当前节点的父节点ID是否等于根节点的ID，即当前节点为其下的子节点
            if (treeNode.getParentId().equals(pNode.getId())) {
                // 再递归进行判断当前节点的情况，调用自身方法
                childTree.add(buildChildTree(treeNode));
            }
        }
        // for循环结束，即节点下没有任何节点，树形构建结束，设置树结果
        pNode.setChildren(childTree);
        return pNode;
    }
 
}

三、测试案例

import com.lbf.common.core.domain.AjaxResult;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
 
import java.util.ArrayList;
import java.util.List;
 
/**
 *  TreeController 树控制层
 *  方式：传递所有数据集合作为参数，调用buildTree()构建树形。
 *
 *  @author LBF
 *  @date 2022/1/10 17:49
 */
@RestController
@RequestMapping("/tree")
public class TreeController {
 
    @GetMapping("/treeTest")
    public AjaxResult treeTest(){
 
        // 模拟测试数据（通常为数据库的查询结果）
        List<TreeNode> treeNodeList = new ArrayList<>();
        treeNodeList.add(new TreeNode(1,0,"顶级节点A"));
        treeNodeList.add(new TreeNode(2,0,"顶级节点B"));
        treeNodeList.add(new TreeNode(3,1,"父节点是A"));
        treeNodeList.add(new TreeNode(4,2,"父节点是B"));
        treeNodeList.add(new TreeNode(5,2,"父节点是B"));
        treeNodeList.add(new TreeNode(6,3,"父节点的ID是3"));
 
        // 创建树形结构（数据集合作为参数）
        TreeBuild treeBuild = new TreeBuild(treeNodeList);
        // 原查询结果转换树形结构
        treeNodeList = treeBuild.buildTree();
        // AjaxResult：个人封装返回的结果体
        return AjaxResult.success("测试数据",treeNodeList);
    }
}

四、测试结果

五、总结

本文介绍Java后台构建树形结构数据的设计思路及实现，在设计过程中，仍然存在许多未能充分理解的位置，若存在错误或更好实现方案，可分享/指出，共同学习进步。

文章目录
  1. 1. 记录
  2. 2. 一、树节点数据类
  3. 3. 二、构建树形类
  4. 4. 三、测试案例
  5. 5. 四、测试结果
  6. 6. 五、总结