下载视频

yzg-util-db/src/main/java/com/yanzuoguang/token/TokenHelper.java

@@ -3,7 +3,6 @@ package com.yanzuoguang.token;
 import com.yanzuoguang.util.YzgError;
 import com.yanzuoguang.util.base.ObjectHelper;
 import com.yanzuoguang.util.cache.MemoryCache;
-import com.yanzuoguang.util.exception.CodeException;
 import com.yanzuoguang.util.helper.JsonHelper;
 import com.yanzuoguang.util.helper.StringHelper;
 import com.yanzuoguang.util.vo.MapRow;
@@ -73,6 +72,18 @@ public class TokenHelper {
      * @return 缓存的数据
      */
     public static <T extends Object> T get(boolean checkFlag, Class<T> cls) {
+        return get(checkFlag, cls, true);
+    }
+
+    /**
+     * 获取当前的登录信息
+     *
+     * @param checkFlag 是否抛出异常
+     * @param cls       需要获取的数据的类型
+     * @param <T>       数据类型
+     * @return 缓存的数据
+     */
+    public static <T extends Object> T get(boolean checkFlag, Class<T> cls, boolean readNewToken) {
         String id = getCurrentId();
         TokenData tokenData = cache.get(id);
         if (tokenData == null) {
@@ -81,20 +92,23 @@ public class TokenHelper {
             }
             return null;
         }
-        boolean isEmpty = tokenData.getData() == null || tokenData.getData().isEmpty();
-        boolean isExpire = tokenData.getExpire() < System.currentTimeMillis();
-        boolean isToken = !StringHelper.isEmpty(tokenData.getToken()) && (isEmpty || isExpire);
-        if (isToken) {
-            tokenData = tokenLoad.load(tokenData.getToken());
-            if (tokenData != null) {
-                cache.put(id, tokenData);
+        // 获取最新的token
+        if (readNewToken) {
+            boolean isEmpty = tokenData.getData() == null || tokenData.getData().isEmpty();
+            boolean isExpire = tokenData.getExpire() < System.currentTimeMillis();
+            boolean isToken = !StringHelper.isEmpty(tokenData.getToken()) && (isEmpty || isExpire);
+            if (isToken) {
+                tokenData = tokenLoad.load(tokenData.getToken());
+                if (tokenData != null) {
+                    cache.put(id, tokenData);
+                }
             }
-        }
-        if (tokenData == null || tokenData.getData() == null) {
-            if (checkFlag) {
-                throw YzgError.getRuntimeException("061");
+            if (tokenData == null || tokenData.getData() == null) {
+                if (checkFlag) {
+                    throw YzgError.getRuntimeException("061");
+                }
+                return null;
             }
-            return null;
         }
 
         if (ObjectHelper.isSub(cls, tokenData.getData().getClass())) {