RateLimiter 笔记

ゝ一世哀愁。 2022-02-20 03:49 62阅读 0赞

关于谷歌的RateLimiter限流的使用工具类记录

package org.limit.example;
    
    import com.google.common.util.concurrent.RateLimiter;
    
    import java.util.concurrent.ConcurrentHashMap;
    
    /**
     * @Title: RateLimiterUtil
     * @Package: org.limit.example
     * @ProjectName: lruDemo
     * @Description: //TODO
     * @author: xupeng
     * @date: 2019/4/11 16:28
     * @version: V1.0
     */
    public class RateLimiterUtil {
    
        private static double defuntQps = 1.0;
    
        private static ConcurrentHashMap<String, RateLimiter> limiterMap = new ConcurrentHashMap<String, RateLimiter>();
    
        public static boolean tryAcquire(String resource, double qps) {
            RateLimiter rateLimiter = limiterMap.get(resource);
            if (rateLimiter == null) {
                rateLimiter = RateLimiter.create(qps);
                RateLimiter rateLitimerByOtherThread = limiterMap.putIfAbsent(resource, rateLimiter);
                if (rateLitimerByOtherThread != null) {
                    rateLimiter = rateLitimerByOtherThread;
                }
            }
            rateLimiter.setRate(qps);
            if (rateLimiter.tryAcquire()) {
                return true;
            }
            return false;
        }
        public static boolean tryAcquire(String resource) {
           return tryAcquire(resource,defuntQps);
        }
    
        
    
        public static void main(String[] args)throws Exception{
            String resource="xxx";
            for (int i = 0; i < 100; i++) {
                if (tryAcquire(resource)){
                    System.out.println(i);
                }
                Thread.sleep(10);
            }
            Thread.sleep(10000);
            System.out.println("--------------------");
            for (int i = 0; i < 100; i++) {
                if (tryAcquire(resource,10)){
                    System.out.println(i);
                }
                Thread.sleep(10);
            }
        }
    }

参考资料  
[https://tech.kujiale.com/ratelimiter-architecture/][https_tech.kujiale.com_ratelimiter-architecture]  
[https://www.jianshu.com/p/3dfae5c15eb9][https_www.jianshu.com_p_3dfae5c15eb9]

[https_tech.kujiale.com_ratelimiter-architecture]: https://tech.kujiale.com/ratelimiter-architecture/
[https_www.jianshu.com_p_3dfae5c15eb9]: https://www.jianshu.com/p/3dfae5c15eb9