MySQL實現樹狀所有子節點查詢的方法

lotus_ruan發表於2021-09-09

本文例項講述了MySQL實現樹狀所有子節點查詢的方法。分享給大家供大家參考,具體如下:

在Oracle 中我們知道有一個 Hierarchical Queries 透過CONNECT BY 我們可以方便的查了所有當前節點下的所有子節點。但很遺憾,在MySQL的目前版本中還沒有對應的功能。

在MySQL中如果是有限的層次,比如我們事先如果可以確定這個樹的最大深度是4, 那麼所有節點為根的樹的深度均不會超過4,則我們可以直接透過left join 來實現。

但很多時候我們無法控制樹的深度。這時就需要在MySQL中用儲存過程來實現或在你的程式中來實現這個遞迴。本文討論一下幾種實現的方法。

樣例資料:

?


mysql> create table treeNodes  -> (  -> id int primary key,  -> nodename varchar(20),  -> pid int  -> );Query OK, 0 rows affected (0.09 sec)mysql> select * from treenodes;+----+----------+------+| id | nodename | pid |+----+----------+------+| 1 | A    |  0 || 2 | B    |  1 || 3 | C    |  1 || 4 | D    |  2 || 5 | E    |  2 || 6 | F    |  3 || 7 | G    |  6 || 8 | H    |  0 || 9 | I    |  8 || 10 | J    |  8 || 11 | K    |  8 || 12 | L    |  9 || 13 | M    |  9 || 14 | N    |  12 || 15 | O    |  12 || 16 | P    |  15 || 17 | Q    |  15 |+----+----------+------+17 rows in set (0.00 sec)

樹形圖如下

?


1:A+-- 2:B|  +-- 4:D|  +-- 5:E+-- 3:C   +-- 6:F     +-- 7:G8:H+-- 9:I|  +-- 12:L|  |  +--14:N|  |  +--15:O|  |    +--16:P|  |    +--17:Q|  +-- 13:M+-- 10:J+-- 11:K

方法一:利用函式來得到所有子節點號。

建立一個function getChildLst, 得到一個由所有子節點號組成的字串.

?


mysql> delimiter //mysql>mysql> CREATE FUNCTION `getChildLst`(rootId INT)  -> RETURNS varchar(1000)  -> BEGIN  ->  DECLARE sTemp VARCHAR(1000);  ->  DECLARE sTempChd VARCHAR(1000);  ->  ->  SET sTemp = '$';  ->  SET sTempChd =cast(rootId as CHAR);  ->  ->  WHILE sTempChd is not null DO  ->   SET sTemp = concat(sTemp,',',sTempChd);  ->   SELECT group_concat(id) INTO sTempChd FROM treeNodes where FIND_IN_SET(pid,sTempChd)>0;  ->  END WHILE;  ->  RETURN sTemp;  -> END  -> //Query OK, 0 rows affected (0.00 sec)mysql>mysql> delimiter ;

使用我們直接利用find_in_set函式配合這個getChildlst來查詢

?


mysql> select getChildLst(1);+-----------------+| getChildLst(1) |+-----------------+| $,1,2,3,4,5,6,7 |+-----------------+1 row in set (0.00 sec)mysql> select * from treeNodes  -> where FIND_IN_SET(id, getChildLst(1));+----+----------+------+| id | nodename | pid |+----+----------+------+| 1 | A    |  0 || 2 | B    |  1 || 3 | C    |  1 || 4 | D    |  2 || 5 | E    |  2 || 6 | F    |  3 || 7 | G    |  6 |+----+----------+------+7 rows in set (0.01 sec)mysql> select * from treeNodes  -> where FIND_IN_SET(id, getChildLst(3));+----+----------+------+| id | nodename | pid |+----+----------+------+| 3 | C    |  1 || 6 | F    |  3 || 7 | G    |  6 |+----+----------+------+3 rows in set (0.01 sec)

優點: 簡單,方便,沒有遞迴呼叫層次深度的限制 (max_sp_recursion_depth,最大255) ;

缺點:長度受限,雖然可以擴大 RETURNS varchar(1000),但總是有最大限制的。

MySQL目前版本( 5.1.33-community)中還不支援function 的遞迴呼叫。

方法二:利用臨時表和過程遞迴

建立儲存過程如下。createChildLst 為遞迴過程,showChildLst為呼叫入口過程,準備臨時表及初始化。

?


mysql> delimiter //mysql>mysql> # 入口過程mysql> CREATE PROCEDURE showChildLst (IN rootId INT)  -> BEGIN  -> CREATE TEMPORARY TABLE IF NOT EXISTS tmpLst  ->  (sno int primary key auto_increment,id int,depth int);  -> DELETE FROM tmpLst;  ->  -> CALL createChildLst(rootId,0);  ->  -> select tmpLst.*,treeNodes.* from tmpLst,treeNodes where tmpLst.id=treeNodes.id order by tmpLst.sno;  -> END;  -> //Query OK, 0 rows affected (0.00 sec)mysql>mysql> # 遞迴過程mysql> CREATE PROCEDURE createChildLst (IN rootId INT,IN nDepth INT)  -> BEGIN  -> DECLARE done INT DEFAULT 0;  -> DECLARE b INT;  -> DECLARE cur1 CURSOR FOR SELECT id FROM treeNodes where pid=rootId;  -> DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;  ->  -> insert into tmpLst values (null,rootId,nDepth);  ->  -> OPEN cur1;  ->  -> FETCH cur1 INTO b;  -> WHILE done=0 DO  ->     CALL createChildLst(b,nDepth+1);  ->     FETCH cur1 INTO b;  -> END WHILE;  ->  -> CLOSE cur1;  -> END;  -> //Query OK, 0 rows affected (0.00 sec)mysql> delimiter ;

