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

背 景:

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

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

一、环境准备

1、建表

1 CREATE TABLE `table_name`  (2   `id` int(11) NOT NULL AUTO_INCREMENT,3   `status` int(255) NULL DEFAULT NULL,4   `pid` int(11) NULL DEFAULT NULL,5   PRIMARY KEY (`id`) USING BTREE6 ) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;

2、插入数据

1 INSERT INTO `table_name` VALUES (1, 12, 0); 2 INSERT INTO `table_name` VALUES (2, 4, 1); 3 INSERT INTO `table_name` VALUES (3, 8, 2); 4 INSERT INTO `table_name` VALUES (4, 16, 3); 5 INSERT INTO `table_name` VALUES (5, 32, 3); 6 INSERT INTO `table_name` VALUES (6, 64, 3); 7 INSERT INTO `table_name` VALUES (7, 128, 6); 8 INSERT INTO `table_name` VALUES (8, 256, 7); 9 INSERT INTO `table_name` VALUES (9, 512, 8);10 INSERT INTO `table_name` VALUES (10, 1024, 9);11 INSERT INTO `table_name` VALUES (11, 2048, 10);

二、MySQL函数的编写

1、查询当前节点的所有父级节点

1 delimiter //  2 CREATE FUNCTION `getParentList`(root_id BIGINT)  3      RETURNS VARCHAR(1000)  4      BEGIN  5           DECLARE k INT DEFAULT 0; 6         DECLARE fid INT DEFAULT 1; 7         DECLARE str VARCHAR(1000) DEFAULT '$'; 8         WHILE rootId > 0 DO 9               SET fid=(SELECT pid FROM table_name WHERE root_id=id); 10               IF fid > 0 THEN11                   SET str = concat(str,',',fid);   12                   SET root_id = fid;  13               ELSE 14                   SET root_id=fid;  15               END IF;  16      END WHILE;17    RETURN str;18  END  //19  delimiter ;

2、查询当前节点的所有子节点

1   2  delimiter // 3  CREATE FUNCTION `getChildList`(root_id BIGINT)  4      RETURNS VARCHAR(1000)  5      BEGIN  6        DECLARE str VARCHAR(1000) ;  7        DECLARE cid VARCHAR(1000) ;  8        DECLARE k INT DEFAULT 0; 9        SET str = '$'; 10        SET cid = CAST(root_id AS CHAR);12        WHILE cid IS NOT NULL DO  13                 IF k > 0 THEN14                   SET str = CONCAT(str,',',cid);15                 END IF;16                 SELECT GROUP_CONCAT(id) INTO cid FROM table_name WHERE FIND_IN_SET(pid,cid)>0;17                 SET k = k + 1;18        END WHILE; 19        RETURN str; 20 END //  21 delimiter ;

三、测试

1、获取当前节点的所有父级

SELECT getParentList(10);

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

SELECT getChildList(3);

本文完......

(0)

相关推荐