Java优化递归查询Mysql节点树数据

发布时间 2023-07-14 14:20:16作者: 人间二两风

示例

目前有一个功能:任务计划管理,必然存在多级子任务的父子级关系,每个任务还会存在其它数据的关联表。

mysql无法一次性递归查出想要的数据结构,想必很多人都会是通过根目录递归查询数据库的方式查出树结构数据。如果节点数较多,就会造成大量请求Mysql查询,效率会很低。

那么如何优化节点树数据查询效率???

表设计

# task(任务主表) [id,name,...]
[110,"XXX项目排期",...]

# task_tree(节点树子任务表) [id,task_id,parent_id,task_name,level,task_num,...]
[1, 110, 110, "XXX项目排期",  0, 0, ...]
[2, 110, 1, "一级主任务1",     1, 1, ...]
[5, 110, 2, "二级子任务1.1",   2, 1, ...]
[6, 110, 2, "二级子任务1.2",   2, 2, ...]
[3, 110, 1, "一级主任务2",     1, 2, ...]
[4, 110, 1, "一级主任务3",     1, 3, ...]
[7, 110, 4, "二级子任务3.1",   2, 1, ...]
[8, 110, 7, "三级子任务3.1.1", 3, 1, ...]
[8, 110, 7, "三级子任务3.1.2", 3, 2, ...]

# task_tree_pre(前置任务表) 关联节点任务表 [id,task_id,task_tree_id,val,...]
# 同时关联任务主表的好处是在删除数据的时候减少不必要的关联查询
[1, 110, 6 , "2SF", ...]
[1, 110, 6 , "7SS", ...]

Java使用Map处理

原理:仅发起一次sql请求把所需数据全查出来,使用Map进行逻辑封装处理,效率非常之快。

public class Test {
    /**
     * 根据主表id查询节点树子任务数据
     */
    public TaskTree tree(String id)
    {
        TaskTree tree = new TaskTree();
        // 根据主表id查询所有的前置任务
        List<TaskTreePre> listPre = "select * from task_tree_pre where task_id = 110";
        Map<String, List<TaskTreePre>> mapPre = new HashMap<>();
        for (TaskTreePre pre : listPre) {
            // 对所有taskTreeId用map封装对应的前置任务列表
            List<TaskTreePre> list = new ArrayList<>();
            if (mapPre.containsKey(pre.getTaskTreeId())) {
                // 先获取到map中已存在的listPre,再继续添加
                list = mapPre.get(pre.getTaskTreeId());
            }
            list.add(pre);
            mapPre.put(pre.getTaskTreeId(), list);
        }
        // 根据主表id查询所有的节点树任务数据
        List<TaskTree> listTree = "select * from task_tree where task_id = 110";
        Map<String, List<TaskTree>> map = new HashMap<>();
        // 设置前置任务
        for (TaskTree task : listTree) {
            if (id.equals(task.getParentId())) {
                // 获取根节点对象
                tree = task;
            }
            // 根据任务表id获取前置任务并添加
            task.setTaskTreePreList(mapPre.get(task.getId()));
            // 对所有parentId用map封装所属的子级任务
            List<TaskTree> list = new ArrayList<>();
            if (map.containsKey(task.getParentId())) {
                list = map.get(task.getParentId());
            }
            list.add(task);
            map.put(task.getParentId(), list);
        }
        // 递归封装节点树结构数据
        tree.setChildren(initTree(map, tree.getId()));
        return tree;
    }

    /**
     * 递归封装节点树结构数据
     * @param map   key包含了所有的parentId, value=所属子级列表
     * @param id    父级id
     * @return
     */
    public List<TaskTree> initTree(Map<String, List<TaskTree>> map, String id) {
        List<TaskTree> list = new ArrayList<>();
        if (map.containsKey(id)) {
            list = map.get(id);
            if (list.size() > 0) {
                for (TaskTree task : list) {
                    task.setChildren(initTree(map, task.getId()));
                }
            }
        }
        return list;
    }
}