RateLimit--使用guava来做接口限流-转载

灰太狼 2022-06-02 07:36 147阅读 0赞

原文链接：

http://blog.csdn.net/jiesa/article/details/50412027

# 一、问题描述   #

某天A君突然发现自己的接口请求量突然涨到之前的10倍，没多久该接口几乎不可使用，并引发连锁反应导致整个系统崩溃。如何应对这种情况呢？生活给了我们答案：比如老式电闸都安装了保险丝，一旦有人使用超大功率的设备，保险丝就会烧断以保护各个电器不被强电流给烧坏。同理我们的接口也需要安装上“保险丝”，以防止非预期的请求对系统压力过大而引起的系统瘫痪，当流量过大时，可以采取拒绝或者引流等机制。

# 二、常用的限流算法 #

常用的限流算法有两种：漏桶算法和令牌桶算法，这篇博文介绍得比较清晰（[过载保护算法浅析][Link 1]）。

漏桶算法思路很简单，请求先进入到漏桶里，漏桶以一定的速度出水，当水请求过大会直接溢出，可以看出漏桶算法能强行限制数据的传输速率。

![Center][]

图1 漏桶算法示意图

对于很多应用场景来说，除了要求能够限制数据的平均传输速率外，还要求允许某种程度的突发传输。这时候漏桶算法可能就不合适了，令牌桶算法更为适合。如图2所示，令牌桶算法的原理是系统会以一个恒定的速度往桶里放入令牌，而如果请求需要被处理，则需要先从桶里获取一个令牌，当桶里没有令牌可取时，则拒绝服务。

![Center 1][]

图2 令牌桶算法示意图

# 三、限流工具类RateLimiter #

google开源工具包guava提供了限流工具类RateLimiter，该类基于“令牌桶算法”，非常方便使用。该类的接口描述请参考：[RateLimiter接口描述][RateLimiter]，具体的使用请参考：[RateLimiter使用实践][RateLimiter 1]。

[Link 1]: http://sharecore.net/blog/2014/06/21/guo-zai-bao-hu-suan-fa-qian-xi/
[Center]: /images/20220602/32e66cc462c044049f27c2ae8d9646ce.png
[Center 1]: /images/20220602/9a00ffcf565e487e9e7de7f6bc6535fc.png
[RateLimiter]: http://docs.guava-libraries.googlecode.com/git-history/master/javadoc/com/google/common/util/concurrent/RateLimiter.html
[RateLimiter 1]: http://java.dzone.com/articles/ratelimiter-discovering-google