Spring Boot中实现接口请求限制

在Spring Boot中实现接口请求限制，可以通过AOP（面向切面编程）结合注解来实现。以下是一个简单的实现步骤：

1. 添加依赖

首先，确保你的pom.xml中包含了Spring Boot AOP的起步依赖和expiringmap依赖，后者用于创建具有过期策略的Map。

<!-- AOP -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>
<!-- ExpiringMap -->
<dependency>
    <groupId>net.jodah</groupId>
    <artifactId>expiringmap</artifactId>
    <version>0.5.10</version> <!-- 请使用最新的版本 -->
</dependency>

2. 创建注解

定义一个注解@LimitRequest，用于标记需要限制请求频率的方法。

@Documented
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface LimitRequest {
    long time() default 6000; // 限制时间，默认6000毫秒
    int count() default 3; // 允许的最大请求次数，默认3次
}

3. 创建AOP切面

创建一个切面LimitRequestAspect，用于拦截标记了@LimitRequest的方法，并实现请求频率限制。

@Aspect
@Component
public class LimitRequestAspect {
    private static ConcurrentHashMap<String, ExpiringMap<String, Integer>> book = new ConcurrentHashMap<>();

    @Pointcut("@annotation(limitRequest)")
    public void excudeService(LimitRequest limitRequest) {
    }

    @Around("excudeService(limitRequest)")
    public Object doAround(ProceedingJoinPoint pjp, LimitRequest limitRequest) throws Throwable {
        RequestAttributes ra = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes sra = (ServletRequestAttributes) ra;
        HttpServletRequest request = sra.getRequest();

        ExpiringMap<String, Integer> 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) {
            uc.put(request.getRemoteAddr(), uCount + 1, ExpirationPolicy.CREATED, limitRequest.time(), TimeUnit.MILLISECONDS);
        } else {
            uc.put(request.getRemoteAddr(), uCount + 1);
        }
        book.put(request.getRequestURI(), uc);
        return pjp.proceed();
    }
}

4. 使用注解

在你的控制器中使用@LimitRequest注解来标记需要限制请求频率的方法。

@RestController
public class DemoController {

    @LimitRequest(time = 60000, count = 5) // 每60秒内最多5次请求
    @GetMapping("/test")
    public String test() {
        return "Hello, World!";
    }
}

这个简单的示例展示了如何在Spring Boot应用中使用AOP和ExpiringMap来实现接口级别的请求频率限制。你可以根据实际需求调整注解的参数和切面的逻辑。