Java 注解限流实现：使用 AOP 拦截器控制请求速率

使用 Java 注解实现接口限流

本文将介绍如何使用 Java 注解和 AOP 拦截器实现接口限流，以控制请求速率，防止突发流量导致系统崩溃。

1. 定义注解 @RateLimiter

首先，我们需要定义一个注解 @RateLimiter，用于标识需要进行限流的方法。

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimiter {
    int value() default 10; // 默认值为每秒最多处理10个请求
}

该注解包含一个 value 属性，用于指定每秒最多处理多少个请求。如果不指定，默认值为 10。

2. 实现限流拦截器

接下来，我们需要实现一个拦截器 RateLimiterInterceptor，用于处理限流逻辑。该拦截器会在被标记为 @RateLimiter 的方法被调用时执行。

public class RateLimiterInterceptor implements MethodInterceptor {
    private final Map<Method, Semaphore> semaphores = new ConcurrentHashMap<>();

    @Override
    public Object invoke(MethodInvocation invocation) throws Throwable {
        Method method = invocation.getMethod();
        RateLimiter rateLimiter = method.getAnnotation(RateLimiter.class);
        if (rateLimiter == null) {
            return invocation.proceed();
        }

        Semaphore semaphore = semaphores.computeIfAbsent(method, key -> new Semaphore(rateLimiter.value()));
        boolean acquired = semaphore.tryAcquire();
        if (!acquired) {
            throw new RuntimeException('too many requests');
        }

        try {
            return invocation.proceed();
        } finally {
            semaphore.release();
        }
    }
}

该拦截器维护了一个 Map，用于存储每个被标记为 @RateLimiter 的方法所对应的信号量。在方法被调用时，拦截器会从 Map 中获取相应的信号量，并尝试获取一个许可。如果获取失败，则说明当前请求已经达到了限流阈值，拦截器会抛出一个 RuntimeException。如果获取成功，则说明当前请求可以被处理，拦截器会继续执行原来的方法。在方法执行完成后，拦截器会释放许可。

3. 应用拦截器

最后，我们需要将拦截器应用到需要进行限流的方法上。

@Service
public class UserService {
    @RateLimiter(value = 5) // 每秒最多处理5个请求
    public void createUser(String username, String password) {
        // 创建用户逻辑
    }
}

在上面的例子中，我们将 @RateLimiter 注解应用到了 createUser 方法上，并指定了每秒最多处理 5 个请求。当有多个请求同时调用该方法时，拦截器会根据限流阈值进行限制，保证系统的稳定性和可用性。

总结

通过使用 Java 注解和 AOP 拦截器，我们可以方便地实现接口限流，有效控制请求速率，提高系统稳定性和可用性。该方案易于扩展，可以根据实际需求进行调整和优化。