From 18cae781ffbf230c961d414be91e2c5137503bd6 Mon Sep 17 00:00:00 2001
From: baizonghao <1719256278@qq.com>
Date: 星期六, 18 三月 2023 11:32:28 +0800
Subject: [PATCH] 部门导入修改

---
 ycl-common/src/main/java/com/ycl/utils/listener/DepartListener.java |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/utils/listener/DepartListener.java b/ycl-common/src/main/java/com/ycl/utils/listener/DepartListener.java
index 267d9a7..3313a43 100644
--- a/ycl-common/src/main/java/com/ycl/utils/listener/DepartListener.java
+++ b/ycl-common/src/main/java/com/ycl/utils/listener/DepartListener.java
@@ -5,7 +5,9 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.ycl.dto.UmsDepartDto;
 import com.ycl.entity.depart.UmsDepart;
+import com.ycl.entity.dict.DataDictionary;
 import com.ycl.mapper.depart.UmsDepartMapper;
+import com.ycl.mapper.dict.DataDictionaryMapper;
 import org.springframework.stereotype.Component;
 
 import javax.annotation.Resource;
@@ -16,6 +18,9 @@
 
     @Resource
     private UmsDepartMapper sccgDepartMapper;
+
+    @Resource
+    DataDictionaryMapper dataDictionaryMapper;
 
 
     @Override
@@ -36,7 +41,12 @@
         umsDepart.setDepartDes(umsDepartDto.getDepartDes());
 
         //  璁剧疆閮ㄩ棬绫诲瀷
-        umsDepart.setDepartType(umsDepartDto.getDepartType());
+        DataDictionary dataDictionary = dataDictionaryMapper.selectOne(new LambdaQueryWrapper<DataDictionary>().eq(DataDictionary::getName, umsDepartDto.getDepartType()));
+        if (Objects.isNull(dataDictionary)){
+            throw new RuntimeException("閮ㄩ棬绫诲瀷涓嶅瓨鍦�");
+        }else {
+            umsDepart.setDepartType(new Long(dataDictionary.getId()).intValue());
+        }
 
         //  璁剧疆鐖秈d
         UmsDepart two = sccgDepartMapper.selectOne(new LambdaQueryWrapper<UmsDepart>().eq(UmsDepart::getDepartName, umsDepartDto.getParentDepartName()));

--
Gitblit v1.8.0