MySQL递归查询上下级菜单

MySQL
598
0
0
2022-03-27

正文

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

现在我们有一张corpinfo单位表,里面有一个belong字段指向上级单位,首先来看一下现在表里有什么数据:

SELECT uid,ubelong FROM corpinfo
MySQL递归查询上下级菜单

现在是类似下面这样的一个三级菜单,uid为1的是我们的顶级菜单,ubelog为0。

MySQL递归查询上下级菜单

现在我们想实现传入一个 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

看下执行结果:

MySQL递归查询上下级菜单

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

MySQL递归查询上下级菜单

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

MySQL递归查询上下级菜单

GROUP_CONCAT()函数

前言:在有 group by 的查询语句中,select指定的字段要么就包含在 group by 语句的后面,作为分组的依据,要么就包含在聚合函数中。

假设我们有一张 user 用户表,我们想查看名字相同的用户的最小年龄,可以这样写:

SELECT name,age FROM user GROUP BY name

执行结果为:

MySQL递归查询上下级菜单

现在我们想查询 name 相同的用户的所有年龄,当然我们可以这样写:

SELECT name,age FROM user ORDER BY name

执行结果为:

MySQL递归查询上下级菜单

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

功能:group by 产生的同一个分组中的值连接起来,返回一个字符串结果。

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

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

执行结果为:

MySQL递归查询上下级菜单

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

FIND_IN_SET函数

假设我们有一张 book 书籍表,有书名和作者两个字段:

SELECT name,author FROM book

执行结果为:

MySQL递归查询上下级菜单

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

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

执行结果为:

MySQL递归查询上下级菜单

实际上这样是不行的,这样只有当 author 字段的值等于'小A'时(和IN前面的字符串完全匹配),查询才有效,否则都得不到结果。

可能你会想到用LIKE实现,我们来试试看:

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

执行结果为:

MySQL递归查询上下级菜单

可以看到把小AA的书籍也查出来了,所以用LIKE无法实现该功能。

那么我们如何使用 FIND_IN_SET 函数来实现呢?

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

执行结果为:

MySQL递归查询上下级菜单

语法:FIND_IN_SET(str,strlist)

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

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

MySQL递归查询上下级菜单

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

既然我们能查出当前单位的所有下级单位,那么应该也能查询所有上级单位,来看下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单位表,执行结果为:

MySQL递归查询上下级菜单

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

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

/**
 * 递归将模块树构建成JSON数组
 */ 
private JSONArray getJsonArray(List<ClCorpinfo> list) {
    Map<Integer, List<ClCorpinfo>> map = new HashMap<>(16);
    List<ClCorpinfo> 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<Integer, List<ClCorpinfo>> 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;
}

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

总结

其实网上也有很多其它的解决方案,比如用代码实现,也可以用存储过程实现,今天我们使用SQL语句来实现并不一定是最好的办法,虽然简单但是比较难懂,我这边只是给大伙提供一个可行的方案,如果有什么不对的地方请多多指教。

MySQL递归查询上下级菜单