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