1. 程式人生 > 實用技巧 >MySQL遞迴查詢corpinfo單位表

MySQL遞迴查詢corpinfo單位表

正文

在傳統的後臺管理系統裡面經常會需要展示多級選單關係,今天我們來學一下如何使用一條SQL語句展示多級選單。

現在我們有一張corpinfo單位表,裡面有一個belong欄位指向上級單位,首先來看一下現在表裡有什麼資料:

SELECT uid,ubelong FROM corpinfo
MySQL遞迴查詢上下級選單

現在是類似下面這樣的一個三級選單,uid為1的是我們的頂級選單,ubelog為0。

image.png

現在我們想實現傳入一個 uid ,把當前 uid 和其下級單位的 uid 都展示出來,當然我們可以使用程式碼或者網上常見的儲存過程來實現,但是今天我們用一條SQL語句來實現該效果:

先來看看我們的SQL語句

SELECT DATA.uid FROM(
      SELECT
      

看下執行結果:

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
        

還是我們的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遞迴查詢上下級選單