From 518b7a20ef1add6c6910eca89b1fef0fbd83e03d Mon Sep 17 00:00:00 2001
From: whycq <you@example.com>
Date: 星期日, 01 一月 2023 07:03:19 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/common/web/BaseController.java |   12 ++++++++++++
 1 files changed, 12 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 ca4ec88..eab395b 100644
--- a/src/main/java/com/zy/crm/common/web/BaseController.java
+++ b/src/main/java/com/zy/crm/common/web/BaseController.java
@@ -8,7 +8,9 @@
 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;
@@ -44,6 +46,8 @@
     private DeptService deptService;
     @Autowired
     private UserLoginService userLoginService;
+    @Autowired
+    private RoleService roleService;
 
     protected Long getHostId(){
         if (getUserId() == 9527) {
@@ -88,6 +92,14 @@
         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){

--
Gitblit v1.9.1