优化防重复提交拦截器，唯一值（没有消息头则使用请求地址）

RuoYi

2020-08-11 28a9582a3e4a573881879947960bae1e848ce97a

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

@@ -5,8 +5,10 @@
import java.util.concurrent.TimeUnit;

import javax.servlet.http.HttpServletRequest;

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

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

import org.springframework.stereotype.Component;

import com.alibaba.fastjson.JSONObject;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.filter.RepeatedlyRequestWrapper;

import com.ruoyi.common.utils.StringUtils;

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

@@ -26,7 +28,9 @@


    public final String REPEAT_TIME = "repeatTime";



    public final String CACHE_REPEAT_KEY = "repeatData";

    // 令牌自定义标识

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

    private String header;



    @Autowired

    private RedisCache redisCache;

@@ -47,8 +51,12 @@
    @Override

    public boolean isRepeatSubmit(HttpServletRequest request)

    {

        RepeatedlyRequestWrapper repeatedlyRequest = (RepeatedlyRequestWrapper) request;

        String nowParams = HttpHelper.getBodyString(repeatedlyRequest);

        String nowParams = "";

        if (request instanceof RepeatedlyRequestWrapper)

        {

            RepeatedlyRequestWrapper repeatedlyRequest = (RepeatedlyRequestWrapper) request;

            nowParams = HttpHelper.getBodyString(repeatedlyRequest);

        }



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

        if (StringUtils.isEmpty(nowParams))

@@ -61,8 +69,18 @@


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

        String url = request.getRequestURI();

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

        String submitKey = request.getHeader(header);

        if (StringUtils.isEmpty(submitKey))

        {

            submitKey = url;

        }



        Object sessionObj = redisCache.getCacheObject(CACHE_REPEAT_KEY);

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

        String cache_repeat_key = Constants.REPEAT_SUBMIT_KEY + submitKey;



        Object sessionObj = redisCache.getCacheObject(cache_repeat_key);

        if (sessionObj != null)

        {

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

@@ -77,7 +95,7 @@
        }

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

        cacheMap.put(url, nowDataMap);

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

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

        return false;

    }