Spring控制重复请求

通过AOP拦截所有请求，控制在规定时间内请求次数。

1：添加maven

<dependency><groupId>net.jodah</groupId><artifactId>expiringmap</artifactId><version>0.5.10</version>
</dependency>

2：全局注解

import java.lang.annotation.*;
import java.util.concurrent.TimeUnit;/*** @Description 是否重复提交注解(在方法上,例: @IsRepeatSubmit 或者 @IsRepeatSubmit(count = 2,time = 2000)，参数不写使用默认值)* @Author WangKun* @Date 2024/5/22 16:52* @Version*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface IsRepeatSubmit {/*** 限制时间单位 单位：毫秒**/TimeUnit TIME_UNIT = TimeUnit.MILLISECONDS;/*** 限制时间 单位：毫秒**/long time() default 1000;/*** 限制时间内允许请求的次数**/int count() default 1;}

3：AOP切面

import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.harmonywisdom.requestaop.IsRepeatSubmit;
import com.harmonywisdom.response.ResponseResult;
import com.harmonywisdom.utils.HttpContextUtil;
import com.harmonywisdom.utils.RequestUtil;
import lombok.extern.slf4j.Slf4j;
import net.jodah.expiringmap.ExpirationPolicy;
import net.jodah.expiringmap.ExpiringMap;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.concurrent.ConcurrentHashMap;/*** @Description 重复提交切面* @Author WangKun* @Date 2024/5/22 16:57* @Version*/
@Slf4j
@Aspect
@Component
public class RepeatSubmitAspect {/*** 全局缓存**/private static final ConcurrentHashMap<String, ExpiringMap<String, Integer>> CE = new ConcurrentHashMap<>();/*** @Description 切点* @param irs* @Throws* @Return void* @Date 2024-05-28 15:41:56* @Author WangKun**/@Pointcut("@annotation(irs)")public void excludeService(IsRepeatSubmit irs) {}/*** @Description 切面* @param pjp* @param irs* @Throws* @Return java.lang.Object* @Date 2024-05-28 15:41:42* @Author WangKun**/@Around("excludeService(irs)")public Object doAround(ProceedingJoinPoint pjp, IsRepeatSubmit irs) throws Throwable {// 从上下文获取requestHttpServletRequest request = HttpContextUtil.getHttpServletRequest();MethodSignature signature = (MethodSignature) pjp.getSignature();Method method = signature.getMethod();Object[] args = pjp.getArgs();//获取request中数据（获取@RequestBody ,@RequestParam注解的参数）String data = RequestUtil.getParameter(request, method, args);JSONObject obj = JSON.parseObject(data);/** 以下根据自己的业务调整**///ipString ip = request.getRemoteAddr();//uriString uri = request.getRequestURI();//getOrDefault方法，没有该key则使用该keyExpiringMap<String, Integer> em = CE.getOrDefault(uri+"_"+ip, ExpiringMap.builder().variableExpiration().build());Integer count = em.getOrDefault(ip, 0);if (count >= irs.count()) {return ResponseResult.error(500,"请勿重复提交");} else if (count == 0) {em.put(ip, count + 1, ExpirationPolicy.CREATED, irs.time(), irs.TIME_UNIT);} else {em.put(ip, count + 1);}CE.put(uri+"_"+ip, em);return pjp.proceed();}}

RequestUtil.getParameter(request, method, args)

获取@RequestBody ,@RequestParam注解的参数-CSDN博客

HttpServletRequest request = HttpContextUtil.getHttpServletRequest();

Spring 获取http上下文-CSDN博客