美文网首页
Guava RateLimiter限流示例

Guava RateLimiter限流示例

作者: an_bear | 来源:发表于2018-09-01 13:59 被阅读0次

本文章主要有两部分。

第一部分主要是提供限流器的一个构建方法,以及提供流速调节的方法。

第二部分是讲述怎么使用限流器。

第一部分:

这一部分主要是提供构建一个限流器,并且提供流速的实时调节

import java.util.HashMap;

import java.util.Map;

import java.util.concurrent.ConcurrentHashMap;

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

/*  

 * 用于限流和流量调节

 * 原理是基于令牌桶算法

 * 使用方式是:

 * 1.通过getRateLimiter获取rateLimiter.

 * 2.然后根据rateLimiter.tryAcquire()的返回的布尔值来进行判断是否可以执行正常流程。

 * 3.如果返回的是true,则表明获取到令牌可以执行正常的流程,否则就意味着没有获取令牌,被限流。

 * 有两点需要注意的:

 * 第一,RateLimiter是按照一定频率往桶里放入令牌,频率是1/QPS秒

 * 第二,RateLimiter最多会存储QPS个令牌

 * 

 */

public class RateLimiterBuilder {

    private static final double DEFALUT_RATE = 200;

    staticObjectobj=newObject();

    private static Map rateLimiterMap=new ConcurrentHashMap();

    /*  

     * 根据key和rate获取一个rateLimiter

     * rate是流速,也就是允许的QPS

     * 1.根据key去map里取rateLimiter,然后还要判断现在的rate和传人的rate是否相等,如果已经有这样一个rateLimiter(key和rate)就直接返回

     * 2.如果传人的rate和现在的不相等则调用changeRate进行调速度

     * 3.否则创建一个新的rateLimiter

     */

public static RateLimiter getRateLimiter(String key,double rate) {

if(rate<= 0) {

rate = DEFALUT_RATE;

}

RateLimiter limiter=rateLimiterMap.get(key);

if(limiter!=null&&limiter.getRate() ==rate) {

return limiter;

}

synchronized (obj) {

limiter = rateLimiterMap.get(key);

if(limiter!=null&&limiter.getRate() ==rate) {

return limiter;

}

if(limiter!=null) {

return changeRate(limiter,rate);

}

limiter= RateLimiter.create(rate);

rateLimiterMap.put(key, limiter);

}

return limiter;

}

    /*  

     * 调整RateLimiter的流量

     */

public static synchronized RateLimiter changeRate(RateLimiterlimiter,doublerate) {

if(limiter.getRate() ==rate){

returnlimiter;

}

        limiter.setRate(rate);

        returnlimiter;

    }

}

第二部分:

使用示例:

double rate = get();//通过配置中心去获取一个限流的速率

RateLimiter limiter= RateLimiterBuilder.getRateLimiter("limiterName",rate);

if(limiter!=null) {

        if(!limiter.tryAcquire()) {

         return weatherForecastResp;//被限流了,直接返回

        }

}

//下面是相关的业务代码。

也可以写一个拦截器,将这些代码包起来,在方法执行前,拦截一下,做限流。

相关文章

网友评论

      本文标题:Guava RateLimiter限流示例

      本文链接:https://www.haomeiwen.com/subject/gmfuwftx.html