From 0e8731a031b810fa4dff7d127dd4912cc58d5f10 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期四, 24 十一月 2022 15:01:13 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/common/web/BaseController.java |   32 ++++++++++++++++++++++++++++++++
 1 files changed, 32 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..eab395b 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,10 @@
 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.entity.Role;
+import com.zy.crm.system.service.DeptService;
+import com.zy.crm.system.service.RoleService;
 import io.swagger.annotations.ApiModelProperty;
 import org.springframework.beans.factory.annotation.Autowired;
 import com.zy.crm.manager.entity.Node;
@@ -39,7 +43,11 @@
     @Autowired
     private NodeService nodeService;
     @Autowired
+    private DeptService deptService;
+    @Autowired
     private UserLoginService userLoginService;
+    @Autowired
+    private RoleService roleService;
 
     protected Long getHostId(){
         if (getUserId() == 9527) {
@@ -79,6 +87,19 @@
         }
         return user;
     }
+
+    protected Long getDeptId() {
+        return getUser().getDeptId();
+    }
+
+    protected Role getRole(){
+        Role role = roleService.selectById(getUser().getRoleId());
+        if (null == role) {
+            throw new CoolException(BaseRes.DENIED);
+        }
+        return role;
+    }
+
     protected String getComment(Class<?> cls, String fieldName){
         Field[] fields = Cools.getAllFields(cls);
         for (Field field : fields){
@@ -99,6 +120,14 @@
             throw new CoolException("搴撳尯鏁版嵁閿欒");
         }
         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){
@@ -170,6 +199,9 @@
             }
             String column = null;
             if (field.isAnnotationPresent(TableField.class)) {
+                if (!field.getAnnotation(TableField.class).exist()) {
+                    continue;
+                }
                 column = field.getAnnotation(TableField.class).value();
             }
             if (Cools.isEmpty(column)) {

--
Gitblit v1.9.1