MySQL使用FIND_IN_SET和func_find_nodes函数查询树状结构所有子节点

 

       最近在编码的时候碰到查询数据库中某树节点所有子节点的需求，以下是MySQL层设计的解决方案。

一、FIND_IN_SET函数

       首先介绍MySQL中的FIND_IN_SET函数。

FIND_IN_SET(str,strlist)
功能：
返回strlist中str所在的位置索引，strlist必须以","分割开，返回结果为null或记录
参数：
str----要查询的字符串
strlist----字段名 参数以”,”分隔 如 (1,2,6,8)
注意：
索引从1开始

       假如字符串str在由N个子链组成的字符串列表strlist 中，则返回值的范围在 1 到 N 之间。 一个字符串列表就是一个由一些被 ‘,’ 符号分开的子链组成的字符串。如果第一个参数是一个常数字符串，而第二个是type SET列，则FIND_IN_SET() 函数被优化，使用比特计算。 如果str不在strlist 或strlist 为空字符串，则返回值为 0 。如任意一个参数为NULL，则返回值为 NULL。这个函数在第一个参数包含一个逗号(‘,’)时将无法正常运行。

       例如，

SELECT FIND_IN_SET('b', 'a,b,c,d'); 

返回 2 。因为b 在strlist集合中放在2的位置 从1开始

select FIND_IN_SET('1', '1'); 

返回 1。 这时候的strlist集合有点特殊 只有一个字符串 其实就是要求前一个字符串 一定要在后一个字符串集合中才返回大于0的数 

select FIND_IN_SET('2', '1，2');

返回2 
 

select FIND_IN_SET('6', '1'); 

返回0

1）当str为某个字段时，使用find_in_set函数一次返回多条记录 

例如

select * from treenodes where FIND_IN_SET(id, '1,2,3,4,5'); 

id 是一个表的字段，然后每条记录分别是id等于1，2，3，4，5的时候返回

2）当strlist是一个字段的时，当在list中包含该字符串时返回

例如

SELECT * from test WHERE FIND_IN_SET('hello',list);

二、自定义SQL函数func_find_nodes

      使用BFS算法将子节点拼接到sTemp上，其中group_concat(id)函数将同一个分组中的值连接起来，返回一个字符串结果至sTempChd变量中，最后返回所有子节点id值。

CREATE DEFINER=`root`@`%` FUNCTION `func_find_nodes`(`paramId` VARCHAR(100)) RETURNS varchar(5000) CHARSET utf8
    COMMENT '包含当前节点'
BEGIN
	DECLARE sTemp VARCHAR(5000);
  DECLARE sTempChd VARCHAR(5000);
  SET sTemp = '$$$';
  SET sTempChd =CAST(IF(paramId IS NULL OR TRIM(paramId)='',0,paramId) AS CHAR);

  WHILE sTempChd is not null DO

    SET sTemp = concat(sTemp,',',sTempChd);

    SELECT group_concat(id) INTO sTempChd FROM CONTACTS_GROUP where FIND_IN_SET(CAST(parent_id AS UNSIGNED),sTempChd)>0;

  END WHILE;

	RETURN sTemp;
END

最后就可以用一句SQL语句就可以查询该节点下的所有子节点啦。

select id from CONTACTS_GROUP where FIND_IN_SET(id,func_find_nodes(1))
#1为当前节点id