Java后端项目实现无限级树 - 案例:部门树 - Department实体类

private List<Department> getDeptTree() {
    // 1. 一次性查询出所有数据,放到内存中,可以使用map  -  做缓存用
    // 2. 从所有数据中找出第一层级/顶级部门:这就是我们需要返回的数据
    // 3. 遍历所有部门只要不是第一层级的,找自己的上级,并将自己放入上级的children中
    //将所有的部门撞到map中  - 每次获取父部门都从这里获取
    List<Department> all = departmentMapper.findAll();
    Map<Long, Department> map = new HashMap<>();
    for (Department department : all) {
        map.put(department.getId(), department);
    }

    //前端需要得到的集合数据 - 部门树
    List<Department> deptTree = new ArrayList<>();
    for (Department department : all) { //循环置值
        if (department.getParentId() == null) {//顶级部门的情况
            deptTree.add(department);
        } else {//不是顶级部门的情况
            //1.想办法找到当前部门的 父部门
            Long parent_id = department.getParentId();
           
            //从上面的Map中去获取的 - 以前装进去的对象还是会存在
            Department parentDepartment = map.get(parent_id);
            if (parentDepartment != null) {  //防止空指针,过滤出错数据
                //2.将自己装到  父部门 的children中
                parentDepartment.getChildren().add(department);
            }
        }
    }
    return deptTree;
}

                        
参考链接:https://blog.csdn.net/2301_77829365/article/details/131818062
posted @ 2024-02-08 17:34  予真  阅读(185)  评论(0)    收藏  举报