SpringBoot在⼀定时间内限制接⼝请求次数的实现⽰例
需要⽤到的知识：注解、AOP、ExpiringMap（带有有效期的映射）
我们可以⾃定义注解，把注解添加到我们的接⼝上。定义⼀个切⾯，执⾏⽅法前去ExpiringMap查询该IP在规定时间内请求了多少次，如超过次数则直接返回请求失败。需要⽤到的依赖
<!-- AOP依赖 -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
<version>2.1.5.RELEASE</version>
</dependency>
<!-- Map依赖 -->
<dependency>
<groupId>net.jodah</groupId>
<artifactId>expiringmap</artifactId>
<version>0.5.8</version>
</dependency>
⾃定义注解@LimitRequest
@Documented
@Target(ElementType.METHOD) // 说明该注解只能放在⽅法上⾯
@Retention(RetentionPolicy.RUNTIME)
public @interface LimitRequest {
long time() default 6000; // 限制时间单位：毫秒
int count() default 1; // 允许请求的次数
}
⾃定义AOP
@Aspect
@Component
public class LimitRequestAspect {
private static ConcurrentHashMap<String, ExpiringMap<String, Integer>> book = new ConcurrentHashMap<>();
// 定义切点
// 让所有有@LimitRequest注解的⽅法都执⾏切⾯⽅法
@Pointcut("@annotation(limitRequest)")
public void excudeService(LimitRequest limitRequest) {
}
@Around("excudeService(limitRequest)")
public Object doAround(ProceedingJoinPoint pjp, LimitRequest limitRequest) throws Throwable {
// 获得request对象
RequestAttributes ra = RequestAttributes();
ServletRequestAttributes sra = (ServletRequestAttributes) ra;
HttpServletRequest request = Request();
// 获取Map对象，如果没有则返回默认值
// 第⼀个参数是key，第⼆个参数是默认值
ExpiringMap<String, Integer> uc = RequestURI(), ExpiringMap.builder().variableExpiration().build());
Integer uCount = uc.RemoteAddr(), 0);
if (uCount >= unt()) { // 超过次数，不执⾏⽬标⽅法
return "接⼝请求超过次数";
springboot aop
} else if (uCount == 0){ // 第⼀次请求时，设置有效时间
//            /** Expires entries based on when they were last accessed */
//            ACCESSED,
//            /** Expires entries based on when they were created */
//            CREATED;
uc.RemoteAddr(), uCount + 1, ExpirationPolicy.CREATED, limitRequest.time(), TimeUnit.MILLISECONDS);
} else { // 未超过次数，记录加⼀
uc.RemoteAddr(), uCount + 1);
}
book.RequestURI(), uc);
// result的值就是被拦截⽅法的返回值
Object result = pjp.proceed();
return result;
}
}
第⼀个静态Map是多线程安全的Map（ConcurrentHashMap），它的key是接⼝对于的url，它的value是⼀个多线程安全且键值对是有有效期的Map（ExpiringMap）。ExpiringMap的key是请求的ip地址，value是已经请求的次数。
最后在⽅法上⾯加上@LimitRequest就⾏了
到此这篇关于SpringBoot在⼀定时间内限制接⼝请求次数的实现⽰例的⽂章就介绍到这了，更多相关SpringBoot 限制接⼝请求次数内容请搜索以前的⽂章或继续浏览下⾯的相关⽂章希望⼤家以后多多⽀持！

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。