SpringBoot + Redis 實現(xiàn)接口限流，一個注解的事！

Hollis的新書限時折扣中，一本深入講解Java基礎(chǔ)的干貨筆記！

Redis 除了做緩存，還能干很多很多事情：分布式鎖、限流、處理請求接口冪等性。。。太多太多了～

今天想和小伙伴們聊聊用 Redis 處理接口限流，這也是最近的 TienChin 項目涉及到這個知識點了，我就拎出來和大家聊聊這個話題，后面視頻也會講。

1. 準備工作

首先我們創(chuàng)建一個 Spring Boot 工程，引入 Web 和 Redis 依賴，同時考慮到接口限流一般是通過注解來標記，而注解是通過 AOP 來解析的，所以我們還需要加上 AOP 的依賴，最終的依賴如下：

<dependency>
????<groupId>org.springframework.bootgroupId>
????<artifactId>spring-boot-starter-data-redisartifactId>
dependency>
<dependency>
????<groupId>org.springframework.bootgroupId>
????<artifactId>spring-boot-starter-webartifactId>
dependency>
<dependency>
????<groupId>org.springframework.bootgroupId>
????<artifactId>spring-boot-starter-aopartifactId>
dependency>

然后提前準備好一個 Redis 實例，這里我們項目配置好之后，直接配置一下 Redis 的基本信息即可，如下：

spring.redis.host=localhost
spring.redis.port=6379
spring.redis.password=123

好啦，準備工作就算是到位了。

2. 限流注解

接下來我們創(chuàng)建一個限流注解，我們將限流分為兩種情況：

1. 針對當前接口的全局性限流，例如該接口可以在 1 分鐘內(nèi)訪問 100 次。
2. 針對某一個 IP 地址的限流，例如某個 IP 地址可以在 1 分鐘內(nèi)訪問 100 次。

針對這兩種情況，我們創(chuàng)建一個枚舉類：

public?enum?LimitType?{
????/**
?????*?默認策略全局限流
?????*/
????DEFAULT,
????/**
?????*?根據(jù)請求者IP進行限流
?????*/
????IP
}

接下來我們來創(chuàng)建限流注解：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public?@interface?RateLimiter?{
????/**
?????*?限流key
?????*/
????String?key()?default?"rate_limit:";

????/**
?????*?限流時間,單位秒
?????*/
????int?time()?default?60;

????/**
?????*?限流次數(shù)
?????*/
????int?count()?default?100;

????/**
?????*?限流類型
?????*/
????LimitType?limitType()?default?LimitType.DEFAULT;
}

第一個參數(shù)限流的 key，這個僅僅是一個前綴，將來完整的 key 是這個前綴再加上接口方法的完整路徑，共同組成限流 key，這個 key 將被存入到 Redis 中。

另外三個參數(shù)好理解，我就不多說了。

好了，將來哪個接口需要限流，就在哪個接口上添加?@RateLimiter?注解，然后配置相關(guān)參數(shù)即可。

3. 定制 RedisTemplate

小伙伴們知道，在 Spring Boot 中，我們其實更習(xí)慣使用 Spring Data Redis 來操作 Redis，不過默認的 RedisTemplate 有一個小坑，就是序列化用的是 JdkSerializationRedisSerializer，不知道小伙伴們有沒有注意過，直接用這個序列化工具將來存到 Redis 上的 key 和 value 都會莫名其妙多一些前綴，這就導(dǎo)致你用命令讀取的時候可能會出錯。

例如存儲的時候，key 是 name，value 是 javaboy，但是當你在命令行操作的時候，get name?卻獲取不到你想要的數(shù)據(jù)，原因就是存到 redis 之后 name 前面多了一些字符，此時只能繼續(xù)使用 RedisTemplate 將之讀取出來。

我們用 Redis 做限流會用到 Lua 腳本，使用 Lua 腳本的時候，就會出現(xiàn)上面說的這種情況，所以我們需要修改 RedisTemplate 的序列化方案。

?

可能有小伙伴會說為什么不用 StringRedisTemplate 呢？StringRedisTemplate 確實不存在上面所說的問題，但是它能夠存儲的數(shù)據(jù)類型不夠豐富，所以這里不考慮。

修改 RedisTemplate 序列化方案，代碼如下：

@Configuration
public?class?RedisConfig?{

????@Bean
????public?RedisTemplate?redisTemplate(RedisConnectionFactory?connectionFactory)?{
????????RedisTemplate?redisTemplate?=?new?RedisTemplate<>();
????????redisTemplate.setConnectionFactory(connectionFactory);
????????//?使用Jackson2JsonRedisSerialize?替換默認序列化(默認采用的是JDK序列化)
????????Jackson2JsonRedisSerializer