From e1199724d209c29a7b640bc96d2ffa32acf381dc Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期二, 20 八月 2024 09:10:46 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/service/TokenService.java |    9 ++++++++-
 1 files changed, 8 insertions(+), 1 deletions(-)

diff --git a/ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/service/TokenService.java b/ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/service/TokenService.java
index de9696d..2cdbed5 100644
--- a/ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/service/TokenService.java
+++ b/ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/service/TokenService.java
@@ -5,6 +5,7 @@
 import java.util.concurrent.TimeUnit;
 import javax.servlet.http.HttpServletRequest;
 
+import com.ruoyi.common.core.exception.user.UserAppletException;
 import com.ruoyi.system.api.model.LoginUserApplet;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
@@ -59,6 +60,7 @@
         Map<String, Object> claimsMap = new HashMap<String, Object>();
         claimsMap.put(SecurityConstants.USER_KEY, token);
         claimsMap.put(SecurityConstants.DETAILS_USER_ID, userId);
+        claimsMap.put(SecurityConstants.USER_TYPE, "system");
         claimsMap.put(SecurityConstants.DETAILS_USERNAME, userName);
 
         // 接口返回信息
@@ -81,6 +83,7 @@
         Map<String, Object> claimsMap = new HashMap<String, Object>();
         claimsMap.put(SecurityConstants.USER_APPLET_KEY, token);
         claimsMap.put(SecurityConstants.DETAILS_USER_ID, userId);
+        claimsMap.put(SecurityConstants.USER_TYPE, "applet");
         claimsMap.put(SecurityConstants.DETAILS_USERNAME, name);
         // 接口返回信息
         Map<String, Object> rspMap = new HashMap<String, Object>();
@@ -89,7 +92,11 @@
         return rspMap;
     }
     public LoginUserApplet getLoginUserApplet() {
-        return getLoginUserAppletToken(ServletUtils.getRequest());
+        LoginUserApplet loginUserAppletToken = getLoginUserAppletToken(ServletUtils.getRequest());
+        if (loginUserAppletToken == null){
+            throw new UserAppletException("登录失效,请重新登录!", 401);
+        }
+        return loginUserAppletToken;
     }
     public LoginUserApplet getLoginUserAppletToken(HttpServletRequest request) {
         // 获取请求携带的令牌

--
Gitblit v1.7.1