guava的限流工具RateLimiter使用

guava限流工具使用

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

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

2，測試代碼：

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，執行結果