当前位置：首页 > 文章列表 > 数据库 > MySQL > 递归查询级联信息

递归查询级联信息

来源：SegmentFault 2023-02-23 20:58:23 0浏览收藏

在IT行业这个发展更新速度很快的行业，只有不停止的学习，才不会被行业所淘汰。如果你是数据库学习者，那么本文《递归查询级联信息》就很适合你！本篇内容主要包括递归查询级联信息，希望对大家的知识积累有所帮助，助力实战开发！

1. 需求背景

在很多场合，我们需要对表中的数据对递归查询。如以下情况：

1. 菜单分类中，我们往往需要由一级菜单获得对应的子菜单。

id	name	pid
1	图书	0
2	服装	0
3	儿童读物	1
4	杂志	1
5	卡通	3
6	睡前故事	3

我们希望得到的结果为，由图书可以查到：

@Mapper
public interface RelationTreeDao {
    List<relationtreedto> selectAllParentsByCode(String code);</relationtreedto>

[code, parentCode, parentCode, parentCode,...]

由于

RelationTreeDTO

是一个树结构，这就涉及到树的遍历。在此，以

树的深度优先搜索算法

，获得上述list。

/**
     * description:深度优先搜索DTO中的所有父节点
     * @author wanghongbing whbing1991@gmail.com
     * @param treeDTO RelationTreeDTO待解析的树结构对象
     * @return list [0]存code, [1]开始存parents
     * 一定会有一个父节点，list长度>=2
     */
    @Override
    public List<string> depthFirst(RelationTreeDTO treeDTO) {

        //list [0]存code, [1]开始存parents
        List<string> list = new ArrayList();
        list.add(treeDTO.getCode()); //list[0]
        
        ArrayDeque<relationtreedto> stack = new ArrayDeque();
        stack.push(treeDTO);

        while (!stack.isEmpty()){
            RelationTreeDTO node =stack.pop();
            list.add(node.getParentCode());
            //获取嵌套节点
            List<relationtreedto> parents = node.getParentCodeList();
            if(parents !=null && parents.size() >0){
                for (int i = parents.size()-1; i >=0 ; i--) {
                    stack.push(parents.get(i));
                }
            }
        }
        return list;
    }</relationtreedto></relationtreedto></string></string>

至此，该方式级联查询结束。
上述实现，

collection

结果为固定的树结构，在解析时，要使用算法（如

DFS

）获取树种的节点信息。虽然在mapper查询时，一次性获得了级联结构，后续解析仍然复杂。下面介绍推荐方式。

2.2 方法2：自定义实现级联查询

@Mapper
public interface RelationDao {
    List<taskrelationdto> selectParentByCode(String code);
   // 其他表
    List<taskrelationdto> selectOtherParentByCode(String code);
}</taskrelationdto></taskrelationdto>

dto（或entity，如数据库对应的话）

public class TaskRelationDTO {
    private String code;
    private String parentCode;
    // getter and setter
}

mapper.xml（假设有relation表和other_relation表，其字段相同。两个表完全为了展示扩展）

<?xml version="1.0" encoding="UTF-8" ?><!-- recommended not modified but can be added --><mapper namespace="com.***.dao.RelationDao"><!--tag--><resultmap id="relationMap" type="com.***.dto.RelationDTO"><result column="code" property="code" jdbctype="VARCHAR" javatype="String"></result><result column="parent_code" property="parentCode" jdbctype="VARCHAR" javatype="String"></result></resultmap><!-- a_relation表中选择当前code的父节点 -->&lt;select id=&quot;selectParentByCode&quot; parametertype=&quot;java.lang.String&quot; resultmap=&quot;relationMap&quot;&gt;
        SELECT
            `code`,`parent_code`
        FROM
            `relation`
        WHERE
            `code` = #{code}
        AND
            `parent_code` is not NULL
    &lt;/select&gt;<!-- other_relation表中选择当前code的父节点 -->&lt;select id=&quot;selectOtherParentByCode&quot; parametertype=&quot;java.lang.String&quot; resultmap=&quot;relationMap&quot;&gt;
        SELECT
            `code`,`parent_code`
        FROM
            `other_relation`
        WHERE
            `code` = #{code}
        AND
            `parent_code` is not NULL
    &lt;/select&gt;</mapper>

说明：上述查询仅为最简单的sql查询，我们将递归查询写在业务方法中。

service

   /**
     *
     * @param code 待查找父任务的子任务
     * @return 返回的list.size()>=2 list[0]当前code，[1]以后去重后的无序parentsCode
     *         如：[tag-depend-2, path-depend-0-p, path-depend-2, tag-depend-0, path-depend-0]
     */
    @Override
    public List<string> getAllParentsByCode(String code) {
        List<string> list = new ArrayList();
        Set<string> parentSet = new HashSet();
        ArrayDeque<string> stack = new ArrayDeque();
        int count = 0;
        final int MAX_LOOP_COUNT = 50;

        // 初始化stack，将code放入stack
        stack.push(code);
        // 将code加入list。如果最终list.isEmpty()，表明没有父节点，将其清空。故list长度最短为2
        list.add(code);

        while (!stack.isEmpty()){
            // 如果入栈次数太多，表明可能出现环形依赖。强行终止
            if(count++ > MAX_LOOP_COUNT){
                LOGGER.error("code为["+code+"]任务其父任务超过"+MAX_LOOP_COUNT+"个，请检查是否有环形依赖");
                list.addAll(parentSet);
                // 仅有taskCode，无parentCode时，将list清空
                if(list.size() == 1){
                    list.clear();
                }
                return list;
            }
            String childCode = stack.pop();
/**
可能会出现两个表交叉依赖情况，故有otherRelation
*/
            List<relationdto> relation =relationDao.selectTagParentByCode(childCode);
            List<taskrelationdto> otherRelation =relationDao.selectOtherParentByCode(childCode);

            // 从relation表中查找pop()元素的父任务，将其加入stack
            if(!relation.isEmpty()){
                for (int i = 0; i </taskrelationdto></relationdto></string></string></string></string>

原理

说明：上述原理，使用