快捷導(dǎo)航

SpringCloud Gateway實(shí)現(xiàn)限流功能詳解

更新時(shí)間：2022年11月17日 14:38:15 作者：做一道光

SpringCloud Gateway 是 Spring Cloud 的一個(gè)全新項(xiàng)目，它旨在為微服務(wù)架構(gòu)提供一種簡(jiǎn)單有效的統(tǒng)一的 API 路由管理方式。這篇文章主要介紹了SpringCloud Gateway實(shí)現(xiàn)限流,需要的朋友可以參考下

1 什么是限流

通俗的說，限流就是限制一段時(shí)間內(nèi)，用戶訪問資源的次數(shù) ，減輕服務(wù)器壓力，限流大致分為兩種：

1. IP 限流（5s 內(nèi)同一個(gè) ip 訪問超過 3 次，則限制不讓訪問，過一段時(shí)間才可繼續(xù)訪問）

2. 請(qǐng)求量限流（只要在一段時(shí)間內(nèi)(窗口期)，請(qǐng)求次數(shù)達(dá)到閥值，就直接拒絕后面來的訪問了，過一段時(shí)間才可以繼續(xù)訪問）（粒度可以細(xì)化到一個(gè) api（url），一個(gè)服務(wù)）

2 本次限流模型

限流模型 : 漏斗算法，令牌桶算法，窗口滑動(dòng)算法計(jì)數(shù)器算法

入不敷出

1）、所有的請(qǐng)求在處理之前都需要拿到一個(gè)可用的令牌才會(huì)被處理；

2）、根據(jù)限流大小，設(shè)置按照一定的速率往桶里添加令牌；

3）、桶設(shè)置最大的放置令牌限制，當(dāng)桶滿時(shí)、新添加的令牌就被丟棄或者拒絕；

4）、請(qǐng)求達(dá)到后首先要獲取令牌桶中的令牌，拿著令牌才可以進(jìn)行其他的業(yè)務(wù)邏輯，處理完業(yè)務(wù)邏輯之后，將令牌直接刪除；

5）、令牌桶有最低限額，當(dāng)桶中的令牌達(dá)到最低限額的時(shí)候，請(qǐng)求處理完之后將不會(huì)刪除令牌，以此保證足夠的限流；

3 Gateway結(jié)合redis實(shí)現(xiàn)請(qǐng)求量限流

Spring Cloud Gateway 已經(jīng)內(nèi)置了一個(gè) RequestRateLimiterGatewayFilterFactory，我們可以直接使用。

目前 RequestRateLimiterGatewayFilterFactory 的實(shí)現(xiàn)依賴于 Redis，所以我們還要引入spring-boot-starter-data-redis-reactive。

3.1 添加依賴

<!--限流要引入 Redis-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-redis-reactive</artifactId>
</dependency>

3.2 修改配置文件

server :

port: 80

spring:

application:

name: gateway-80

cloud:

gateway :

enabled: true

routes:

id: user-service

uri: lb: //consumer-user-service

predicates:

- Path=/info/**

filters:

- name : RequestRateLimiter

args:

key-resolver:'#{@hostAddrKeyResolver} '

redis-rate-limiter.replenishRate: 1

redis-rate-limiter . burstCapacity: 3

redis:#redis的配置

host: 192.168.226.128

port: 6379

database: 6

eureka :

instance:

instance-id: ${spring.application.name}:${server.port}

prefer-ip-address: true

client:

service-url:

defaultzone: http:/ / localhost:8761/eureka/

3.3 配置文件說明

在上面的配置文件，配置了 redis 的信息，并配置了 RequestRateLimiter 的限流過濾器，該過濾器需要配置三個(gè)參數(shù)：

burstCapacity：令牌桶總?cè)萘俊?/p>

replenishRate：令牌桶每秒填充平均速率。

key-resolver：用于限流的鍵的解析器的 Bean 對(duì)象的名字。它使用 SpEL 表達(dá)式根據(jù)

3.4 創(chuàng)建配置類RequestRateLimiterConfig

@Configuration
public class RequestRateLimiterConfig {
/**
* IP 限流
* 把用戶的 IP 作為限流的 Key
*
* @return
*/
@Bean
@Primary
public KeyResolver hostAddrKeyResolver() {
return (exchange) -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}
/**
* 用戶 id 限流
* 把用戶 ID 作為限流的 key
*
* @return
*/
@Bean
public KeyResolver userKeyResolver() {
return exchange -> Mono.just(exchange.getRequest().getQueryParams().getFirst("userId"));
}
/**
* 請(qǐng)求接口限流
* 把請(qǐng)求的路徑作為限流 key
*
* @return
*/
@Bean
public KeyResolver apiKeyResolver() {
return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
}