| | |
| | | import javax.servlet.http.HttpServletRequest;
|
| | | import org.springframework.beans.factory.annotation.Autowired;
|
| | | import org.springframework.beans.factory.annotation.Value;
|
| | | import org.springframework.http.MediaType;
|
| | | import org.springframework.stereotype.Component;
|
| | | import com.alibaba.fastjson.JSONObject;
|
| | | import com.ruoyi.common.constant.Constants;
|
| | | import com.alibaba.fastjson2.JSON;
|
| | | import com.ruoyi.common.constant.CacheConstants;
|
| | | import com.ruoyi.common.filter.RepeatedlyRequestWrapper;
|
| | | import com.ruoyi.common.utils.StringUtils;
|
| | | import com.ruoyi.common.utils.http.HttpHelper;
|
| | | import com.ruoyi.framework.interceptor.RepeatSubmitInterceptor;
|
| | | import com.ruoyi.framework.interceptor.annotation.RepeatSubmit;
|
| | | import com.ruoyi.framework.redis.RedisCache;
|
| | |
|
| | | /**
|
| | |
| | | @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)
|
| | | public boolean isRepeatSubmit(HttpServletRequest request, RepeatSubmit annotation)
|
| | | {
|
| | | String nowParams = "";
|
| | | if (request instanceof HttpServletRequest && StringUtils.equalsAnyIgnoreCase(request.getContentType(),
|
| | | MediaType.APPLICATION_JSON_VALUE, MediaType.APPLICATION_JSON_UTF8_VALUE))
|
| | | if (request instanceof RepeatedlyRequestWrapper)
|
| | | {
|
| | | RepeatedlyRequestWrapper repeatedlyRequest = (RepeatedlyRequestWrapper) request;
|
| | | nowParams = HttpHelper.getBodyString(repeatedlyRequest);
|
| | |
| | | // body参数为空,获取Parameter的数据
|
| | | if (StringUtils.isEmpty(nowParams))
|
| | | {
|
| | | nowParams = JSONObject.toJSONString(request.getParameterMap());
|
| | | nowParams = JSON.toJSONString(request.getParameterMap());
|
| | | }
|
| | | Map<String, Object> nowDataMap = new HashMap<String, Object>();
|
| | | nowDataMap.put(REPEAT_PARAMS, nowParams);
|
| | |
| | | // 请求地址(作为存放cache的key值)
|
| | | String url = request.getRequestURI();
|
| | |
|
| | | // 唯一标识(指定key + 消息头)
|
| | | String cache_repeat_key = Constants.REPEAT_SUBMIT_KEY + request.getHeader(header);
|
| | | // 唯一值(没有消息头则使用请求地址)
|
| | | String submitKey = StringUtils.trimToEmpty(request.getHeader(header));
|
| | |
|
| | | Object sessionObj = redisCache.getCacheObject(cache_repeat_key);
|
| | | // 唯一标识(指定key + url + 消息头)
|
| | | String cacheRepeatKey = CacheConstants.REPEAT_SUBMIT_KEY + url + 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))
|
| | | if (compareParams(nowDataMap, preDataMap) && compareTime(nowDataMap, preDataMap, annotation.interval()))
|
| | | {
|
| | | return true;
|
| | | }
|
| | |
| | | }
|
| | | Map<String, Object> cacheMap = new HashMap<String, Object>();
|
| | | cacheMap.put(url, nowDataMap);
|
| | | redisCache.setCacheObject(cache_repeat_key, cacheMap, intervalTime, TimeUnit.SECONDS);
|
| | | redisCache.setCacheObject(cacheRepeatKey, cacheMap, annotation.interval(), TimeUnit.MILLISECONDS);
|
| | | return false;
|
| | | }
|
| | |
|
| | |
| | | /**
|
| | | * 判断两次间隔时间
|
| | | */
|
| | | private boolean compareTime(Map<String, Object> nowMap, Map<String, Object> preMap)
|
| | | private boolean compareTime(Map<String, Object> nowMap, Map<String, Object> preMap, int interval)
|
| | | {
|
| | | long time1 = (Long) nowMap.get(REPEAT_TIME);
|
| | | long time2 = (Long) preMap.get(REPEAT_TIME);
|
| | | if ((time1 - time2) < (this.intervalTime * 1000))
|
| | | if ((time1 - time2) < interval)
|
| | | {
|
| | | return true;
|
| | | }
|