From 3640ae3908b0ad75baee22ba588e5cce4319e216 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期五, 16 九月 2022 08:53:33 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/common/web/BaseController.java |   17 +++++++++++++++++
 1 files changed, 17 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/zy/crm/common/web/BaseController.java b/src/main/java/com/zy/crm/common/web/BaseController.java
index 8fbe2a9..c8fdd34 100644
--- a/src/main/java/com/zy/crm/common/web/BaseController.java
+++ b/src/main/java/com/zy/crm/common/web/BaseController.java
@@ -7,6 +7,8 @@
 import com.core.common.Cools;
 import com.core.controller.AbstractBaseController;
 import com.core.exception.CoolException;
+import com.zy.crm.system.entity.Dept;
+import com.zy.crm.system.service.DeptService;
 import io.swagger.annotations.ApiModelProperty;
 import org.springframework.beans.factory.annotation.Autowired;
 import com.zy.crm.manager.entity.Node;
@@ -38,6 +40,8 @@
     private TagService tagService;
     @Autowired
     private NodeService nodeService;
+    @Autowired
+    private DeptService deptService;
     @Autowired
     private UserLoginService userLoginService;
 
@@ -79,6 +83,11 @@
         }
         return user;
     }
+
+    protected Long getDeptId() {
+        return getUser().getDeptId();
+    }
+
     protected String getComment(Class<?> cls, String fieldName){
         Field[] fields = Cools.getAllFields(cls);
         for (Field field : fields){
@@ -101,6 +110,14 @@
         return node;
     }
 
+    protected Dept getOriginDept(){
+        Dept dept = deptService.getTop(getHostId());
+        if (dept == null) {
+            throw new CoolException("閮ㄩ棬鏁版嵁閿欒");
+        }
+        return dept;
+    }
+
     protected <T> void hostEq(EntityWrapper<T> wrapper){
         Long hostId = getHostId();
         if (hostId != null) {

--
Gitblit v1.9.1