基于Redis的限流器的實作

1.概述

系統中的接口通常都有限流，比如 70次/秒 ，如何保證我們的接口的調用次數在超過第三方接口限流的時候快速失敗呢？這時候就需要限流器了。下面是筆者用redis實作限流器的流程圖。 

2.代碼

/**
 * 擷取限流權限
 * @param key
 * @param millisecond 毫秒數
 * @param limitCount 限流次數
 * @return
 */
public static boolean getCurrentLimiting(String key, Long millisecond, Integer limitCount){
    try {
        boolean currentLimitingLock = RedisDistributeLockUtil.lock(key);
        if(currentLimitingLock){
            Long llen = RedisClient.llen(getCurrentLimitingRedisKey(key));
            if(llen < limitCount){
                RedisClient.lpush(getCurrentLimitingRedisKey(key), DateUtil.current(false) + "");
                return true;
            }else{
                Long lastTime = Long.parseLong(RedisClient.lindex(getCurrentLimitingRedisKey(key), -1));
                if((DateUtil.current(false) - lastTime) >= millisecond){
                    RedisClient.lpush(getCurrentLimitingRedisKey(key), DateUtil.current(false) + "");
                    RedisClient.ltrim(getCurrentLimitingRedisKey(key), 0, limitCount);
                    return true;
                }
            }
        }
    }finally {
        RedisDistributeLockUtil.unlock(key);
    }
    return false;
}
           

--------------------- 

作者：秦霜 

來源：CSDN 

原文：https://blog.csdn.net/wang258533488/article/details/78913827 

版權聲明：本文為部落客原創文章，轉載請附上博文連結！