支持登录IP黑名单限制

RuoYi

2023-02-21 3ef9c108c9e90a01aee36b708983dcb42175c429

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

@@ -7,12 +7,13 @@
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.IdUtils;

import com.ruoyi.common.utils.ServletUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.ip.AddressUtils;

import com.ruoyi.common.utils.ip.IpUtils;

import com.ruoyi.common.utils.uuid.IdUtils;

import com.ruoyi.framework.redis.RedisCache;

import com.ruoyi.framework.security.LoginUser;

import eu.bitwalker.useragentutils.UserAgent;

@@ -60,12 +61,18 @@
        String token = getToken(request);

        if (StringUtils.isNotEmpty(token))

        {

            Claims claims = parseToken(token);

            // 解析对应的权限以及用户信息

            String uuid = (String) claims.get(Constants.LOGIN_USER_KEY);

            String userKey = getTokenKey(uuid);

            LoginUser user = redisCache.getCacheObject(userKey);

            return user;

            try

            {

                Claims claims = parseToken(token);

                // 解析对应的权限以及用户信息

                String uuid = (String) claims.get(Constants.LOGIN_USER_KEY);

                String userKey = getTokenKey(uuid);

                LoginUser user = redisCache.getCacheObject(userKey);

                return user;

            }

            catch (Exception e)

            {

            }

        }

        return null;

    }

@@ -149,7 +156,7 @@
    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());

@@ -214,6 +221,6 @@


    private String getTokenKey(String uuid)

    {

        return Constants.LOGIN_TOKEN_KEY + uuid;

        return CacheConstants.LOGIN_TOKEN_KEY + uuid;

    }

}