优化代码

RuoYi

2025-02-28 db28017ec154131d7baf08283dd9b40fdae91373

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

@@ -4,15 +4,18 @@
import java.util.Map;

import java.util.concurrent.TimeUnit;

import javax.servlet.http.HttpServletRequest;

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.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;

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

public class TokenService

{

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



    // 令牌自定义标识

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

    private String header;

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

            catch (Exception e)

            {

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

            }

        }

        return null;

@@ -114,6 +120,7 @@


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

        claims.put(Constants.LOGIN_USER_KEY, token);

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

        return createToken(claims);

    }



@@ -155,7 +162,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());

@@ -220,6 +227,6 @@


    private String getTokenKey(String uuid)

    {

        return Constants.LOGIN_TOKEN_KEY + uuid;

        return CacheConstants.LOGIN_TOKEN_KEY + uuid;

    }

}