Merge branch 'master' of http://120.76.84.145:10101/gitblit/r/java/mx_charg...

无关风月

2025-02-13 8236caa8042288fcb9b584c21d6157f713183626

 ruoyi-service/ruoyi-integration/src/main/java/com/ruoyi/integration/drainage/TokenUtil.java

New file
@@ -0,0 +1,159 @@
package com.ruoyi.integration.drainage;

import com.ruoyi.common.core.constant.CacheConstants;
import com.ruoyi.common.core.constant.SecurityConstants;
import com.ruoyi.common.core.exception.auth.NotLoginException;
import com.ruoyi.common.core.utils.JwtUtils;
import com.ruoyi.common.core.utils.ServletUtils;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.utils.ip.IpUtils;
import com.ruoyi.common.core.utils.uuid.IdUtils;
import com.ruoyi.common.redis.service.RedisService;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.other.api.domain.Operator;
import com.ruoyi.system.api.model.LoginUser;
import com.ruoyi.system.api.model.LoginUserApplet;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * token验证处理
 *
 * @author ruoyi
 */
@Component
public class TokenUtil {
   @Autowired
   private RedisService redisService;
	
   protected static final long MILLIS_SECOND = 1000;
	
   protected static final long MILLIS_MINUTE = 60 * MILLIS_SECOND;
	
   private final static long expireTime = CacheConstants.EXPIRATION;
   private final static long expireAppletTime = CacheConstants.EXPIRATION_APPLET;
	
   private final static String ACCESS_TOKEN = CacheConstants.LOGIN_TOKEN_KEY;
	
   private final static Long MILLIS_MINUTE_TEN = CacheConstants.REFRESH_TIME * MILLIS_MINUTE;
	
   /**
    * 创建令牌
    */
   public Map<String, Object> createToken(Operator operator) {
      String token = IdUtils.fastUUID();
      operator.setToken(token);
      refreshToken(operator);
		
      // Jwt存储信息
      Map<String, Object> claimsMap = new HashMap<String, Object>();
      claimsMap.put(SecurityConstants.USER_KEY, token);
      claimsMap.put(SecurityConstants.DETAILS_USER_ID, operator.getOperatorId());
      claimsMap.put(SecurityConstants.USER_TYPE, "system");
      claimsMap.put(SecurityConstants.DETAILS_USERNAME, operator.getName());
		
      // 接口返回信息
      Map<String, Object> rspMap = new HashMap<String, Object>();
      rspMap.put("access_token", JwtUtils.createToken(claimsMap));
      rspMap.put("expires_in", expireTime);
      return rspMap;
   }
	
   /**
    * 获取用户身份信息
    *
    * @return 用户信息
    */
   public Operator getLoginUser() {
      Operator loginUser = getLoginUser(ServletUtils.getRequest());
      if (loginUser == null) {
         throw new NotLoginException("令牌已过期，请重新登录！");
      }
      return loginUser;
   }
	
   /**
    * 获取用户身份信息
    *
    * @return 用户信息
    */
   public Operator getLoginUser(HttpServletRequest request) {
      // 获取请求携带的令牌
      String token = SecurityUtils.getToken(request);
      return getLoginUser(token);
   }
	
   /**
    * 获取用户身份信息
    *
    * @return 用户信息
    */
   public Operator getLoginUser(String token) {
      Operator user = null;
      try {
         if (StringUtils.isNotEmpty(token)) {
            String userkey = JwtUtils.getUserKey(token);
            user = redisService.getCacheObject(getTokenKey(userkey));
            return user;
         }
      } catch (Exception e) {
         e.printStackTrace();
      }
      return user;
   }
	
   /**
    * 设置用户身份信息
    */
   public void setLoginUser(Operator operator) {
      if (StringUtils.isNotNull(operator) && StringUtils.isNotEmpty(operator.getToken())) {
         refreshToken(operator);
      }
   }
	
   /**
    * 删除用户缓存信息
    */
   public void delLoginUser(String token) {
      if (StringUtils.isNotEmpty(token)) {
         String userkey = JwtUtils.getUserKey(token);
         redisService.deleteObject(getTokenKey(userkey));
      }
   }
	
   /**
    * 验证令牌有效期，相差不足120分钟，自动刷新缓存
    *
    * @param operator
    */
   public void verifyToken(Operator operator) {
      long expireTime = operator.getExpireTime();
      long currentTime = System.currentTimeMillis();
      if (expireTime - currentTime <= MILLIS_MINUTE_TEN) {
         refreshToken(operator);
      }
   }
	
   /**
    * 刷新令牌有效期
    *
    * @param operator 登录信息
    */
   public void refreshToken(Operator operator) {
      operator.setLoginTime(System.currentTimeMillis());
      operator.setExpireTime(operator.getLoginTime() + expireTime * MILLIS_MINUTE);
      // 根据uuid将loginUser缓存
      String userKey = getTokenKey(operator.getToken());
      redisService.setCacheObject(userKey, operator, expireTime, TimeUnit.MINUTES);
   }

	
   private String getTokenKey(String token) {
      return ACCESS_TOKEN + token;
   }
}