Mysql 實現樹狀遞迴查詢

iteye_5104發表於2013-11-15

在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:G  

 8: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.  

 

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 B   

WHERE A.ID=B.ID   

ORDER 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中用儲存過程比較簡單的實現方法。  

轉載自:http://www.2cto.com/database/201209/152513.html

              http://www.iteye.com/problems/72935

              http://oraclestudy.iteye.com/blog/1233012

相關文章