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語句來實現並不必定是最好的辦法，雖然簡單可是比較難懂，我這邊只是給大夥提供一個可行的方案，若是有什麼不對的地方請多多指教。