移动专线

最新推荐文章于 2022-04-17 16:44:05 发布

H411068

最新推荐文章于 2022-04-17 16:44:05 发布

阅读量1.7k

点赞数

本文链接：https://blog.csdn.net/u014336021/article/details/107722825

版权

本文介绍了如何使用 Spring Cloud Gateway 结合 Guava 的 Bucket4j 实现基于 IP 的限流策略。在高并发场景下，通过令牌桶限流保护服务，控制速率和并发，避免超过服务承受范围。首先引入 Bucket4j 依赖，然后自定义 GatewayFilter，利用 ServerWebExchange 获取 IP 信息创建令牌桶，并在过滤器中实现限流逻辑。当令牌用尽时，返回429状态码，阻止请求。配置路由规则，将限流过滤器应用于特定路径，实现定制化的流量控制。

摘要由CSDN通过智能技术生成

当服务在短时间内迎来高并发，并发量超过服务承受的范围就需要使用限流。

通过请求限速或者对一个时间窗口内的请求进行限速来保护服务。当达到限制速率则可以拒绝请求，返回错误代码，或者定向到友好页面。

一般的中间件都会有单机限流框架，支持两种限流模式：
控制速率
控制并发

这里通过 Guava 中的 Bucket4j 来实现限流操作。按照惯例引入 Bucket4j 的依赖：

<dependency>
    <groupId>com.github.vladimir-bukhtoyarov</groupId>
    <artifactId>bucket4j-core</artifactId>
    <version>4.0.0</version>
</dependency>

由于需要对于用户请求进行监控，因此通过实现 GatewayFilter 的方式自定义 Filter，然后再通过 Gateway API Application 应用这个自定义的 Filter。

这里我们使用的是令牌桶的方式进行限流，因此需要设置桶的容量（capacity），每次填充的令牌数量（refillTokens）以及填充令牌的间隔时间（refillDuration）。

初始化这三个参数以后，通过 createNewBucket 方法针对请求建立令牌桶（bucket），在 Filter 方法中实现限流的主要逻辑。

通过 ServerWebExchange 获取请求的上下文中的 IP 信息，针对 IP 建立对应的令牌桶，这个 IP 与令牌桶的对应关系放到了 LOCAL_CACHE 中