自定义限流注解

Redis 除了做缓存，还能干很多很多事情：分布式锁、限流、处理请求接口幂等性。。。太多太多了～

代码：https://github.com/1040580896/rate_limiter

1. 准备工作

首先我们创建一个 Spring Boot 工程，引入 Web 和 Redis 依赖，同时考虑到接口限流一般是通过注解来标记，而注解是通过 AOP 来解析的，所以我们还需要加上 AOP 的依赖，最终的依赖如下：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

配置信息

然后提前准备好一个 Redis 实例，这里我们项目配置好之后，直接配置一下 Redis 的基本信息即可，如下：

spring.redis.port=6379
spring.redis.host=120.24.87.xxx
spring.redis.password=asd112211

2. 限流注解

接下来我们创建一个限流注解，我们将限流分为两种情况：

1. 针对当前接口的全局性限流，例如该接口可以在 1 分钟内访问 100 次。
2. 针对某一个 IP 地址的限流，例如某个 IP 地址可以在 1 分钟内访问 100 次。

针对这两种情况，我们创建一个枚举类：

/**
 * 限流类型
 */
public enum LimitType {
   

    /**
     * 默认的限流侧月，针对某一个接口进行限流
     */
    Default,
    /**
     * 针对某一个 IP 进行限流
     */
    IP
}

接下来我们来创建限流注解：

@Retention(RetentionPolicy.RUNTIME)
@Target({
   ElementType.METHOD})
public @interface RateLimiter {
   

    /**
     * 限流的key，主要是指前缀
     * @return
     */
    String key() default "rate_limit";

    /**
     * 限流时间窗
     * @return
     */
    int time() default 60;


    /**
     * 在时间窗限流次数
     * @return
     */
    int