update [重大改动] 将 framework 与 system 模块 解耦

疯狂的狮子li

2021-09-17 50599487f894ed3bf0297a2906f172e91f3ffba2

 ruoyi-framework/src/main/java/com/ruoyi/framework/interceptor/impl/SameUrlDataInterceptor.java

@@ -1,126 +1,114 @@
package com.ruoyi.framework.interceptor.impl;



import cn.hutool.core.lang.Validator;

import com.alibaba.fastjson.JSONObject;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.core.redis.RedisCache;

import com.ruoyi.common.filter.RepeatedlyRequestWrapper;

import com.ruoyi.common.utils.http.HttpHelper;

import com.ruoyi.framework.interceptor.RepeatSubmitInterceptor;

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

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

import org.springframework.stereotype.Component;



import javax.servlet.http.HttpServletRequest;

import java.util.HashMap;

import java.util.Map;

import java.util.concurrent.TimeUnit;



/**

 * 判断请求url和数据是否和上一次相同，

 * 如果和上次相同，则是重复提交表单。 有效时间为10秒内。

 * 
 * @author ruoyi

 */

@Component

public class SameUrlDataInterceptor extends RepeatSubmitInterceptor

{

    public final String REPEAT_PARAMS = "repeatParams";



    public final String REPEAT_TIME = "repeatTime";



    // 令牌自定义标识

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

    private String header;



    @Autowired

    private RedisCache redisCache;



    /**

     * 间隔时间，单位:秒 默认10秒

     * 
     * 两次相同参数的请求，如果间隔时间大于该参数，系统不会认定为重复提交的数据

     */

    private int intervalTime = 10;



    public void setIntervalTime(int intervalTime)

    {

        this.intervalTime = intervalTime;

    }



    @SuppressWarnings("unchecked")

    @Override

    public boolean isRepeatSubmit(HttpServletRequest request)

    {

        String nowParams = "";

        if (request instanceof RepeatedlyRequestWrapper)

        {

            RepeatedlyRequestWrapper repeatedlyRequest = (RepeatedlyRequestWrapper) request;

            nowParams = HttpHelper.getBodyString(repeatedlyRequest);

        }



        // body参数为空，获取Parameter的数据

        if (Validator.isEmpty(nowParams))

        {

            nowParams = JSONObject.toJSONString(request.getParameterMap());

        }

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

        nowDataMap.put(REPEAT_PARAMS, nowParams);

        nowDataMap.put(REPEAT_TIME, System.currentTimeMillis());



        // 请求地址（作为存放cache的key值）

        String url = request.getRequestURI();



        // 唯一值（没有消息头则使用请求地址）

        String submitKey = request.getHeader(header);

        if (Validator.isEmpty(submitKey))

        {

            submitKey = url;

        }



        // 唯一标识（指定key + 消息头）

        String cacheRepeatKey = Constants.REPEAT_SUBMIT_KEY + submitKey;



        Object sessionObj = redisCache.getCacheObject(cacheRepeatKey);

        if (sessionObj != null)

        {

            Map<String, Object> sessionMap = (Map<String, Object>) sessionObj;

            if (sessionMap.containsKey(url))

            {

                Map<String, Object> preDataMap = (Map<String, Object>) sessionMap.get(url);

                if (compareParams(nowDataMap, preDataMap) && compareTime(nowDataMap, preDataMap))

                {

                    return true;

                }

            }

        }

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

        cacheMap.put(url, nowDataMap);

        redisCache.setCacheObject(cacheRepeatKey, cacheMap, intervalTime, TimeUnit.SECONDS);

        return false;

    }



    /**

     * 判断参数是否相同

     */

    private boolean compareParams(Map<String, Object> nowMap, Map<String, Object> preMap)

    {

        String nowParams = (String) nowMap.get(REPEAT_PARAMS);

        String preParams = (String) preMap.get(REPEAT_PARAMS);

        return nowParams.equals(preParams);

    }



    /**

     * 判断两次间隔时间

     */

    private boolean compareTime(Map<String, Object> nowMap, Map<String, Object> preMap)

    {

        long time1 = (Long) nowMap.get(REPEAT_TIME);

        long time2 = (Long) preMap.get(REPEAT_TIME);

        if ((time1 - time2) < (this.intervalTime * 1000))

        {

            return true;

        }

        return false;

    }

}

