From ded93a205fc7176b98c24c6e56ad76e9db998b02 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Fri, 10 Jul 2020 11:40:11 +0800 Subject: [PATCH] modify --- src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java | 25 ++++++++++++++++++++----- 1 files changed, 20 insertions(+), 5 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 6ebf4d8..2284d12 100644 --- a/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java +++ b/src/main/java/com/xcong/excoin/configurations/security/TokenFilter.java @@ -4,10 +4,12 @@ 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; import com.xcong.excoin.configurations.properties.SecurityProperties; +import com.xcong.excoin.modules.member.entity.MemberEntity; import com.xcong.excoin.utils.RedisUtils; import com.xcong.excoin.utils.SpringContextHolder; import lombok.extern.slf4j.Slf4j; @@ -43,12 +45,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<>()); + MemberEntity loginUser = JSONObject.parseObject(loginStr, MemberEntity.class); + Authentication authentication = new UsernamePasswordAuthenticationToken(loginUser, token, new ArrayList<>()); SecurityContextHolder.getContext().setAuthentication(authentication); - redisUtils.expire(AppContants.APP_LOGIN_PREFIX + token, 300000); + redisUtils.expire(redisKey, 300000); } else { SecurityContextHolder.clearContext(); } @@ -67,13 +77,18 @@ */ private String resolveToken(HttpServletRequest request) { try { + // TODO debug模式下写死用户 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, ""); RSA rsa = new RSA(securityProperties.getPrivateKey(), null); String[] tokens = StrUtil.split(rsa.decryptStr(rsaToken, KeyType.PrivateKey), "_"); - if (verifyTokenExpired(Long.parseLong(tokens[1]))) { return tokens[0]; } -- Gitblit v1.9.1