springboot redis实现接口限流

Spring Boot结合Redis实现接口限的步骤如下[^1][^2]：

1. 引入依赖：在Spring Boot项目的pom.xml文件中添加Redis和AOP的相关依赖。

2. 配置RedisTemplate：在Spring Boot的配置文件中配置Redis连接信息，包括主机名、端口号、密码等。

3. 创建自定义注解：使用@RateLimiter注解来标记需要进行接口限流的方法。

4. 编写切面类：创建一个切面类，使用@Aspect注解标记，并在该类中编写切点和通知方法。

5. 实现接口限流逻辑：在通知方法中，使用Redis的原子操作来实现接口限流的逻辑。可以使用Redis的incr命令来对接口的访问次数进行计数，然后根据设定的阈值来判断是否限流。

6. 配置切面：在Spring Boot的配置类中，使用@EnableAspectJAutoProxy注解开启AOP功能，并将切面类添加到容器中。

7. 在需要进行接口限流的方法上添加注解：在需要进行接口限流的方法上添加@RateLimiter注解，并配置相关参数，如限流的阈值、时间窗口大小等。

8. 测试接口限流效果：启动Spring Boot应用程序，并访问被限流的接口，观察接口的访问频率是否受到限制。

以下是一个示例代码，演示了如何使用Spring Boot和Redis实现接口限流：

// 1. 创建自定义注解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RateLimiter {
    int value() default 10; // 默认限流阈值为10
    int window() default 60; // 默认时间窗口为60秒
}

// 2. 编写切面类
@Aspect
@Component
public class RateLimiterAspect {
    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Around("@annotation(rateLimiter)")
    public Object around(ProceedingJoinPoint joinPoint, RateLimiter rateLimiter) throws Throwable {
        String methodName = joinPoint.getSignature().getName();
        String key = "rate_limiter:" + methodName;
        int limit = rateLimiter.value();
        int window = rateLimiter.window();

        // 使用Redis的incr命令对接口的访问次数进行计数
        Long count = redisTemplate.opsForValue().increment(key, 1);
        if (count == 1) {
            // 设置过期时间，保证计数器在一定时间后自动清零
            redisTemplate.expire(key, window, TimeUnit.SECONDS);
        }

        if (count > limit) {
            // 超过限流阈值，抛出异常或返回错误信息
            throw new RuntimeException("接口访问频率超过限制");
        }

        // 执行原方法
        return joinPoint.proceed();
    }
}

// 3. 在需要进行接口限流的方法上添加注解
@RestController
public class DemoController {
    @RateLimiter(value = 5, window = 60) // 每分钟最多访问5次
    @GetMapping("/demo")
    public String demo() {
        return "Hello World!";
    }
}