rate-limit 一款 java 開源漸進式分散式限流框架使用介紹

老馬嘯西風發表於2022-12-17

專案簡介

rate-limit 是一個為 java 設計的漸進式限流工具。

目的是為了深入學習和使用限流,後續將會持續迭代。

特性

  • 漸進式實現
  • 支援獨立於 spring 使用
  • 支援整合 spring
  • 支援整合 spring-boot
  • 內建多種限流策略

在這裡插入圖片描述

快速開始

需求

  • jdk 1.7
  • maven 3.x+

maven 匯入

<dependency>
    <groupId>com.github.houbb</groupId>
    <artifactId>rate-limit-core</artifactId>
    <version>1.1.0</version>
</dependency>

入門例子

方法定義

@RateLimit 限流注解放在方法上,指定對應的限制頻率。

也可以定義在類上,預設下面的所有方法生效。方法上的優先順序高於類。

屬性說明預設值
value方法訪問一次消耗的令牌數1
timeUnit時間單位TimeUnit.SECONDS
interval時間間隔60
count可呼叫次數1000
enable是否啟用true

預設為 60S 內,可以呼叫 1000 次。

public class UserService {

    @RateLimit(interval = 2, count = 5)
    public void limitCount() {
        log.info("{}", Thread.currentThread().getName());
    }

}

這個例子中我們 2S 內最多呼叫 5 次。

程式碼測試

RateLimitProxy.getProxy(xxx) 透過位元組碼獲取方法對應的方法代理。

@Test(expected = RateLimitRuntimeException.class)
public void limitCountErrorTest() {
    UserService userService = RateLimitProxy.getProxy(new UserService());
    for(int i = 0; i < 3; i++) {
        userService.limitCount();
    }
}

當呼叫超出限制時,預設丟擲 RateLimitRuntimeException 異常。

這裡預設使用的是令牌桶演算法,所以會出現異常。

重複註解 @RateLimits

有時候我們希望同時做多個的限制:

(1)一分鐘不超過 10 次

(2)一小時不超過 30 次

為了支援多個配置,我們引入了新的註解 @RateLimits,可以指定一個 @RateLimit 陣列。

方法上同時使用 @RateLimits + @RateLimit 是可以同時生效的,不過為了簡單,一般不建議混合使用。

@RateLimits({@RateLimit(interval = 2, count = 5)})
public void limitCount() {
    //...
}

指定引導類

RateLimitProxy.getProxy(new UserService());

等價於

RateLimitProxy.getProxy(new UserService(), RateLimitBs.newInstance());

下面我們來一起看一下 RateLimitBs 引導類。

引導類

RateLimitBs 作為引導類,便於使用者自定義配置。

方法說明預設值
rateLimit限流策略RateLimits.tokenBucket() 令牌桶演算法
timer時間策略Timers.system() 系統時間
cacheService快取策略CommonCacheServiceMap 基於本地 map 的快取策略
cacheKeyNamespace快取KEY名稱空間RATE-LIMIT 避免不同的應用,命名衝突。
configService限制配置策略RateLimitConfigService 預設基於方法上的註解
tokenService身份標識策略RateLimitTokenService 預設基於 IP
methodService方法標識策略RateLimitMethodService 預設基於方法名+引數型別
rejectListener拒絕策略RateLimitRejectListenerException 限流時丟擲異常

其中 rateLimit 內建 RateLimits 工具中的策略如下:

方法說明
fixedWindow()固定視窗
slideWindow(int windowNum)滑動視窗,可指定視窗大小
slideWindow()滑動視窗,預設為 10
slideWindowQueue()滑動視窗,基於佇列的實現
leakyBucket()漏桶演算法
tokenBucket()令牌桶演算法

配置建議

  1. 分散式系統,cacheService 建議使用基於 redis 的集中式快取策略。
  2. configService 如果想更加靈活,可以基於資料庫的配置查詢

RateLimitBs 引導類

RateLimitBs 預設配置如下:

RateLimitBs.newInstance()
      .timer(Timers.system())
      .methodService(new RateLimitMethodService())
      .tokenService(new RateLimitTokenService())
      .rejectListener(new RateLimitRejectListenerException())
      .configService(new RateLimitConfigService())
      .cacheService(new CommonCacheServiceMap())
      .rateLimit(RateLimits.tokenBucket())
      .cacheKeyNamespace(RateLimitConst.DEFAULT_CACHE_KEY_NAMESPACE);

spring 整合

maven 引入

<dependency>
    <groupId>com.github.houbb</groupId>
    <artifactId>rate-limit-spring</artifactId>
    <version>1.1.0</version>
</dependency>

類定義

方法

和上面使用類似,直接在方法上宣告 @RateLimit 註解即可。

@Service
public class UserService {

    private static final Log log = LogFactory.getLog(UserService.class);

    @RateLimit(interval = 2, count = 5)
    public void limitCount() {
        log.info("{}", Thread.currentThread().getName());
    }

}

配置

透過 @EnableRateLimit 宣告啟用限流。

@Configuration
@ComponentScan("com.github.houbb.rate.limit.test.core")
@EnableRateLimit
public class SpringConfig {

}

@EnableRateLimit 的屬性配置和 RateLimitBs 屬性是以一一對應的。

方法說明預設值
rateLimit限流策略令牌桶演算法
timer時間策略系統時間
cacheService快取策略基於本地 map 的快取策略
cacheKeyNamespace快取KEY名稱空間RATE-LIMIT 避免不同的應用,命名衝突。
configService限制配置策略預設基於方法上的註解
tokenService身份標識策略預設基於 IP
methodService方法標識策略預設基於方法名+引數型別
rejectListener拒絕策略限流時丟擲異常

這裡的屬性值,都是對應的 spring bean 名稱,支援使用者自定義。

spring-boot 整合

maven 引入

<dependency>
    <groupId>com.github.houbb</groupId>
    <artifactId>rate-limit-springboot-starter</artifactId>
    <version>1.1.0</version>
</dependency>

使用

其他和 spring 保持一致。

快取相關工具

cache: 手寫漸進式 redis

common-cache: 通用快取標準定義

redis-config: 相容各種常見的 redis 配置模式

lock: 開箱即用的分散式鎖

resubmit: 防重複提交

rate-limit: 限流

相關文章