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

执行结果为：

可是这样同一个名字出现屡次，看上去很是不直观。有没有更直观的方法，既让每一个名字都只出现一次，又可以显示全部的名字相同的人的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<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语句来实现并不必定是最好的办法，虽然简单可是比较难懂，我这边只是给大伙提供一个可行的方案，若是有什么不对的地方请多多指教。