From 27e6bc3df3e39e0d0b147b155a89ad6837ea972b Mon Sep 17 00:00:00 2001
From: cjl <909710561@qq.com>
Date: Mon, 06 Jan 2025 09:19:24 +0800
Subject: [PATCH] Merge branch 'cjl' into dev

---
 screen-common/src/main/java/com/moral/util/TokenUtils.java |  160 +++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 136 insertions(+), 24 deletions(-)

diff --git a/screen-common/src/main/java/com/moral/util/TokenUtils.java b/screen-common/src/main/java/com/moral/util/TokenUtils.java
index cafd03b..e51b7c0 100644
--- a/screen-common/src/main/java/com/moral/util/TokenUtils.java
+++ b/screen-common/src/main/java/com/moral/util/TokenUtils.java
@@ -1,18 +1,20 @@
 package com.moral.util;
 
-import com.moral.constant.Constants;
 import com.moral.constant.ResponseCodeEnum;
 import com.moral.exception.TokenException;
-import com.sun.org.apache.bcel.internal.classfile.ConstantString;
+
 import lombok.extern.slf4j.Slf4j;
+
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Component;
+import org.springframework.util.ObjectUtils;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
 
 
-import java.util.HashMap;
-import java.util.Map;
+import javax.servlet.http.HttpServletRequest;
 import java.util.concurrent.TimeUnit;
 
 /**
@@ -25,7 +27,6 @@
 @Component
 @Slf4j
 public class TokenUtils {
-
     private static RedisTemplate redisTemplate;
 
     @Autowired
@@ -35,10 +36,20 @@
     }
 
     //token��������� ������������
-    private static final int validity_time = 60*30;
+    private static final int validity_time = 60 * 30;
+
+    //������user_token���������������������token
+    public static boolean hHasKey(String uid) {
+        return redisTemplate.opsForHash().hasKey("user_token", uid);
+    }
+
+    //������������id������token
+    public static Object hget(String uid) {
+        return redisTemplate.opsForHash().get("user_token", uid);
+    }
 
     /**
-     * @Description: ������token,���������������������������������
+     * @Description: ������token, ���������������������������������
      * @Param: [uid] type��� ���������������manage  ���������������api
      * @return: java.lang.String
      * @Author: ���������
@@ -55,26 +66,50 @@
                 redisTemplate.delete(oldToken);
             //���token���������value���
             redisTemplate.opsForValue().set(token, userInfo);
-            redisTemplate.expire(token, validity_time, TimeUnit.SECONDS);
+            //redisTemplate.expire(token, validity_time, TimeUnit.SECONDS);
             //���token���������Hash���
             redisTemplate.opsForHash().put("user_token", uid, token);
             return token;
         } catch (Exception e) {
-            log.error("token���������������"+e.getMessage());
+            log.error("token���������������" + e.getMessage());
+            throw new TokenException(ResponseCodeEnum.TOKEN_CREATE_ERROR.getCode(),
+                    ResponseCodeEnum.TOKEN_CREATE_ERROR.getMsg());
+        }
+    }
+    public static String getTokenApp(String uid, Object userInfo) {
+        //������������token
+        try {
+            //������token
+            String token = TokenEncryptUtils.encoded(uid + "/" + System.currentTimeMillis() / 1000);
+            //������������token
+            String oldToken = (String) redisTemplate.opsForHash().get("user_token", uid);
+            if (oldToken != null)
+                redisTemplate.delete(oldToken);
+            //���token���������value���
+            getRefreshTokenApp(token,userInfo);
+            //redisTemplate.expire(token, validity_time, TimeUnit.SECONDS);
+            //���token���������Hash���
+            redisTemplate.opsForHash().put("user_token", uid, token);
+            return token;
+        } catch (Exception e) {
+            log.error("token���������������" + e.getMessage());
             throw new TokenException(ResponseCodeEnum.TOKEN_CREATE_ERROR.getCode(),
                     ResponseCodeEnum.TOKEN_CREATE_ERROR.getMsg());
         }
     }
 
+    public static void getRefreshTokenApp(String token, Object userInfo) {
+        redisTemplate.opsForValue().set(token, userInfo);
+    }
 
     /**
      * @Description: ������token
      * @Param: [type, token] type��� ���������������manage  ���������������api
-     * @return: java.util.Map<java.lang.String   ,   java.lang.Object>
+     * @return: java.util.Map<java.lang.String                                                                                                                               ,                                                                                                                               java.lang.Object>
      * @Author: ���������
      * @Date: 2021/3/10
      */
