專案簡介

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()	令牌桶演演算法

設定建議

分散式系統，cacheService 建議使用基於 redis 的集中式快取策略。
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>