From 0b41cffa071b82e96c53f9f0fc394259442e24d1 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 08 Jun 2020 16:40:39 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java |   25 +++++++++++++++++--------
 1 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
index 0e5624d..8b98b0b 100644
--- a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
+++ b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java
@@ -4,6 +4,7 @@
 import cn.hutool.crypto.asymmetric.KeyType;
 import cn.hutool.crypto.asymmetric.RSA;
 import com.alibaba.fastjson.JSONObject;
+import com.xcong.excoin.common.LoginUserUtils;
 import com.xcong.excoin.common.contants.AppContants;
 import com.xcong.excoin.common.system.bean.LoginUserBean;
 import com.xcong.excoin.configurations.properties.ApplicationProperties;
@@ -43,12 +44,20 @@
         HttpServletRequest request = (HttpServletRequest) servletRequest;
         String token = resolveToken(request);
         if (StrUtil.isNotBlank(token)) {
-            String loginStr = (String) redisUtils.get(AppContants.APP_LOGIN_PREFIX + token);
+            String redisKey = "";
+            // 根据user-agent判断pc端还是app端
+            if (LoginUserUtils.isBrowser(request)) {
+                redisKey = AppContants.PC_LOGIN_PREFIX + token;
+            } else {
+                redisKey = AppContants.APP_LOGIN_PREFIX + token;
+            }
+
+            String loginStr = (String) redisUtils.get(redisKey);
             if (StrUtil.isNotBlank(loginStr)) {
                 LoginUserBean loginUser = JSONObject.parseObject(loginStr, LoginUserBean.class);
                 Authentication authentication = new UsernamePasswordAuthenticationToken(loginUser.getMemberEntity(), token, new ArrayList<>());
                 SecurityContextHolder.getContext().setAuthentication(authentication);
-                redisUtils.expire(AppContants.APP_LOGIN_PREFIX + token, 300000);
+                redisUtils.expire(redisKey, 300000);
             } else {
                 SecurityContextHolder.clearContext();
             }
@@ -68,12 +77,12 @@
     private String resolveToken(HttpServletRequest request) {
         try {
             // TODO debug模式下写死用户
-            String bearerToken = "";
-            if (applicationProperties.isDebug()) {
-                bearerToken = "Bearer JSEre1ZUKEu2Ga5ORM+juxXv6yBwmt+FgLhxaeHf1EEJfIb3oRir4pXqe5JDhS6sXfLYOXRIAyBpq+SYBwAtGigxwzGVPn+k4Pt6vNxZ4h8Pk4IeG4+FqbFD0guzvu3WN2eRnnzYqCepl429v9Ju7n4jSG0Hj5ViM3MHQZs3qHo=";
-            } else {
-                bearerToken = request.getHeader(AppContants.TOKEN_HEADER);
-            }
+            String bearerToken = request.getHeader(AppContants.TOKEN_HEADER);
+//            if (applicationProperties.isDebug()) {
+//                bearerToken = "Bearer JSEre1ZUKEu2Ga5ORM+juxXv6yBwmt+FgLhxaeHf1EEJfIb3oRir4pXqe5JDhS6sXfLYOXRIAyBpq+SYBwAtGigxwzGVPn+k4Pt6vNxZ4h8Pk4IeG4+FqbFD0guzvu3WN2eRnnzYqCepl429v9Ju7n4jSG0Hj5ViM3MHQZs3qHo=";
+//            } else {
+//                bearerToken = request.getHeader(AppContants.TOKEN_HEADER);
+//            }
             if (StringUtils.hasText(bearerToken) && bearerToken.startsWith(AppContants.TOKEN_START_WITH)) {
                 // 去掉令牌前缀
                 String rsaToken = bearerToken.replace(AppContants.TOKEN_START_WITH, "");

--
Gitblit v1.9.1