package com.ruoyi.framework.interceptor.impl;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.io.IoUtil;
import com.ruoyi.common.annotation.RepeatSubmit;
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.filter.RepeatedlyRequestWrapper;
import com.ruoyi.common.utils.JsonUtils;
import com.ruoyi.common.utils.RedisUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.framework.config.properties.RepeatSubmitProperties;
import com.ruoyi.common.properties.TokenProperties;
import com.ruoyi.framework.interceptor.RepeatSubmitInterceptor;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

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

/**
 * 判断请求url和数据是否和上一次相同，
 * 如果和上次相同，则是重复提交表单。
 *
 * @author Lion Li
 */
@Slf4j
@RequiredArgsConstructor(onConstructor_ = @Autowired)
@Component
public class SameUrlDataInterceptor extends RepeatSubmitInterceptor {
   public final String REPEAT_PARAMS = "repeatParams";

   public final String REPEAT_TIME = "repeatTime";

   private final TokenProperties tokenProperties;
   private final RepeatSubmitProperties repeatSubmitProperties;


   @SuppressWarnings("unchecked")
   @Override
   public boolean isRepeatSubmit(RepeatSubmit repeatSubmit, HttpServletRequest request) {
      // 如果注解不为0 则使用注解数值
      long intervalTime = repeatSubmitProperties.getIntervalTime();
      if (repeatSubmit.intervalTime() > 0) {
         intervalTime = repeatSubmit.timeUnit().toMillis(repeatSubmit.intervalTime());
      }
      String nowParams = "";
      if (request instanceof RepeatedlyRequestWrapper) {
         RepeatedlyRequestWrapper repeatedlyRequest = (RepeatedlyRequestWrapper) request;
         try {
            nowParams = IoUtil.readUtf8(repeatedlyRequest.getInputStream());
         } catch (IOException e) {
            log.warn("读取流出现问题！");
         }
      }

      // body参数为空，获取Parameter的数据
      if (StringUtils.isEmpty(nowParams)) {
         nowParams = JsonUtils.toJsonString(request.getParameterMap());
      }
      Map<String, Object> nowDataMap = new HashMap<String, Object>();
      nowDataMap.put(REPEAT_PARAMS, nowParams);
      nowDataMap.put(REPEAT_TIME, System.currentTimeMillis());

      // 请求地址（作为存放cache的key值）
      String url = request.getRequestURI();

      // 唯一值（没有消息头则使用请求地址）
      String submitKey = request.getHeader(tokenProperties.getHeader());
      if (StringUtils.isEmpty(submitKey)) {
         submitKey = url;
      }

      // 唯一标识（指定key + 消息头）
      String cacheRepeatKey = Constants.REPEAT_SUBMIT_KEY + submitKey;

      Object sessionObj = RedisUtils.getCacheObject(cacheRepeatKey);
      if (sessionObj != null) {
         Map<String, Object> sessionMap = (Map<String, Object>) sessionObj;
         if (sessionMap.containsKey(url)) {
            Map<String, Object> preDataMap = (Map<String, Object>) sessionMap.get(url);
            if (compareParams(nowDataMap, preDataMap) && compareTime(nowDataMap, preDataMap, intervalTime)) {
               return true;
            }
         }
      }
      Map<String, Object> cacheMap = new HashMap<String, Object>();
      cacheMap.put(url, nowDataMap);
      RedisUtils.setCacheObject(cacheRepeatKey, cacheMap, Convert.toInt(intervalTime), TimeUnit.MILLISECONDS);
      return false;
   }

   /**
    * 判断参数是否相同
    */
   private boolean compareParams(Map<String, Object> nowMap, Map<String, Object> preMap) {
      String nowParams = (String) nowMap.get(REPEAT_PARAMS);
      String preParams = (String) preMap.get(REPEAT_PARAMS);
      return nowParams.equals(preParams);
   }

   /**
    * 判断两次间隔时间
    */
   private boolean compareTime(Map<String, Object> nowMap, Map<String, Object> preMap, long intervalTime) {
      long time1 = (Long) nowMap.get(REPEAT_TIME);
      long time2 = (Long) preMap.get(REPEAT_TIME);
      return (time1 - time2) < intervalTime;
   }
}