限流算法

最新推荐文章于 2025-04-25 18:31:28 发布

sinat_25076521

最新推荐文章于 2025-04-25 18:31:28 发布

阅读量205

点赞数

分类专栏： Java基础

本文链接：https://blog.youkuaiyun.com/sinat_25076521/article/details/81212750

版权

Java基础专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个基于Guava的流量整形器实现。该实现利用了ConcurrentMap和RateLimiter来跟踪和限制每个用户的请求速率。文章详细展示了如何更新和移除用户限速设置，以及如何判断请求是否可以被接受。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import java.util.concurrent.ConcurrentMap;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.collect.Maps;
import com.google.common.util.concurrent.RateLimiter;

/**
* @author fuyongqian
*
*/
public class TrafficShaper {
   private static final Logger log = LoggerFactory.getLogger(TrafficShaper.class);
   private static final ConcurrentMap<String,Double> resoureMap = Maps.newConcurrentMap();
   private static final ConcurrentMap<String,RateLimiter> limiterMap = Maps.newConcurrentMap();

   public static void updateResourceQps(String userkey,double qps){
       resoureMap.put(userkey, qps);
       if(qps > 0.0){
           RateLimiter limiter = limiterMap.get(userkey);
           if (limiter!=null ){
               if (limiter.getRate() != qps){
                   limiter.setRate(qps);
               }
           }else{
               limiter = RateLimiter.create(qps);
               RateLimiter putByOtherThread = limiterMap.putIfAbsent(userkey, limiter);
               if (putByOtherThread != null) limiter = putByOtherThread;
           }
       }
   }

   public static void removeResourcesQps(String userkey){
       resoureMap.remove(userkey);
   }

   public static boolean enter (String userkey){
       if (resoureMap == null){
           return true;
       }
       double qps = resoureMap.get(userkey) != null ? resoureMap.get(userkey):0;
       if (qps <= 0.0){
           return true;
       }
       RateLimiter limiter = limiterMap.get(userkey);
       if (limiter == null){
           log.info("userkey为"+userkey+"的限流对象初始化异常或没有初始化");
           return true;
       }
       return limiter.tryAcquire();
   }

}