基于Redis的限流器的實(shí)現(xiàn)(示例講解)

發(fā)布時(shí)間：2020-08-22 14:43:12 來(lái)源：腳本之家閱讀：310 作者：秦霜欄目：數(shù)據(jù)庫(kù)

1 概述

系統(tǒng)中的接口通常都有限流，比如 70次/秒，如何保證我們的接口的調(diào)用次數(shù)在超過(guò)第三方接口限流的時(shí)候快速失敗呢？這時(shí)候就需要限流器了。下面是筆者用redis實(shí)現(xiàn)限流器的流程圖。

2 代碼

/**
 * 獲取限流權(quán)限
 * @param key
 * @param millisecond 毫秒數(shù)
 * @param limitCount 限流次數(shù)
 * @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;
}

以上這篇基于Redis的限流器的實(shí)現(xiàn)(示例講解)就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

基于Redis的限流器的實(shí)現(xiàn)(示例講解)

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