SpringBoot在一定时间内限制接口请求次数的实现示例

SpringBoot在一定时间内限制接口请求次数的实现示例

需要用到的知识：注解、AOP、ExpiringMap（带有有效期的映射）

我们可以自定义注解，把注解添加到我们的接口上。定义一个切面，执行方法前去ExpiringMap查询该IP在规定时间内请求了多少次，如超过次数则直接返回请求失败。

需要用到的依赖

org.springframework.boot

spring-boot-starter-aop

2.1.5.RELEASE

net.jodah

expiringmap

0.5.8

自定义注解@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> 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 = RequestContextHolder.getRequestAttributes();

ServletRequestAttributes sra = (ServletRequestAttributes) ra;

HttpServletRequest request = sra.getRequest();

// 获取Map对象， 如果没有则返回默认值

// 第一个参数是key， 第二个参数是默认值

ExpiringMap uc = book.getOrDefault(request.getRequestURI(), ExpiringMap.builder().variableExpiration().build());

Integer uCount = uc.getOrDefault(request.getRemoteAddr(), 0);

if (uCount >= limitRequest.count()) { // 超过次数，不执行目标方法

return "接口请求超过次数";

} else if (uCount == 0){ // 第一次请求时，设置有效时间

// /** Expires entries based on when they were last accessed */

// ACCESSED,

// /** Expires entries based on when they were created */

// CREATED;

uc.put(request.getRemoteAddr(), uCount + 1, ExpirationPolicy.CREATED, limitRequest.time(), TimeUnit.MILLISECONDS);

} else { // 未超过次数， 记录加一

uc.put(request.getRemoteAddr(), uCount + 1);

}

book.pugrDGKt(request.getRequestURI(), uc);

// result的值就是被拦截方法的返回值

Object result = pjp.proceed();

return result;

}

}

第一个静态Map是多线程安全的Map（ConcurrentHashMap），它的key是接口对于的url，它的value是一个多线程安全且键值对是有有效期的Map（ExpiringMap）。

ExpiringMap的key是请求的ip地址，value是已经请求的次数。

ExpiringMap更多的使用方法可以参考：https://github.com/jhalterman/expiringmap

最后在方法上面加上@LimitRequest就行了

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。