首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >MySQL实现树的遍历

MySQL实现树的遍历

作者头像
用户1148526
发布2018-01-03 12:12:08
发布2018-01-03 12:12:08
1.7K0
举报
文章被收录于专栏:Hadoop数据仓库Hadoop数据仓库

经常在一个表中有父子关系的两个字段,比如empno与manager,这种结构中需要用到树的遍历。在Oracle 中可以使用connect by简单解决问题,但MySQL 5.1中还不支持(据说已纳入to do中),要自己写过程或函数来实现。

一、建立测试表和数据:

[c-sharp]

  1. DROP TABLE IF EXISTS `channel`;  
  2. CREATE TABLE `channel` (  
  3.   `id` int(11) NOT NULL AUTO_INCREMENT,    
  4.   `cname` varchar(200) DEFAULT NULL,  
  5.   `parent_id` int(11) DEFAULT NULL,  
  6.   PRIMARY KEY (`id`)  
  7. ) ENGINE=MyISAM AUTO_INCREMENT=19 DEFAULT CHARSET=utf8;  
  8. /*Data for the table `channel` */
  9. insert  into `channel`(`id`,`cname`,`parent_id`)   
  10. values (13,'首页',-1),  
  11.        (14,'TV580',-1),  
  12.        (15,'生活580',-1),  
  13.        (16,'左上幻灯片',13),  
  14.        (17,'帮忙',14),  
  15.        (18,'栏目简介',17);  

二、利用临时表和递归过程实现树的遍历(mysql的UDF不能递归调用):

