自定义限流注解

最新推荐文章于 2024-07-23 07:45:00 发布

小开心

最新推荐文章于 2024-07-23 07:45:00 发布

阅读量699

点赞数

分类专栏：若依框架文章标签： redis java 缓存

本文链接：https://blog.csdn.net/qq_47142993/article/details/125033365

版权

本文介绍了如何使用Redis实现接口限流，包括创建限流注解、定制RedisTemplate、编写Lua脚本以及注解解析和接口测试。通过自定义AOP切面处理限流注解，结合Lua脚本实现原子操作，确保限流的准确性和效率。

摘要由CSDN通过智能技术生成

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 分钟内访问 100 次。
针对某一个 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