Merge branch 'master' of http://114.132.189.42:9002/r/product-inventory-man...

liyong

昨天 ba4af275c3b8d073f863be7b12e43974b3ef14a2

 src/main/java/com/ruoyi/framework/security/service/TokenService.java

@@ -1,12 +1,21 @@
package com.ruoyi.framework.security.service;



import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.concurrent.TimeUnit;

import java.util.stream.Collectors;

import javax.servlet.http.HttpServletRequest;



import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;

import com.ruoyi.project.system.domain.SysUserDept;

import com.ruoyi.project.system.mapper.SysUserDeptMapper;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.stereotype.Component;

import com.ruoyi.common.constant.CacheConstants;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.utils.ServletUtils;

import com.ruoyi.common.utils.StringUtils;

@@ -19,6 +28,7 @@
import io.jsonwebtoken.Claims;

import io.jsonwebtoken.Jwts;

import io.jsonwebtoken.SignatureAlgorithm;

import org.springframework.util.CollectionUtils;



/**

 * token验证处理

@@ -28,6 +38,8 @@
@Component

public class TokenService

{

    private static final Logger log = LoggerFactory.getLogger(TokenService.class);



    // 令牌自定义标识

    @Value("${token.header}")

    private String header;

@@ -44,7 +56,7 @@


    protected static final long MILLIS_MINUTE = 60 * MILLIS_SECOND;



    private static final Long MILLIS_MINUTE_TEN = 20 * 60 * 1000L;

    private static final Long MILLIS_MINUTE_TWENTY = 20 * 60 * 1000L;



    @Autowired

    private RedisCache redisCache;

@@ -71,6 +83,7 @@
            }

            catch (Exception e)

            {

                log.error("获取用户信息异常'{}'", e.getMessage());

            }

        }

        return null;

@@ -114,24 +127,28 @@


        Map<String, Object> claims = new HashMap<>();

        claims.put(Constants.LOGIN_USER_KEY, token);

        claims.put(Constants.JWT_USERNAME, loginUser.getUsername());

        return createToken(claims);

    }



    /**

     * 验证令牌有效期，相差不足20分钟，自动刷新缓存

     * 

     * @param token 令牌

     * @param loginUser 登录信息

     * @return 令牌

     */

    public void verifyToken(LoginUser loginUser)

    {

        long expireTime = loginUser.getExpireTime();

        long currentTime = System.currentTimeMillis();

        if (expireTime - currentTime <= MILLIS_MINUTE_TEN)

        if (expireTime - currentTime <= MILLIS_MINUTE_TWENTY)

        {

            refreshToken(loginUser);

        }

    }



    @Autowired

    private SysUserDeptMapper sysUserDeptMapper;



    /**

     * 刷新令牌有效期

@@ -142,11 +159,18 @@
    {

        loginUser.setLoginTime(System.currentTimeMillis());

        loginUser.setExpireTime(loginUser.getLoginTime() + expireTime * MILLIS_MINUTE);

        LambdaQueryWrapper<SysUserDept> sysUserDeptLambdaQueryWrapper = new LambdaQueryWrapper<>();

        sysUserDeptLambdaQueryWrapper.eq(SysUserDept::getUserId, loginUser.getUserId());

        List<SysUserDept> sysUserDept = sysUserDeptMapper.selectList(sysUserDeptLambdaQueryWrapper);

        if(!CollectionUtils.isEmpty(sysUserDept)){

            List<Long> collect = sysUserDept.stream().map(SysUserDept::getDeptId).collect(Collectors.toList());

            loginUser.setDeptId(collect.toArray(new Long[0]));

        }

        // 根据uuid将loginUser缓存

        String userKey = getTokenKey(loginUser.getToken());

        redisCache.setCacheObject(userKey, loginUser, expireTime, TimeUnit.MINUTES);

    }

    


    /**

     * 设置用户代理信息

     * 

@@ -155,13 +179,13 @@
    public void setUserAgent(LoginUser loginUser)

    {

        UserAgent userAgent = UserAgent.parseUserAgentString(ServletUtils.getRequest().getHeader("User-Agent"));

        String ip = IpUtils.getIpAddr(ServletUtils.getRequest());

        String ip = IpUtils.getIpAddr();

        loginUser.setIpaddr(ip);

        loginUser.setLoginLocation(AddressUtils.getRealAddressByIP(ip));

        loginUser.setBrowser(userAgent.getBrowser().getName());

        loginUser.setOs(userAgent.getOperatingSystem().getName());

    }

    


    /**

     * 从数据声明生成令牌

     *

@@ -220,6 +244,6 @@


    private String getTokenKey(String uuid)

    {

        return Constants.LOGIN_TOKEN_KEY + uuid;

        return CacheConstants.LOGIN_TOKEN_KEY + uuid;

    }

}