guava的限流工具RateLimiter使用

guava限流工具使用html

很是詳細的一篇使用博客:https://www.cnblogs.com/yeyinfu/p/7316972.htmljava

1,原理:Guava RateLimiter基於令牌桶算法,咱們只須要告訴RateLimiter系統限制的QPS是多少,那麼RateLimiter將以這個速度往桶裏面放入令牌,而後請求的時候,經過tryAcquire()方法向RateLimiter獲取許可(令牌)。算法

2,測試代碼:ide

/**
 * @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,執行結果工具

相關文章
相關標籤/搜索