Guava RateLimiter限流示例

本文章主要有两部分。

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

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

第一部分:

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

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;//被限流了，直接返回

        }

}

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

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