基于RateLimiter的接口限流器

最新推荐文章于 2024-08-21 09:00:39 发布

Py_Wang

最新推荐文章于 2024-08-21 09:00:39 发布

阅读量1k

点赞数 1

分类专栏： Java 文章标签： RateLimiter

本文链接：https://blog.csdn.net/Py_Wang/article/details/103603754

版权

本文介绍了Google Guava库中的RateLimiter如何用于接口限流，详细阐述了其工作原理，包括两种创建方法及限流策略。并展示了通过RateLimiterFactory创建实例、注解方式和代码插入方式实现限流的详细步骤。

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

知识前述

RateLimiter是Google开源工具包Guava提供的限流工具类，该类基于令牌桶算法实现流量限制。

RateLimiter有两种构造方法：

（1）通过RateLimiter.create(double permitsPerSecond)创建实例，该实例以固定的速率生成令牌，参数permitsPerSecond代表每秒生成的令牌桶数量，令牌桶存放一秒的令牌数量；

（2）通过RateLimiter.create(double permitsPerSecond, long warmupPeriod, TimeUnit unit)方法创建限流器实例，该实例初始生成令牌的速率为零，经过一定时间段warmupPeriod后，速率达到指定的permitsPerSecond。

限流方法：

（1）通过ratelimiter.acquire(int permits)获取指定数量的令牌，等待直至获取到令牌；

（2）通过ratelimiter.tryAcquire(int permits, long timeout, TimeUnit unit)在指定时间内获取指定数量令牌，通过计算获取到令牌的时间，若小于指定时间，则等待直至获取到令牌返回true，若大于指定时间，则直接返回false。

限流器的功能

限流器需要实现两个功能：

（1）提供注解的方式，采用AOP对指定方法限流；

（2）采用代码插入方式限流。

由于一个RateLimiter实例生成令牌的速度固定，若多个地方同时使用，则该RateLimiter限制的速度是几个地方的调用速度之和。因此，为明确每个需要限流的接口的限流速率，需要为每个限流接口生成一个RateLimiter实例。

限流器的具体实现

1、RateLimiterFactory工厂类——生成RateLimiter实例

为了保证并发状态下，同一个接口多次调用，使用同一个RateLimiter实例，通过一个ConcurrentHashMap存储所有RateLimiter实例，key为该实例所在接口的信息。

调用getRateLimiter(String location, int rate)方法获取RateLimiter，并调整令牌生成速率rate。若该location没有对应的RateLimiter，则新建实例。

import com.google.common.util.concurrent.RateLimiter;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurr