From 09ee5added9d59e90310a2586e846137ea597b19 Mon Sep 17 00:00:00 2001
From: LSH
Date: 星期五, 01 十二月 2023 13:15:42 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/common/web/BaseController.java |   13 +++++++++++++
 1 files changed, 13 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 6216688..8855433 100644
--- a/src/main/java/com/zy/crm/common/web/BaseController.java
+++ b/src/main/java/com/zy/crm/common/web/BaseController.java
@@ -80,6 +80,11 @@
         return Long.parseLong(String.valueOf(request.getAttribute("userId")));
     }
 
+    protected Long getUserId(String token){
+        UserLogin userLogin = userLoginService.selectOne(new EntityWrapper<UserLogin>().eq("token", token));
+        return userLogin.getUserId();
+    }
+
     protected User getUser(){
         User user = userService.selectById(getUserId());
         if (null == user) {
@@ -88,6 +93,14 @@
         return user;
     }
 
+    protected User getUser(String token){
+        User user = userService.selectById(getUserId(token));
+        if (null == user) {
+            throw new CoolException(BaseRes.DENIED);
+        }
+        return user;
+    }
+
     protected Long getDeptId() {
         return getUser().getDeptId();
     }

--
Gitblit v1.9.1