[c-sharp]

  1. DELIMITER $$  
  2. USE `db1`$$  
  3. -- 从某节点向下遍历子节点  
  4. -- 递归生成临时表数据  
  5. DROP PROCEDURE IF EXISTS `createChildLst`$$  
  6. CREATE PROCEDURE `createChildLst`(IN rootId INT,IN nDepth INT)  
  7. BEGIN  
  8.       DECLARE done INT DEFAULT 0;  
  9.       DECLARE b INT;  
  10.       DECLARE cur1 CURSOR FOR SELECT id FROM channel WHERE parent_id=rootId;  
  11.       DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;  
  12.       SET max_sp_recursion_depth=12;  
  13.       INSERT INTO tmpLst VALUES (NULL,rootId,nDepth);  
  14.       OPEN cur1;  
  15.       FETCH cur1 INTO b;  
  16.       WHILE done=0 DO  
  17.               CALL createChildLst(b,nDepth+1);  
  18.               FETCH cur1 INTO b;  
  19.       END WHILE;  
  20.       CLOSE cur1;  
  21.      END$$  
  22. -- 从某节点向上追溯根节点  
  23. -- 递归生成临时表数据  
  24. DROP PROCEDURE IF EXISTS `createParentLst`$$  
  25. CREATE PROCEDURE `createParentLst`(IN rootId INT,IN nDepth INT)  
  26. BEGIN  
  27.       DECLARE done INT DEFAULT 0;  
  28.       DECLARE b INT;  
  29.       DECLARE cur1 CURSOR FOR SELECT parent_id FROM channel WHERE id=rootId;  
  30.       DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;  
  31.       SET max_sp_recursion_depth=12;  
  32.       INSERT INTO tmpLst VALUES (NULL,rootId,nDepth);  
  33.       OPEN cur1;  
  34.       FETCH cur1 INTO b;  
  35.       WHILE done=0 DO  
  36.               CALL createParentLst(b,nDepth+1);  
  37.               FETCH cur1 INTO b;  
  38.       END WHILE;  
  39.       CLOSE cur1;  
  40.      END$$  
  41. -- 实现类似Oracle SYS_CONNECT_BY_PATH的功能  
  42. -- 递归过程输出某节点id路径  
  43. DROP PROCEDURE IF EXISTS `createPathLst`$$  
  44. CREATE PROCEDURE `createPathLst`(IN nid INT,IN delimit VARCHAR(10),INOUT pathstr VARCHAR(1000))  
  45. BEGIN                    
  46.       DECLARE done INT DEFAULT 0;  
  47.       DECLARE parentid INT DEFAULT 0;        
  48.       DECLARE cur1 CURSOR FOR   
  49.       SELECT t.parent_id,CONCAT(CAST(t.parent_id AS CHAR),delimit,pathstr)  
  50.         FROM channel AS t WHERE t.id = nid;  
  51.       DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;  
  52.       SET max_sp_recursion_depth=12;                    
  53.       OPEN cur1;  
  54.       FETCH cur1 INTO parentid,pathstr;  
  55.       WHILE done=0 DO             
  56.               CALL createPathLst(parentid,delimit,pathstr);  
  57.               FETCH cur1 INTO parentid,pathstr;  
  58.       END WHILE;  
  59.       CLOSE cur1;   
  60.      END$$  
  61. -- 递归过程输出某节点name路径  
  62. DROP PROCEDURE IF EXISTS `createPathnameLst`$$  
  63. CREATE PROCEDURE `createPathnameLst`(IN nid INT,IN delimit VARCHAR(10),INOUT pathstr VARCHAR(1000))  
  64. BEGIN                    
  65.       DECLARE done INT DEFAULT 0;  
  66.       DECLARE parentid INT DEFAULT 0;        
  67.       DECLARE cur1 CURSOR FOR   
  68.       SELECT t.parent_id,CONCAT(t.cname,delimit,pathstr)  
  69.         FROM channel AS t WHERE t.id = nid;  
  70.       DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;  
  71.       SET max_sp_recursion_depth=12;                    
  72.       OPEN cur1;  
  73.       FETCH cur1 INTO parentid,pathstr;  
  74.       WHILE done=0 DO             
  75.               CALL createPathnameLst(parentid,delimit,pathstr);  
  76.               FETCH cur1 INTO parentid,pathstr;  
  77.       END WHILE;  
  78.       CLOSE cur1;   
  79.      END$$  
  80. -- 调用函数输出id路径  
  81. DROP FUNCTION IF EXISTS `fn_tree_path`$$  
  82. CREATE FUNCTION `fn_tree_path`(nid INT,delimit VARCHAR(10)) RETURNS VARCHAR(2000) CHARSET utf8  
  83. BEGIN    
  84.   DECLARE pathid VARCHAR(1000);  
  85.   SET @pathid=CAST(nid AS CHAR);  
  86.   CALL createPathLst(nid,delimit,@pathid);  
  87.   RETURN @pathid;  
  88. END$$  
  89. -- 调用函数输出name路径  
  90. DROP FUNCTION IF EXISTS `fn_tree_pathname`$$  
  91. CREATE FUNCTION `fn_tree_pathname`(nid INT,delimit VARCHAR(10)) RETURNS VARCHAR(2000) CHARSET utf8  
  92. BEGIN    
  93.   DECLARE pathid VARCHAR(1000);  
  94.   SET @pathid='';      
  95.   CALL createPathnameLst(nid,delimit,@pathid);  
  96.   RETURN @pathid;  
  97. END$$  
  98. -- 调用过程输出子节点  
  99. DROP PROCEDURE IF EXISTS `showChildLst`$$  
  100. CREATE PROCEDURE `showChildLst`(IN rootId INT)  
  101. BEGIN  
  102.       DROP TEMPORARY TABLE IF EXISTS tmpLst;  
  103.       CREATE TEMPORARY TABLE IF NOT EXISTS tmpLst   
  104.        (sno INT PRIMARY KEY AUTO_INCREMENT,id INT,depth INT);        
  105.       CALL createChildLst(rootId,0);  
  106.       SELECT channel.id,CONCAT(SPACE(tmpLst.depth*2),'--',channel.cname) NAME,channel.parent_id,tmpLst.depth,fn_tree_path(channel.id,'/') path,fn_tree_pathname(channel.id,'/') pathname  
  107.       FROM tmpLst,channel WHERE tmpLst.id=channel.id ORDER BY tmpLst.sno;  
  108.      END$$  
  109. -- 调用过程输出父节点  
  110. DROP PROCEDURE IF EXISTS `showParentLst`$$  
  111. CREATE PROCEDURE `showParentLst`(IN rootId INT)  
  112. BEGIN  
  113.       DROP TEMPORARY TABLE IF EXISTS tmpLst;  
  114.       CREATE TEMPORARY TABLE IF NOT EXISTS tmpLst   
  115.        (sno INT PRIMARY KEY AUTO_INCREMENT,id INT,depth INT);        
  116.       CALL createParentLst(rootId,0);  
  117.       SELECT channel.id,CONCAT(SPACE(tmpLst.depth*2),'--',channel.cname) NAME,channel.parent_id,tmpLst.depth,fn_tree_path(channel.id,'/') path,fn_tree_pathname(channel.id,'/') pathname  
  118.       FROM tmpLst,channel WHERE tmpLst.id=channel.id ORDER BY tmpLst.sno;  
  119.      END$$  
  120. DELIMITER ;  

三、测试

[c-sharp]

  1. CALL showChildLst(-1);  
  2. CALL showChildLst(13);  
  3. CALL showChildLst(14);  
  4. CALL showChildLst(17);  
  5. CALL showChildLst(18);  
  6. CALL showParentLst(-1);  
  7. CALL showParentLst(13);  
  8. CALL showParentLst(14);  
  9. CALL showParentLst(17);  
  10. CALL showParentLst(18);  

四、遗留问题 1. 因为mysql对动态游标的支持不够,所以要想做成通用的过程或函数比较困难,可以利用两个临时表来转换(同时去掉了递归调用),是个相对通用的实现。 2. 目前来看无论哪种实现,效率都不太好,希望mysql自己能实现oracle 的connect by 功能,应该会比较优化。

参考:MySQL中进行树状所有子节点的查询

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2016年12月29日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档