Spring Boot内Redis实现的令牌桶算法限流

Spring Boot内Redis实现的令牌桶算法限流

最近研究了下接口的请求速率控制，有几种常见的算法，资料有很多，这里就不赘述了，这里使用的是Redis + Lua实现的令牌桶算法
参考了Spring Gateway的限流算法，直接搬运了它的lua脚本以及核心代码，简单封装了下，用注解实现在Controller层的速率限制

参考

Spring Cloud的限流实现，核心代码在
org.springframework.cloud.gateway.filter.ratelimit.RedisRateLimiter.isAllowed()

完整源码

不喜欢罗嗦的直接去看这里
Github Redis+Lua令牌桶限流

自定义注解

比较简单，组要是对接口限流能进行一些自定义配置

• 如按不同类型(ip/用户/接口)限制
• 对令牌桶进行自定义

/**
 * description: Limiter <br>
 * date: 2021/3/12 13:22 <br>
 * author: ws <br>
 * version: 1.0 <br>
 */
@Target({
   ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Documented
public @interface Limiter {
   
    LimiterType type() default LimiterType.PATH;

    // 每秒生产令牌数量
    int replenishRate() default 1;

    // 桶容量（并发数量）
    int burstCapacity() default 2;

    // 每次请求消耗令牌数量
    int requestedTokens() default 1;
}

Lua脚本

脚本直接使用@Bean来注入

...