转载自   高并发场景下的限流策略

在高并发的场景下，我们的优化和保护系统的方式通常有：多级缓存、资源隔离、熔断降级、限流等等。

今天我们来聊聊限流。

 

为什么需要限流

举个比较简单的例子，正常来说，一个员工A他每天能够处理的工作是10个，突然某一天来了100个工作量，这时候，如果员工A还处理100个，只有一种可能，这个员工被压垮。

如果我们能预先知道会有100个任务会来，我们通过增加员工数或定义消息队列等等来临时解决。

但是我们很多时候无法预料这些意外的。根据墨菲定律，坏事往往会接踵而来，有可能某个点挂了会引起全局的挂掉（雪崩）。因此我们不得不对我们的系统做一些保护措施。限流是其中之一。

针对秒杀这类场景，我们也可以做一些限流措施，而不影响到系统全局。

 

限流方式之计数器(滑动窗口协议)

思路：限速，我们可能第一个想到的应该是，我通过一个计数器，进行技术，如果超过了计数器阀值，表示速度太快了。一秒一个计数器。

为了便于阅读，我只截图了主要的代码片段。

这样有个问题就是：粒度太大了，不均匀，针对1秒以下的，没法辨析。

我们能不能把粒度拆细了，1秒拆成10个100毫秒。每一个100毫秒有一个计数器。了解TCP/IP的应该知道，TCP/IP为了增加传输速度和控制传输速度，有个叫“滑动窗口协议”。

就算拆得再细，也无法解决匀速限制速度的问题。

而且还有个临界点问题，假如，一秒限制10个请求，在第1秒和第2秒之间，第1秒后半段时间10个请求，第2秒前半段10个请求，那第1秒后半段+第2秒前半段时间组成的一秒钟里就有20个请求，没有起到限速的作用。

有没有更好的办法呢？

 

限速方式之漏桶算法

在生活中，如果一桶有一个细眼，我们往里面装水，可以看到水是一滴一滴匀速的下落的，哪我们能不能通过程序来实现这种方式呢。

思路：桶为容器，一滴水为一请求。如果桶满了就拒绝请求，没满处理请求。

代码片段

在段代码中

• 首先计算这次请求与上次请求来的时候，总共漏了多少水。

• 看一下桶里面还剩多少水，有没有溢出。

• 如果溢出了拒绝请求，如果没有添加当前一滴水。处理请求。

对于很多应用场景来说，除了要求能够限制数据的平均传输速率外，还要求允许某种程度的突发传输。这时候漏桶算法可能就不合适了，令牌桶算法更为适合。

什么意思呢？就是说我服务前面闲了很久，突然来了很多请求（在桶的容量内），我得快速的把这些处理了。

 

限速方式之令牌桶算法

思路：匀速的产生令牌，往桶里面丢，每次请求来，看是否有多余的令牌。如果有获取令牌执行正常业务，偌没有限速。

代码片段

通过这种方式可以允许瞬时的大量处理，然后做限速处理。

• 请求来的时候先计算目前放入桶中的令牌数，这里计算，就可以不用启动一个线程匀速放置令牌了，这个叫惰性计算。

• 然后计算桶拥有的令牌数。然后获取令牌。做拒绝还是处理动作。

以上代码，可在Github查看。
https://github.com/hirudy/java_lib/tree/master/src/main/java/com/hirudy/limiter 

 

单机限速器RateLimiter

安利大家一个高效的限速器。

google的基础库guava中包含了一个基于令牌桶的限速器RateLimiter。使用也很简单。