Java后台返回树形结构的二种实现方式

Ja va后台返回树形结构

后台返回树形结构，算是日常开发中高频遇到的需求了——部门树、菜单树、分类树，几乎每个项目都绕不开。实现方式也不少，今天重点聊两种：一种是自己写递归，另一种是直接用Hutool工具类，省心省力。

方式一、使用递归方式实现

    private List getSysDepts(String deptId) {
        // 1、获取表中所有数据 (自行根据实际场景拿到所有表数据)
        List all = getAllDept();
        // 3、返回的结果集
        List tree = new ArrayList<>();
        // 4、获取到最外层的部门信息
        List parentSysDept = all.stream().filter(dept -> dept.getDeptId().equals(deptId)).collect(Collectors.toList());
        if (CollectionUtils.isNotEmpty(parentSysDept)){
            // 5、取出部门信息
            SysDept dept = parentSysDept.get(0);
            // 6、放入集合中
            tree.add(dept);
            // 7、添加子节点
            addChildDept(dept, all);
        }
        return tree;
    }
    private void addChildDept(SysDept sysDept, List all) {
        // 1、拿到所传部门的字部门列表
        List tempList = all.stream()
                .filter(dept -> sysDept.getDeptId().equals(dept.getParentId()))
                .collect(Collectors.toList());
        sysDept.setChildren(tempList);
        tempList.forEach(dept -> {
            2、添加子节点
            addChildDept(dept, all);
        });
    }

效果展示：

递归的思路很直观：先找到根节点，然后写一个递归方法，每次把当前节点的子节点找出来并挂上去，直到子节点为空。代码虽然不多，但有个细节容易踩坑——递归深度，如果层级特别深，小心栈溢出。另外，每次递归都会全表扫描一次，数据量大了性能会掉。

方式二、利用Hutool工具进行实现

    private List> getSysDepts(String deptId) {
        // 获取所有数据
        List all = getAllDept();
        //配置
        TreeNodeConfig treeNodeConfig = new TreeNodeConfig();
        // 自定义属性名 都要默认值的
        treeNodeConfig.setIdKey("deptId");
        // 最大递归深度
        treeNodeConfig.setDeep(4);

        //转换器 （这里参数的deptId,指的是最外层的deptId值）
        List> treeNodes = TreeUtil.build(all, deptId, treeNodeConfig,
                (treeNode, tree) -> {
                    tree.setParentId(treeNode.getParentId());
                    // 扩展属性 ...(可以自行设置需要返回的字段)
                    tree.putExtra("deptId", treeNode.getDeptId());
                    tree.putExtra("status", treeNode.getStatus());
                });
        return treeNodes;
    }

效果展示：

相比自己写递归，Hutool的TreeUtil明显更省事。只需要配置好主键名（idKey）和递归深度，然后把原始列表传进去，它就自动帮你组装树了。还能通过putExtra把需要的额外字段塞进去，灵活性也不错。如果项目里已经引入了Hutool，强烈推荐用这种方式，代码量少，也不容易出bug。

总结

两种方式各有适用场景：递归实现适合对第三方依赖有顾虑、或者需要精细控制递归逻辑的情况；而Hutool工具类更适合快速开发，简单直接。实际选哪种，看团队偏好和项目现状就好。关键是把树形结构的核心逻辑理清楚——找准根节点，确定父子关系，剩下的就好办了。