当前位置：首页 > 文章列表 > 数据库 > MySQL > MySQL递归查询上下级菜单

MySQL递归查询上下级菜单

来源：SegmentFault 2023-01-24 12:15:18 0浏览收藏

本篇文章向大家介绍《MySQL递归查询上下级菜单》，主要包括MySQL、存储过程，具有一定的参考价值，需要的朋友可以参考一下。

正文

在传统的后台管理系统里面经常会需要展示多级菜单关系，今天我们来学一下如何使用一条SQL语句展示多级菜单。

现在我们有一张

corpinfo

单位表，里面有一个

belong

字段指向上级单位，首先来看一下现在表里有什么数据：

SELECT uid,ubelong FROM corpinfo

现在是类似下面这样的一个三级菜单，

uid

为1的是我们的顶级菜单，

ubelog

为0。

现在我们想实现传入一个 uid ，把当前 uid 和其下级单位的 uid 都展示出来，当然我们可以使用代码或者网上常见的存储过程来实现，但是今天我们用一条SQL语句来实现该效果：

先来看看我们的SQL语句

SELECT DATA.uid FROM(
      SELECT
      @ids AS _ids,
      (
      SELECT @ids := GROUP_CONCAT(uid)
      FROM corpinfo
      WHERE FIND_IN_SET(ubelong, @ids)
      ) AS cids,
      @l := @l+1 AS level
      FROM corpinfo, (SELECT @ids := (参数) , @l := 0 ) b
      WHERE @ids IS NOT NULL
      ) ID, corpinfo DATA
WHERE FIND_IN_SET(DATA.uid, ID._ids)
ORDER BY level,uid

看下执行结果：

可以看到传入 uid 为 1 后，列出了 uid 为 1 的所有下级单位，连第三级的菜单也列出来了。

比较核心的有下面几个地方：

GROUP_CONCAT()函数

前言：在有

group by

的查询语句中，

select

指定的字段要么就包含在

group by

语句的后面，作为分组的依据，要么就包含在聚合函数中。

假设我们有一张

user

用户表，我们想查看名字相同的用户的最小年龄，可以这样写：

SELECT name,age FROM user GROUP BY name

执行结果为：

现在我们想查询

name

相同的用户的所有年龄，当然我们可以这样写：

SELECT name,age FROM user ORDER BY name

用

ORDER BY

对名字排序，执行结果为：

但是这样同一个名字出现多次，看上去非常不直观。有没有更直观的方法，既让每个名字都只出现一次，又能够显示所有的名字相同的人的id呢？——使用

GROUP_CONCAT()

函数

功能：将

group by

产生的同一个分组中的值连接起来，返回一个字符串结果。

语法：GROUP_CONCAT( [distinct] 要连接的字段 [ORDER BY 排序字段 ASC/DESC ] [separator '分隔符'] )

SELECT name,GROUP_CONCAT(age) FROM user GROUP BY name

执行结果为：

可以看到相同用户名的年龄都放到一起了，以逗号分割。

FIND_IN_SET函数

假设我们有一张

book

书籍表，有书名和作者两个字段：

SELECT name,author FROM book

执行结果为：

现在我们想查作者包含小A 的书籍，我们来试一下：

SELECT name,author FROM book WHERE author IN ('小A')

执行结果为：

实际上这样是不行的，这样只有当

author

字段的值等于

'小A'

时（和IN前面的字符串完全匹配），查询才有效，否则都得不到结果。

可能你会想到用

LIKE

实现，我们来试试看：

SELECT name,author FROM book WHERE author LIKE '%小A%';

执行结果为：

可以看到把

小AA

的书籍也查出来了，所以用

LIKE

无法实现该功能。

那么我们如何使用

FIND_IN_SET

函数来实现呢？

SELECT name,author FROM book WHERE FIND_IN_SET('小A',author);

执行结果为：

语法：FIND_IN_SET(str,strlist)

str ：要查询的字符串
strlist ：字段名参数以”,”分隔如 (1,2,6,8)
查询字段(strlist)中包含(str)的结果，返回结果为null或记录

知道了这两个函数后，现在回过头来看看前面的SQL语句：

运行选中的代码后可以看到列出了上下级的关系，至于细节这里不再展开描述。

既然我们能查出当前单位的所有下级单位，那么应该也能查询所有上级单位，来看下SQL：

   SELECT uid FROM(
        SELECT
        @id AS _id,
        ( SELECT @id := ubelong
        FROM corpinfo
        WHERE uid = @id
        ) AS _pid,
        @l := @l+1 as level
        FROM corpinfo,
        (SELECT @id := (参数), @l := 0 ) b
        WHERE @id > 0
        ) ID, corpinfo DATA
        WHERE ID._id = DATA.uid
   ORDER BY level DESC

还是我们的

corpinfo

单位表，执行结果为：

可以看到当输入 uid 为 5 时，列出了当前单位及其上级所有单位，SQL和上面的差不多，这里不再细说。

最后补充一段代码，既然我们已经拿到想要的单位编号了，接下来就是要递归构建我们的单位树了，来看下代码：

/**
 * 递归将模块树构建成JSON数组
 */
private JSONArray getJsonArray(List list) {
    Map> map = new HashMap(16);
    List sonList;
    for (ClCorpinfo clCorpinfo : list) {
        if (map.get(clCorpinfo.getUbelong()) != null) {
            sonList = map.get(clCorpinfo.getUbelong());
        } else {
            sonList = new ArrayList();
        }
        sonList.add(clCorpinfo);
        map.put(clCorpinfo.getUbelong(), sonList);
    }
    JSONArray array = new JSONArray();
    if (list.size() > 0) {
        array = getChildrenTree(map, 0, 0);
    }
    return array;
}
/**
 * 递归构建模块树的子类
 */
public JSONArray getChildrenTree(Map> map, Integer uparentid, Integer level) {
    JSONArray array = new JSONArray();
    for (ClCorpinfo clCorpinfo : map.get(uparentid)) {
        JSONObject obj = new JSONObject();
        obj.put("uid", clCorpinfo.getUid());
        obj.put("ubelong", clCorpinfo.getUbelong());
        obj.put("ucorpname", clCorpinfo.getUcorpname());
        obj.put("uparentname", clCorpinfo.getUparentname());
        if (map.get(clCorpinfo.getUid()) != null) {
            level++;
            obj.put("children", getChildrenTree(map, clCorpinfo.getUid(), level));
        } else {
            obj.put("children", null);
        }
        array.add(obj);
    }
    return array;
}

上面这段只要传入单位集合，接下来会递归来构建我们的单位树，接下来只要前端渲染上去就完事了。