-    public static void checkToken(String token) {
+    public static boolean checkToken(String token) {
         try {
             String[] tokenArray = TokenEncryptUtils.decoded(token).split("/");
             //������token������������
@@ -88,25 +123,30 @@
                         ResponseCodeEnum.TOKEN_INVALID.getMsg());
             }
         } catch (Exception e) {
-            log.error("token���������������token������" + e.getMessage());
-            throw new TokenException(ResponseCodeEnum.TOKEN_INVALID.getCode(),
-                    ResponseCodeEnum.TOKEN_INVALID.getMsg());
+            return false;
         }
+        return true;
     }
 
+
     /**
-     * @Description: ������token������������������  ������������
+     * @Description: ������token������������������
      * @Param: [token]
-     * @return: java.util.Map<java.lang.String   ,   java.lang.Object>
+     * @return: java.util.Map<java.lang.String                                                                                                                               ,                                                                                                                               java.lang.Object>
      * @Author: ���������
      * @Date: 2021/3/11
      */
     public static Object getUserInfoByToken(String token) {
         Object userInfo = redisTemplate.opsForValue().get(token);
-        if(userInfo==null)
+        if (userInfo == null)
             throw new TokenException(ResponseCodeEnum.TOKEN_INVALID.getCode(),
                     ResponseCodeEnum.TOKEN_INVALID.getMsg());
         return userInfo;
+    }
+
+    public static Object getUserInfo() {
+        String token = getCurrentToken();
+        return getUserInfoByToken(token);
     }
 
     /**
@@ -117,18 +157,90 @@
      * @Date: 2021/3/11
      */
     public static void destoryToken(String uid, String token) {
+        if(ObjectUtils.isEmpty(uid)||ObjectUtils.isEmpty(token))
+            return;
         redisTemplate.delete(token);
         redisTemplate.opsForHash().delete("user_token", uid);
     }
 
+
     /**
-    * @Description: token������
-            * @Param: [token]
-            * @return: void
-            * @Author: ���������
-            * @Date: 2021/3/11
-            */
-    public static void extendTokenTime(String token)  {
+     * @Description: ������token
+     * @Param: [token]
+     * @return: void
+     * @Author: ���������
+     * @Date: 2021/4/1
+     */
+    public static void destoryToken(String token) {
+        destoryToken(getUidByToken(token), token);
+    }
+
+
+    /**
+     * @Description: ���������������������token
+     * @Param: []
+     * @return: void
+     * @Author: ���������
+     * @Date: 2021/5/21
+     */
+    public static void destoryToken() {
+        destoryToken(getCurrentToken());
+    }
+
+    public static void destoryToken(Integer id) {
+        String token = getTokenById(id);
+        destoryToken(String.valueOf(id),token);
+    }
+
+    /**
+     * @Description: ������id������token
+     * @Param: []
+     * @return: java.lang.String
+     * @Author: ���������
+     * @Date: 2021/5/21
+     */
+    public static String getTokenById(Integer id) {
+         return (String) getTokenById(id.toString());
+    }
+
+    public static String getTokenById(String id) {
+        return (String)redisTemplate.opsForHash().get("user_token", String.valueOf(id));
+    }
+
+
+    /**
+     * @Description: ������TOKEN������Id
+     * @Param: [token]
+     * @return: void
+     * @Author: ���������
+     * @Date: 2021/4/1
+     */
+    public static String getUidByToken(String token) {
+        String[] string = TokenEncryptUtils.decoded(token).split("/");
+        return string[0];
+    }
+
+    /**
+     * @Description: ���������������������token
+     * @Param: []
+     * @return: java.lang.String
+     * @Author: ���������
+     * @Date: 2021/5/21
+     */
+    public static String getCurrentToken() {
+        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
+        return request.getHeader("token");
+    }
+
+
+    /**
+     * @Description: token������
+     * @Param: [token]
+     * @return: void
+     * @Author: ���������
+     * @Date: 2021/3/11
+     */
+    public static void extendTokenTime(String token) {
         redisTemplate.expire(token, validity_time, TimeUnit.SECONDS);
     }
 }

--
Gitblit v1.8.0