【接口限流】java中springboot实现接口限流防抖处理（JUC注解版）

文章目录

- 1、添加pom项目依赖
- 2、注解类RateLimit
- 3、限流切面RateLimitAspect
- 4、controller层使用注解
- 小结

1、添加pom项目依赖

		<!--AspectJ来实现切面，在方法执行前进行限流检查--><dependency><groupId>org.aspectj</groupId><artifactId>aspectjweaver</artifactId><version>1.9.7</version></dependency>

2、注解类RateLimit

java">import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimit {int limit() default 0;long timeUnit() default 0;boolean byIp() default false;boolean byRequestParam() default false;String paramName() default "";
}

3、限流切面RateLimitAspect

java">
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;/*** @author ZY.Jia* @date 2024/10/12 16:32* 限流切面类RateLimitAspect，使其支持根据 IP 和请求参数进行限流。**/
@Aspect
@Component
public class RateLimitAspect {@Autowiredprivate RateLimitConfig rateLimitConfig;private ConcurrentHashMap<String, AtomicInteger> requestCounts = new ConcurrentHashMap<>();private ConcurrentHashMap<String, AtomicLong> lastAccessTimes = new ConcurrentHashMap<>();@Around("@annotation(com.cmst.financialmiddleplatform.platformmaincenter.annotation.RateLimit)")public Object rateLimit(ProceedingJoinPoint joinPoint) throws Throwable {HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();String requestIp = String.valueOf(request.getRemoteAddr()).trim();String requestURI = request.getRequestURI();// 获取被调用方法的方法名String methodKey = request.getMethod();Class<?>[] parameterTypes = new Class<?>[joinPoint.getArgs().length];for (int i = 0; i < joinPoint.getArgs().length; i++) {parameterTypes[i] = joinPoint.getArgs()[i].getClass();}Method method = joinPoint.getTarget().getClass().getMethod(joinPoint.getSignature().getName(), parameterTypes);RateLimit rateLimitAnnotation = method.getAnnotation(RateLimit.class);//注解方式获取参数int limit = rateLimitAnnotation.limit();long timeUnit = rateLimitAnnotation.timeUnit();boolean byIp = rateLimitAnnotation.byIp();boolean byRequestParam = rateLimitAnnotation.byRequestParam();String paramName = rateLimitAnnotation.paramName();//配置文件获取参数/*int limit = rateLimitConfig.getLimit();long timeUnit = rateLimitConfig.getTimeUnit();boolean byIp = rateLimitConfig.isByIp();boolean byRequestParam = rateLimitConfig.isByRequestParam();String paramName = rateLimitConfig.getParamName();*/Object requestParam = new Object();Object[] args = joinPoint.getArgs();if (args != null || args.length > 0) {for (Object arg : args) {requestParam = arg;}}Map<String, String> stringStringHashMap = new HashMap<String, String>();stringStringHashMap = JSONObject.parseObject(requestParam.toString(), Map.class);String key = methodKey + "_" + requestURI;if (byIp) {key += "_" + requestIp;}if (byRequestParam && stringStringHashMap.containsKey(paramName)) {key += "_" + paramName + "_" + stringStringHashMap.get(paramName);}// 如果该方法的计数不存在于 map 中，则初始化一个新的 AtomicInteger 并放入 map，否则返回已有的计数AtomicInteger count = requestCounts.computeIfAbsent(key, k -> new AtomicInteger(0));AtomicLong lastAccessTime = lastAccessTimes.computeIfAbsent(key, k -> new AtomicLong(System.currentTimeMillis()));long currentTime = System.currentTimeMillis();if (currentTime - lastAccessTime.get() > timeUnit) {// 时间超过指定时间单位，重置计数和时间count.set(0);lastAccessTime.set(currentTime);}if (count.get() < limit) {count.incrementAndGet();return joinPoint.proceed();} else {throw new RuntimeException("超过限制次数，请联系服务商！。");}}
}

4、controller层使用注解

注解解释：不对ip限流，只对参数中参数名为systemCode的参数值进行监控限流，一周的访问上限为350次

java">	@PostMapping("/123")@Syslog(description = "123")@RateLimit(limit = 350, timeUnit = 7*24*60*60*1000, byIp = false, byRequestParam = true, paramName = "systemCode")public MessageResult method123(@RequestBody String json) throws Exception {//。。。。。return new MessageResult("测试限流注解");}