zoukankan      html  css  js  c++  java
  • MySQL实现树状所有子节点查询的方法

    本文实例讲述了MySQL实现树状所有子节点查询的方法。分享给大家供大家参考,具体如下:

    在Oracle 中我们知道有一个 Hierarchical Queries 通过CONNECT BY 我们可以方便的查了所有当前节点下的所有子节点。但很遗憾,在MySQL的目前版本中还没有对应的功能。

    在MySQL中如果是有限的层次,比如我们事先如果可以确定这个树的最大深度是4, 那么所有节点为根的树的深度均不会超过4,则我们可以直接通过left join 来实现。

    但很多时候我们无法控制树的深度。这时就需要在MySQL中用存储过程来实现或在你的程序中来实现这个递归。本文讨论一下几种实现的方法。

    样例数据:

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    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
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    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, 得到一个由所有子节点号组成的字符串.

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    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来查找

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    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为调用入口过程,准备临时表及初始化。

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    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 ;

    调用时传入结点

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    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.

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

    优点 : 可以更灵活处理,及层数的显示。并且可以按照树的遍历顺序得到结果。

    缺点 : 递归有255的限制。

    方法三:利用中间表和过程

    (本方法由yongyupost2000提供样子改编)
    创建存储过程如下。由于MySQL中不允许在同一语句中对临时表多次引用,只以使用普通表tmpLst来实现了。当然你的程序中负责在用完后清除这个表。

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    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 为排序字段。

    使用方法

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    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中用存储过程比较简单的实现方法。

  • 相关阅读:
    Isolation Forest原理总结
    python 黑客书籍 ——扫描+暴力破解
    DNS解析过程
    DNS污染——domain name的解析被劫持了返回无效的ip
    leetcode 427. Construct Quad Tree
    todo
    leetcode 172. Factorial Trailing Zeroes
    leetcode 26. Remove Duplicates from Sorted Array
    滴滴快车奖励政策,高峰奖励,翻倍奖励,按成交率,指派单数分级(3月3日)
    北京Uber优步司机奖励政策(3月2日)
  • 原文地址:https://www.cnblogs.com/zhming26/p/6419621.html
Copyright © 2011-2022 走看看