From 30f0bdc754cc1d85a72c5cb86b84ca20a67ef20c Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期六, 16 四月 2022 14:08:11 +0800
Subject: [PATCH] update 使用 spring cglib 替换 停止维护的 cglib

---
 ruoyi-common/src/main/java/com/ruoyi/common/helper/LoginHelper.java |   37 +++++++++++++++++++++++++++++++------
 1 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/helper/LoginHelper.java b/ruoyi-common/src/main/java/com/ruoyi/common/helper/LoginHelper.java
index ddd5018..7875c56 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/helper/LoginHelper.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/helper/LoginHelper.java
@@ -2,6 +2,7 @@
 
 import cn.dev33.satoken.stp.StpUtil;
 import cn.hutool.core.util.ObjectUtil;
+import com.ruoyi.common.constant.UserConstants;
 import com.ruoyi.common.core.domain.model.LoginUser;
 import com.ruoyi.common.enums.DeviceType;
 import com.ruoyi.common.enums.UserType;
@@ -12,34 +13,43 @@
 
 /**
  * 鐧诲綍閴存潈鍔╂墜
- * 涓洪�傞厤澶氱鐧诲綍鑰屽皝瑁�
+ * 
+ * user_type 涓� 鐢ㄦ埛绫诲瀷 鍚屼竴涓敤鎴疯〃 鍙互鏈夊绉嶇敤鎴风被鍨� 渚嬪 pc,app
+ * deivce 涓� 璁惧绫诲瀷 鍚屼竴涓敤鎴风被鍨� 鍙互鏈� 澶氱璁惧绫诲瀷 渚嬪 web,ios
+ * 鍙互缁勬垚 鐢ㄦ埛绫诲瀷涓庤澶囩被鍨嬪瀵瑰鐨� 鏉冮檺鐏垫椿鎺у埗
+ *
+ * 澶氱敤鎴蜂綋绯� 閽堝 澶氱鐢ㄦ埛绫诲瀷 浣嗘潈闄愭帶鍒朵笉涓�鑷�
+ * 鍙互缁勬垚 澶氱敤鎴风被鍨嬭〃涓庡璁惧绫诲瀷 鍒嗗埆鎺у埗鏉冮檺
  *
  * @author Lion Li
  */
 @NoArgsConstructor(access = AccessLevel.PRIVATE)
 public class LoginHelper {
 
-    private static final String LOGIN_USER_KEY = "loginUser";
+    public static final String JOIN_CODE = ":";
+    public static final String LOGIN_USER_KEY = "loginUser";
+
     private static final ThreadLocal<LoginUser> LOGIN_CACHE = new ThreadLocal<>();
 
     /**
      * 鐧诲綍绯荤粺
-     * 閽堝涓ゅ鐢ㄦ埛浣撶郴
      *
      * @param loginUser 鐧诲綍鐢ㄦ埛淇℃伅
      */
     public static void login(LoginUser loginUser) {
+        LOGIN_CACHE.set(loginUser);
         StpUtil.login(loginUser.getLoginId());
         setLoginUser(loginUser);
     }
 
     /**
      * 鐧诲綍绯荤粺 鍩轰簬 璁惧绫诲瀷
-     * 閽堝涓�濂楃敤鎴蜂綋绯�
+     * 閽堝鐩稿悓鐢ㄦ埛浣撶郴涓嶅悓璁惧
      *
      * @param loginUser 鐧诲綍鐢ㄦ埛淇℃伅
      */
     public static void loginByDevice(LoginUser loginUser, DeviceType deviceType) {
+        LOGIN_CACHE.set(loginUser);
         StpUtil.login(loginUser.getLoginId(), deviceType.getDevice());
         setLoginUser(loginUser);
     }
@@ -49,7 +59,6 @@
      */
     public static void setLoginUser(LoginUser loginUser) {
         StpUtil.getTokenSession().set(LOGIN_USER_KEY, loginUser);
-        LOGIN_CACHE.set(loginUser);
     }
 
     /**
@@ -80,7 +89,9 @@
             String userId = null;
             for (UserType value : UserType.values()) {
                 if (StringUtils.contains(loginId, value.getUserType())) {
-                    userId = StringUtils.replace(loginId, value.getUserType(), StringUtils.EMPTY);
+                    String[] strs = StringUtils.split(loginId, JOIN_CODE);
+                    // 鐢ㄦ埛id鍦ㄦ�绘槸鍦ㄦ渶鍚�
+                    userId = strs[strs.length - 1];
                 }
             }
             if (StringUtils.isBlank(userId)) {
@@ -113,4 +124,18 @@
         return UserType.getUserType(loginId);
     }
 
+    /**
+     * 鏄惁涓虹鐞嗗憳
+     *
+     * @param userId 鐢ㄦ埛ID
+     * @return 缁撴灉
+     */
+    public static boolean isAdmin(Long userId) {
+        return UserConstants.ADMIN_ID.equals(userId);
+    }
+
+    public static boolean isAdmin() {
+        return isAdmin(getUserId());
+    }
+
 }

--
Gitblit v1.9.3