From bb5e59c8d33db35940a28dcde66ae6ae16a88558 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期一, 12 九月 2022 16:39:56 +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..f157038 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();
+        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