本文主要研究的是RateLimit——使用番石榴来做接口限流的相关内容,具体如下。
某天一个君突然发现自己的接口请求量突然涨到之前的10倍,没多久该接口几乎不可使用,并引发连锁反应导致整个系统崩溃。如何应对这种情况呢?生活给了我们答案:比如老式电闸都安装了保险丝,一旦有人使用超大功率的设备,保险丝就会烧断以保护各个电器不被强电流给烧坏。同理我们的接口也需要安装上“保险丝”,以防止非预期的请求对系统压力过大而引起的系统瘫痪,当流量过大时,可以采取拒绝或者引流等机制。
常用的限流算法有两种:漏桶算法和令牌桶算法。
漏桶算法思路很简单,请求先进入到漏桶里,漏桶以一定的速度出水,当水请求过大会直接溢出,可以看出漏桶算法能强行限制数据的传输速率。
<强>图1漏桶算法示意图强>
对于很多应用场景来说,除了要求能够限制数据的平均传输速率外,还要求允许某种程度的突发传输。这时候漏桶算法可能就不合适了,令牌桶算法更为适合。如图2所示,令牌桶算法的原理是系统会以一个恒定的速度往桶里放入令牌,而如果请求需要被处理,则需要先从桶里获取一个令牌,当桶里没有令牌可取时,则拒绝服务。
<强>图2令牌桶算法示意图
强>
谷歌开源工具包番石榴提供了限流工具类RateLimiter,该类基于“令牌桶算法”,非常方便使用。该类的接口具体的使用请参考:RateLimiter使用实践。
RateLimiter使用演示
包ratelimite; 进口com.google.common.util.concurrent.RateLimiter; 公开课RateLimiterDemo { 公共静态void main (String [] args) { testNoRateLimiter (); testWithRateLimiter (); } 公共静态孔隙testNoRateLimiter () { 长开始=System.currentTimeMillis (); for (int i=0;我& lt;10;我+ +){ system . out。println(“调用执行. .”+ i); } 长债=System.currentTimeMillis (); system . out。println(结束-开始); } 公共静态孔隙testWithRateLimiter () { 长开始=System.currentTimeMillis (); RateLimiter限幅器=RateLimiter.create (10.0);//每秒不超过10个任务被提交 for (int i=0;我& lt;10;我+ +){ limiter.acquire ();//请求RateLimiter,超过许可证会被阻塞 system . out。println(“调用执行. .”+ i); } 长债=System.currentTimeMillis (); system . out。println(结束-开始); } }
<强>概念强>
ListenableFuture顾名思义就是可以监听的未来,它是对java原生未来的扩展增强。我们知道未来表示一个异步计算任务,当任务完成时可以得到计算结果。如果我们希望一旦计算完成就拿到结果展示给用户或者做另外的计算,就必须使用另一个线程不断的查询计算状态。这样做,代码复杂,而且效率低下。使用ListenableFuture番石榴帮我们检测未来是否完成了,如果完成就自动调用回调函数,这样可以减少并发程序的复杂度。
推荐使用第二种方法,因为第二种方法可以直接得到未来的返回值,或者处理错误情况。本质上第二种方法是通过调动第一种方法实现的,做了进一步的封装。
另外ListenableFuture还有其他几种内置实现:
SettableFuture:不需要实现一个方法来计算返回值,而只需要返回一个固定值来做为返回值,可以通过程序设置此未来的返回值或者异常信息
CheckedFuture:这是一个继承自ListenableFuture接口,他提供了checkedGet()方法,此方法在未来执行发生异常时,可以抛出指定类型的异常。
RateLimiter类似于JDK的信号量Semphore,他用来限制对资源并发访问的线程数,本文介绍RateLimiter使用
<强>代码示例强>
进口java.util.concurrent.Callable; 进口java.util.concurrent.ExecutionException; 进口java.util.concurrent.Executors; 进口java.util.concurrent.TimeUnit; 进口com.google.common.util.concurrent.FutureCallback; 进口com.google.common.util.concurrent.Futures; 进口com.google.common.util.concurrent.ListenableFuture; 进口com.google.common.util.concurrent.ListeningExecutorService; 进口com.google.common.util.concurrent.MoreExecutors; 进口com.google.common.util.concurrent.RateLimiter; 公开课ListenableFutureDemo { 公共静态void main (String [] args) { testRateLimiter (); testListenableFuture (); }/* * * RateLimiter类似于JDK的信号量Semphore,他用来限制对资源并发访问的线程数 */公共静态孔隙testRateLimiter () { ListeningExecutorService executorService=MoreExecutors .listeningDecorator (Executors.newCachedThreadPool ()); RateLimiter限幅器=RateLimiter.create (5.0);//每秒不超过4个任务被提交 for (int i=0;我& lt;10;我+ +){ limiter.acquire ();//请求RateLimiter,超过许可证会被阻塞 最后ListenableFutureRateLimit——使用番石榴来做接口限流代码示例