本篇文章小编给大家分享一下MySQL通过自定义函数实现递归查询父级ID或者子级ID代码实例,小编觉得挺不错的,现在分享给大家供大家参考,有需要的小伙伴们可以来看看。
背 景:
在MySQL中如果是有限的层次,比如我们事先如果可以确定这个树的最大深度, 那么所有节点为根的树的深度均不会超过树的最大深度,则我们可以直接通过left join来实现。
但很多时候我们是无法控制或者是知道树的深度的。这时就需要在MySQL中用存储过程(函数)来实现或者在程序中使用递归来实现。本文讨论在MySQL中使用函数来实现的方法:
一、环境准备
1、建表
CREATE TABLE `table_name` ( `id` int(11) NOT NULL AUTO_INCREMENT, `status` int(255) NULL DEFAULT NULL, `pid` int(11) NULL DEFAULT NULL, PRIMARY KEY (`id`) USING BTREE ) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;
2、插入数据
INSERT INTO `table_name` VALUES (1, 12, 0); INSERT INTO `table_name` VALUES (2, 4, 1); INSERT INTO `table_name` VALUES (3, 8, 2); INSERT INTO `table_name` VALUES (4, 16, 3); INSERT INTO `table_name` VALUES (5, 32, 3); INSERT INTO `table_name` VALUES (6, 64, 3); INSERT INTO `table_name` VALUES (7, 128, 6); INSERT INTO `table_name` VALUES (8, 256, 7); INSERT INTO `table_name` VALUES (9, 512, 8); INSERT INTO `table_name` VALUES (10, 1024, 9); INSERT INTO `table_name` VALUES (11, 2048, 10);
二、MySQL函数的编写
1、查询当前节点的所有父级节点
delimiter // CREATE FUNCTION `getParentList`(root_id BIGINT) RETURNS VARCHAR(1000) BEGIN DECLARE k INT DEFAULT 0; DECLARE fid INT DEFAULT 1; DECLARE str VARCHAR(1000) DEFAULT '$'; WHILE rootId > 0 DO SET fid=(SELECT pid FROM table_name WHERE root_id=id); IF fid > 0 THEN SET str = concat(str,',',fid); SET root_id = fid; ELSE SET root_id=fid; END IF; END WHILE; RETURN str; END // delimiter ;
2、查询当前节点的所有子节点
delimiter // CREATE FUNCTION `getChildList`(root_id BIGINT) RETURNS VARCHAR(1000) BEGIN DECLARE str VARCHAR(1000) ; DECLARE cid VARCHAR(1000) ; DECLARE k INT DEFAULT 0; SET str = '$'; SET cid = CAST(root_id AS CHAR);12 WHILE cid IS NOT NULL DO IF k > 0 THEN SET str = CONCAT(str,',',cid); END IF; SELECT GROUP_CONCAT(id) INTO cid FROM table_name WHERE FIND_IN_SET(pid,cid)>0; SET k = k + 1; END WHILE; RETURN str; END // delimiter ;
三、测试
1、获取当前节点的所有父级
SELECT getParentList(10);
2、获取当前节点的所有字节
SELECT getChildList(3);