首页 > 数据库> 文章详细

mysql自定义函数递归查询部门层级

2021-12-11 15:03:16 阅读：178 来源： 互联网

标签：orgNamePath SET 自定义 pOrgId orgName 层级 mysql org id

最近有个项目需要在很多前端页面显示当前员工的部门，但是这个部门是显示当前部门的所有父级部门，比如当前为研发部，在前台页面需要显示为

“集团公司>子公司>研发部” 的格式，因为很多页面都需要这种数据，在查询时关联查询可能比较麻烦，所以选择自定义一个mysql函数来查询这种格式。

现在需要的是从子级查询父级部门，是从下往上查，可能父级部门只有一个，也可能有多个父级部门，所以我们需要for循环来递归查询。

为了举个栗子，新建表结构及插入数据

SET NAMES utf8mb4;
SET FOREIGN_KEY_CHECKS = 0;

-- ----------------------------
-- Table structure for sa_org
-- ----------------------------
DROP TABLE IF EXISTS `sa_org`;
CREATE TABLE `sa_org`  (
  `id` varchar(32) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL,
  `p_org_id` varchar(25) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,
  `org_name` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,
  PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;

-- ----------------------------
-- Records of sa_org
-- ----------------------------
INSERT INTO `sa_org` VALUES ('27193728124848239', '0', '集团公司');
INSERT INTO `sa_org` VALUES ('27193728124848240', '27193728124848239', '子公司1');
INSERT INTO `sa_org` VALUES ('27193728124848241', '27193728124848240', '合规部');

SET FOREIGN_KEY_CHECKS = 1;

这里插入三条数据，比如当前某个员工是合规部，需要显示为如下格式：

这里我们设置最顶级的部门id为0，当当前部门的id为0时默认退出循环，所以我们可以这样写函数：

DROP FUNCTION IF EXISTS getOrgParent;

CREATE  FUNCTION `getOrgParent`(orgId varchar(100)) RETURNS varchar(1000) CHARSET utf8
        
BEGIN 
DECLARE orgNamePath VARCHAR(1000);
DECLARE pOrgId VARCHAR(32);
DECLARE orgName VARCHAR(128);

SET orgNamePath = '';
SET pOrgId = orgId;
set orgName = '';

WHILE pOrgId <> '0'
DO

SELECT p_org_id, org_name INTO pOrgId, orgName FROM sa_org WHERE id = pOrgId;

IF orgNamePath = '' THEN
    SET orgNamePath = orgName;
ELSE
    SET orgNamePath = CONCAT(orgName, '>', orgNamePath);
END IF;

END WHILE;
RETURN orgNamePath;
END

执行完成，查询试一下：

select getOrgParent('27193728124848241')

select getOrgParent('27193728124848240')

可以看到实现了我们的需求，在查询表格数据时，直接使用这个函数就可以将org_id变为部门层级展示了。

但是当我们输入一个不存在于这个表的数据时，会发现这条查询一直在执行，构成了死循环，所以我们改成如下：

drop  FUNCTION if EXISTS `getOrgParent`;
CREATE FUNCTION `getOrgParent`(orgId varchar(100)) RETURNS varchar(1000) CHARSET utf8
BEGIN 
DECLARE orgNamePath VARCHAR(1000);
DECLARE pOrgId VARCHAR(32);
DECLARE orgName VARCHAR(128);
DECLARE count INT;

SET orgNamePath = '';
SET pOrgId = orgId;
set orgName = '';
set count = 1;

WHILE pOrgId <> '0' and count <> 0
DO

SELECT p_org_id, org_name, count(1) INTO pOrgId, orgName, count FROM sa_org WHERE id = pOrgId;

IF orgNamePath = '' THEN
    SET orgNamePath = orgName;
ELSE
    SET orgNamePath = CONCAT(orgName, '>', orgNamePath);
END IF;

END WHILE;
RETURN orgNamePath;
END

可以看到即使现在这条数据不存在也不会死循环了。

标签：orgNamePath,SET,自定义,pOrgId,orgName,层级,mysql,org,id
来源： https://www.cnblogs.com/LiuFqiang/p/15675634.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

mysql自定义函数递归查询部门层级