modify

Helius

2021-01-22 81406fbab9b6be9c61c115f5daf3f9bbfa4d256a

 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);
@@ -51,7 +66,7 @@
        }

        String userStr = redisClient.getCachedValue(token);
        redisClient.resetExpire(token);
//        redisClient.resetExpire(token);
        if (StringUtils.isBlank(userStr)) {
            ajaxResult.setInfo("login time out");
            response.getWriter().write(new ObjectMapper().writeValueAsString(ajaxResult));
@@ -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 "";
                }