存储过程递归树结构实现（某一节点下所有子节点）三种方式

2018年11月01日 15:08:41 peach桃阅读数：75

最近项目中碰到需要写递归，就特意实现了集中实现方式，总结一下~

情景：通过给出的机构id，得到该机构id以及所有的子机构信息，机构id是UUID;

通过三种方式去实现，亲自实现过，可直接用：

第一种：通过多次操作数据库获得所有子机构信息。实现如下：

/**
* 递归获取某个机构id下面的所有子机构
* @param comTreeBeanHashSet 节点集合
* @param mechanismId 当前机构ID
*/
private void findChildMechanisms(Set<ComTreeBean> comTreeBeanHashSet, String mechanismId) {
ComTreeBean dep=comTreeDao.findByMechanismId(mechanismId);
List<ComTreeBean> comTreeBeans = new ArrayList<ComTreeBean>();
comTreeBeans.add(dep);
/**
* 加入当前机构
*/
comTreeBeanHashSet.addAll(comTreeBeans.stream().collect(Collectors.toList()));
/**
* 获取子机构
*/
List<ComTreeBean> comTreeBeanPars = comTreeDao.findByParentId(mechanismId);
for (ComTreeBean d : comTreeBeanPars) {
/**
* 递归子单位
*/
findChildMechanisms(comTreeBeanHashSet, d.getUid());
}
}

第二种：不会多次操作数据库，通过一次查询，获取所有机构树信息，再从当前机构去递归获取所有子机构信息

直接使用能获取所有子机构信息，但是不能获取传入机构信息，如果想保留当前传入id机构，在初始化子节点集合时放入当前机构即可，即为：

当然，这两种方式都是机构树的数据不是很大，考虑到如果其他情况下，比如数据量大，就考虑使用第三种方式，通过创建存储过程实现，我也将其实现了一遍，表实例就不给出了，具体实现如下，可自己创建表实例实验：

第三种：存储过程

首先定义执行递归操作的存储过程，如下

-- 首先定义执行递归操作的存储过程：findChildList
CREATE PROCEDURE findChildList(IN id varchar(100))
BEGIN
DECLARE v_dep varchar(100) DEFAULT -1;
DECLARE done varchar(100) DEFAULT 0;
DECLARE C_dep CURSOR FOR SELECT d.id FROM T_COM_INFO d WHERE d.par_com_uid = id;
DECLARE CONTINUE HANDLER FOR NOT found set done=1;
SET @@max_sp_recursion_depth = 10;
INSERT INTO tmp_Dep VALUES (id);
OPEN C_dep;
FETCH C_dep INTO v_dep;
WHILE (done =0)
DO
CALL findChildList(v_dep);
FETCH C_dep INTO v_dep;
END WHILE;
END

然后定义调用递归操作的存储过程，如下

-- 再定义调用递归操作的存储过程：findDepList
CREATE PROCEDURE findDepList(IN id varchar(100))
BEGIN
DROP TEMPORARY TABLE IF EXISTS tmp_Dep;
CREATE TEMPORARY TABLE tmp_Dep(depId varchar(100));
DELETE FROM tmp_Dep;
CALL findChildList(id);
SELECT distinct tmp_Dep.depId,T_COM_INFO.com_name FROM tmp_Dep,T_COM_INFO where tmp_Dep.depId = T_COM_INFO.id order by depId;
END

调用存储过程及删除存储过程，如下：

具体就是这样，顺便给出本人写测试时，id为int类型的存储过程，如下

-- 首先定义执行递归操作的存储过程：findChildList
CREATE PROCEDURE findChildList(IN id BIGINT)
BEGIN
DECLARE v_dep integer DEFAULT -1;
DECLARE done integer DEFAULT 0;
DECLARE C_dep CURSOR FOR SELECT d.id FROM treenodes d WHERE d.pid = id;
DECLARE CONTINUE HANDLER FOR NOT found set done=1;
SET @@max_sp_recursion_depth = 10;
INSERT INTO tmp_Dep VALUES (id);
OPEN C_dep;
FETCH C_dep INTO v_dep;
WHILE (done =0)
DO
CALL findChildList(v_dep);
FETCH C_dep INTO v_dep;
END WHILE;
END
-- 再定义调用递归操作的存储过程：findDepList
CREATE PROCEDURE findDepList(IN id BIGINT)
BEGIN
DROP TEMPORARY TABLE IF EXISTS tmp_Dep;
CREATE TEMPORARY TABLE tmp_Dep(depId INTEGER);
DELETE FROM tmp_Dep;
CALL findChildList(id);
SELECT distinct tmp_Dep.depId,treenodes.nodename FROM tmp_Dep,treenodes where tmp_Dep.depId = treenodes.id order by depId;
END
drop procedure if exists findChildList
drop procedure if exists findDepList
call findDepList(3)

【SQL Server】存储过程遍历树