From c554c7ad5405bafa149dbb6429e8245f78b33ffc Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Wed, 23 Dec 2020 15:59:53 +0800 Subject: [PATCH] modify --- zq-erp/src/main/java/com/matrix/system/common/interceptor/ApiUserLoginInterceptor.java | 24 ++++++++++++++++++++---- 1 files changed, 20 insertions(+), 4 deletions(-) diff --git a/zq-erp/src/main/java/com/matrix/system/common/interceptor/ApiUserLoginInterceptor.java b/zq-erp/src/main/java/com/matrix/system/common/interceptor/ApiUserLoginInterceptor.java index f783a97..f538f16 100644 --- a/zq-erp/src/main/java/com/matrix/system/common/interceptor/ApiUserLoginInterceptor.java +++ b/zq-erp/src/main/java/com/matrix/system/common/interceptor/ApiUserLoginInterceptor.java @@ -10,6 +10,9 @@ import com.matrix.core.tools.RSAUtils; import com.matrix.core.tools.StringUtils; import com.matrix.system.common.bean.SysUsers; +import com.matrix.system.common.dao.SysUsersDao; +import com.matrix.system.hive.bean.SysVipInfo; +import com.matrix.system.hive.dao.SysVipInfoDao; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; @@ -32,11 +35,23 @@ @Value("${login_private_key}") private String privateKey; + @Value("${evn}") + private String evn; + + @Autowired + private SysUsersDao sysUsersDao; + private final String TOKEN_HEADER = "Authorization"; private final String TOKEN_START_WITH = "Bearer "; @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { + if ("dev".equals(evn)) { + SysUsers sysUsers = sysUsersDao.selectById(1012L); + request.getSession().setAttribute(MatrixConstance.LOGIN_KEY, sysUsers); + return true; + } + String token = resolveToken(request); AjaxResult ajaxResult = new AjaxResult(); ajaxResult.setStatus(AjaxResult.STATUS_LOGIN_INVALID); @@ -82,21 +97,22 @@ */ private String resolveToken(HttpServletRequest request) { String headToken = request.getHeader(TOKEN_HEADER); - StringBuffer sb = request.getRequestURL(); + String sb = request.getRequestURI(); if (StringUtils.isNotBlank(headToken) && headToken.startsWith(TOKEN_START_WITH)) { // 去掉令牌前缀 String rsaToken = headToken.replace(TOKEN_START_WITH, ""); try { - String decryptTokten = new String(RSAUtils.decryptByPrivateKey(rsaToken, privateKey)); + String decryptToken = new String(RSAUtils.decryptByPrivateKey(rsaToken, privateKey)); - String[] s = decryptTokten.split("_"); + String[] s = decryptToken.split("_"); if (s == null || s.length != 3) { return ""; } - if (!sb.toString().equals(s[2])) { + LogUtil.info("请求路径:{} -- {}", sb, s[2]); + if (!sb.equals(s[2])) { return ""; } -- Gitblit v1.9.1