呼叫時傳入結點

?


mysql> call showChildLst(1);+-----+------+-------+----+----------+------+| sno | id  | depth | id | nodename | pid |+-----+------+-------+----+----------+------+|  4 |  1 |   0 | 1 | A    |  0 ||  5 |  2 |   1 | 2 | B    |  1 ||  6 |  4 |   2 | 4 | D    |  2 ||  7 |  5 |   2 | 5 | E    |  2 ||  8 |  3 |   1 | 3 | C    |  1 ||  9 |  6 |   2 | 6 | F    |  3 || 10 |  7 |   3 | 7 | G    |  6 |+-----+------+-------+----+----------+------+7 rows in set (0.13 sec)Query OK, 0 rows affected, 1 warning (0.14 sec)mysql>mysql> call showChildLst(3);+-----+------+-------+----+----------+------+| sno | id  | depth | id | nodename | pid |+-----+------+-------+----+----------+------+|  1 |  3 |   0 | 3 | C    |  1 ||  2 |  6 |   1 | 6 | F    |  3 ||  3 |  7 |   2 | 7 | G    |  6 |+-----+------+-------+----+----------+------+3 rows in set (0.11 sec)Query OK, 0 rows affected, 1 warning (0.11 sec)

depth 為深度,這樣可以在程式進行一些顯示上的格式化處理。類似於oracle中的 level 偽列。sno 僅供排序控制。這樣你還可以透過臨時表tmpLst與資料庫中其它表進行聯接查詢。

MySQL中你可以利用系統引數 max_sp_recursion_depth 來控制遞迴呼叫的層數上限。如下例設為12.

?

12 mysql> set max_sp_recursion_depth=12;Query OK, 0 rows affected (0.00 sec)

優點 : 可以更靈活處理,及層數的顯示。並且可以按照樹的遍歷順序得到結果。

缺點 : 遞迴有255的限制。

方法三:利用中間表和過程

(本方法由yongyupost2000提供樣子改編)
建立儲存過程如下。由於MySQL中不允許在同一語句中對臨時表多次引用,只以使用普通表tmpLst來實現了。當然你的程式中負責在用完後清除這個表。

?


delimiter //drop PROCEDURE IF EXISTS showTreeNodes_yongyupost2000//CREATE PROCEDURE showTreeNodes_yongyupost2000 (IN rootid INT)BEGIN DECLARE Level int ; drop TABLE IF EXISTS tmpLst; CREATE TABLE tmpLst ( id int, nLevel int, sCort varchar(8000) ); Set Level=0 ; INSERT into tmpLst SELECT id,Level,ID FROM treeNodes WHERE PID=rootid; WHILE ROW_COUNT()>0 DO SET Level=Level+1 ; INSERT into tmpLst  SELECT A.ID,Level,concat(B.sCort,A.ID) FROM treeNodes A,tmpLst B  WHERE A.PID=B.ID AND B.nLevel=Level-1 ; END WHILE;END;//delimiter ;CALL showTreeNodes_yongyupost2000(0);

執行完後會產生一個tmpLst表,nLevel 為節點深度,sCort 為排序欄位。

使用方法

?


SELECT concat(SPACE(B.nLevel*2),'+--',A.nodename)FROM treeNodes A,tmpLst BWHERE A.ID=B.IDORDER BY B.sCort;+--------------------------------------------+| concat(SPACE(B.nLevel*2),'+--',A.nodename) |+--------------------------------------------+| +--A                    ||  +--B                   ||   +--D                  ||   +--E                  ||  +--C                   ||   +--F                  ||    +--G                 || +--H                    ||  +--J                   ||  +--K                   ||  +--I                   ||   +--L                  ||    +--N                 ||    +--O                 ||     +--P                ||     +--Q                ||   +--M                  |+--------------------------------------------+17 rows in set (0.00 sec)

優點 : 層數的顯示。並且可以按照樹的遍歷順序得到結果。沒有遞迴限制。

缺點 : MySQL中對臨時表的限制,只能使用普通表,需做事後清理。

以上是幾個在MySQL中用儲存過程比較簡單的實現方法。

更多關於MySQL相關內容感興趣的讀者可檢視本站專題:《MySQL日誌操作技巧大全》、《MySQL事務操作技巧彙總》、《MySQL儲存過程技巧大全》、《MySQL資料庫鎖相關技巧彙總》及《MySQL常用函式大彙總》

希望本文所述對大家MySQL資料庫計有所幫助。

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/1762/viewspace-2805569/,如需轉載,請註明出處,否則將追究法律責任。

相關文章