MySQL通过自定义函数实现递归查询父级ID或者子级ID

时间:2022-08-28 16:57:53

背 景:

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

但很多时候我们是无法控制或者是知道树的深度的。这时就需要在MySQL中用存储过程(函数)来实现或者在程序中使用递归来实现。本文讨论在MySQL中使用函数来实现的方法:

一、环境准备

1、建表

?
1
2
3
4
5
6
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、插入数据

?
1
2
3
4
5
6
7
8
9
10
11
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、查询当前节点的所有父级节点

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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、查询当前节点的所有子节点

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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、获取当前节点的所有父级

?
1
SELECT getParentList(10);

2、获取当前节点的所有字节

?
1
SELECT getChildList(3);

总结

以上所述是小编给大家介绍的MySQL通过自定义函数实现递归查询父级ID或者子级ID,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对服务器之家网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

原文链接:https://www.cnblogs.com/cmacro/archive/2019/11/26/11937341.html