guava的限流工具RateLimiter使用

guava限流工具使用

非常詳細的一篇使用博客：https://www.cnblogs.com/yeyinfu/p/7316972.html

1，原理：Guava RateLimiter基於令牌桶算法，我們只需要告訴RateLimiter系統限制的QPS是多少，那么RateLimiter將以這個速度往桶里面放入令牌，然后請求的時候，通過tryAcquire()方法向RateLimiter獲取許可（令牌）。

2，測試代碼：

/**
 * @Auther: wangbao02
 * @Date: 2018/6/29 12:33
 * @Description:谷歌的guava限流工具
 */
public class GuavaRateLimiter {

    public static ConcurrentHashMap<String, RateLimiter> resourceRateLimiter = new ConcurrentHashMap<String, RateLimiter>();

    //初始化限流工具RateLimiter
    static {
        createResourceRateLimiter("order", 50);
    }

    public static void createResourceRateLimiter(String resource, double qps) {
        if (resourceRateLimiter.contains(resource)) {
            resourceRateLimiter.get(resource).setRate(qps);
        } else {
            //創建限流工具，每秒發出50個令牌指令
            RateLimiter rateLimiter = RateLimiter.create(qps);
            resourceRateLimiter.putIfAbsent(resource, rateLimiter);

        }

    }

    public static void main(String[] args) {
        for (int i = 0; i < 5000; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    //如果獲得令牌指令，則執行業務邏輯
                    if (resourceRateLimiter.get("order").tryAcquire(10, TimeUnit.MICROSECONDS)) {
                        System.out.println("執行業務邏輯");
                    } else {
                        System.out.println("限流");
                    }
                }
            }).start();
        }

    }
}

3，執行結果