簡介

市面上很多介紹redis如何實(shí)現(xiàn)限流的，但是大部分都有一個缺點(diǎn)，就是只能實(shí)現(xiàn)單一的限流，比如1分鐘訪問1次或者60分鐘訪問10次這種，但是如果想一個接口兩種規(guī)則都需要滿足呢，我們的項(xiàng)目又是分布式項(xiàng)目，應(yīng)該如何解決，下面就介紹一下redis實(shí)現(xiàn)分布式多規(guī)則限流的方式。

思考

如何一分鐘只能發(fā)送一次驗(yàn)證碼，一小時只能發(fā)送10次驗(yàn)證碼等等多種規(guī)則的限流

如何防止接口被惡意打擊（短時間內(nèi)大量請求）

如何限制接口規(guī)定時間內(nèi)訪問次數(shù)

解決方法

記錄某IP訪問次數(shù)

使用 String結(jié)構(gòu) 記錄固定時間段內(nèi)某用戶IP訪問某接口的次數(shù)

RedisKey = prefix : className : methodName

RedisVlue = 訪問次數(shù)

攔截請求:

初次訪問時設(shè)置 「[RedisKey] [RedisValue=1] [規(guī)定的過期時間]」

獲取 RedisValue 是否超過規(guī)定次數(shù)，超過則攔截，未超過則對 RedisKey 進(jìn)行加1

分析: 規(guī)則是每分鐘訪問 1000 次

考慮并發(fā)問題

假設(shè)目前 RedisKey => RedisValue 為 999

目前大量請求進(jìn)行到第一步( 獲取Redis請求次數(shù) )，那么所有線程都獲取到了值為999，進(jìn)行判斷都未超過限定次數(shù)則不攔截，導(dǎo)致實(shí)際次數(shù)超過 1000 次

「解決辦法:」

保證方法執(zhí)行原子性（加鎖、lua）

考慮在臨界值進(jìn)行訪問

思考下圖

代碼實(shí)現(xiàn): 比較簡單，

Zset解決臨界值問題

使用 Zset 進(jìn)行存儲，解決臨界值訪問問題

網(wǎng)上幾乎都有實(shí)現(xiàn)，這里就不過多介紹

實(shí)現(xiàn)多規(guī)則限流

先確定最終需要的效果

能實(shí)現(xiàn)多種限流規(guī)則

能實(shí)現(xiàn)防重復(fù)提交

通過以上要求設(shè)計注解（先想象出最終實(shí)現(xiàn)效果）

@RateLimiter(
rules={
//60秒內(nèi)只能訪問10次
@RateRule(count=10,time=60,timeUnit=TimeUnit.SECONDS),
//120秒內(nèi)只能訪問20次
@RateRule(count=20,time=120,timeUnit=TimeUnit.SECONDS)

},
//防重復(fù)提交（5秒鐘只能訪問1次）
preventDuplicate=true
)

編寫注解（RateLimiter，RateRule）

編寫 RateLimiter 注解。

/**
*@Description:請求接口限制
*@Author:yiFei
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Inherited
public@interfaceRateLimiter{

/**
*限流key
*/
Stringkey()defaultRedisKeyConstants.RATE_LIMIT_CACHE_PREFIX;

/**
*限流類型(默認(rèn)Ip模式)
*/
LimitTypeEnumlimitType()defaultLimitTypeEnum.IP;

/**
*錯誤提示
*/
ResultCodemessage()defaultResultCode.REQUEST_MORE_ERROR;

/**
*限流規(guī)則（規(guī)則不可變，可多規(guī)則）
*/
RateRule[]rules()default{};

/**
*防重復(fù)提交值
*/
booleanpreventDuplicate()defaultfalse;

/**
*防重復(fù)提交默認(rèn)值
*/
RateRulepreventDuplicateRule()default@RateRule(count=1,time=5);
}

編寫RateRule注解

@Target(ElementType.ANNOTATION_TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Inherited
public@interfaceRateRule{

/**
*限流次數(shù)
*/
longcount()default10;

/**
*限流時間
*/
longtime()default60;

/**
*限流時間單位
*/
TimeUnittimeUnit()defaultTimeUnit.SECONDS;

}

攔截注解 RateLimiter

確定redis存儲方式

RedisKey = prefix : className : methodName

RedisScore = 時間戳

RedisValue = 任意分布式不重復(fù)的值即可

編寫生成 RedisKey 的方法

/**
*通過rateLimiter和joinPoint拼接prefix:ip/userId:classSimpleName-methodName
*
*@paramrateLimiter提供prefix
*@paramjoinPoint提供classSimpleName:methodName
*@return
*/
publicStringgetCombineKey(RateLimiterrateLimiter,JoinPointjoinPoint){
StringBufferkey=newStringBuffer(rateLimiter.key());
//不同限流類型使用不同的前綴
switch(rateLimiter.limitType()){
//XXX可以新增通過參數(shù)指定參數(shù)進(jìn)行限流
caseIP:
key.append(IpUtil.getIpAddr(((ServletRequestAttributes)Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest())).append(":");
break;
caseUSER_ID:
SysUserDetailsuser=SecurityUtil.getUser();
if(!ObjectUtils.isEmpty(user))key.append(user.getUserId()).append(":");
break;
caseGLOBAL:
break;
}
MethodSignaturesignature=(MethodSignature)joinPoint.getSignature();
Methodmethod=signature.getMethod();
ClasstargetClass=method.getDeclaringClass();
key.append(targetClass.getSimpleName()).append("-").append(method.getName());
returnkey.toString();
}

編寫lua腳本

編寫lua腳本 （兩種將時間添加到Redis的方法）。

Zset的UUID value值

UUID（可用其他有相同的特性的值）為Zset中的value值

參數(shù)介紹

KEYS[1] = prefix : ? : className : methodName

KEYS[2] = 唯一ID

KEYS[3] = 當(dāng)前時間

ARGV = [次數(shù)，單位時間，次數(shù)，單位時間, 次數(shù), 單位時間 ...]

由java傳入分布式不重復(fù)的 value 值

--1.獲取參數(shù)
localkey=KEYS[1]
localuuid=KEYS[2]
localcurrentTime=tonumber(KEYS[3])
--2.以數(shù)組最大值為ttl最大值
localexpireTime=-1;
--3.遍歷數(shù)組查看是否超過限流規(guī)則
fori=1,#ARGV,2do
localrateRuleCount=tonumber(ARGV[i])
localrateRuleTime=tonumber(ARGV[i+1])
--3.1判斷在單位時間內(nèi)訪問次數(shù)
localcount=redis.call('ZCOUNT',key,currentTime-rateRuleTime,currentTime)
--3.2判斷是否超過規(guī)定次數(shù)
iftonumber(count)>=rateRuleCountthen
returntrue
end
--3.3判斷元素最大值，設(shè)置為最終過期時間
ifrateRuleTime>expireTimethen
expireTime=rateRuleTime
end
end
--4.redis中添加當(dāng)前時間
redis.call('ZADD',key,currentTime,uuid)
--5.更新緩存過期時間
redis.call('PEXPIRE',key,expireTime)
--6.刪除最大時間限度之前的數(shù)據(jù)，防止數(shù)據(jù)過多
redis.call('ZREMRANGEBYSCORE',key,0,currentTime-expireTime)
returnfalse

根據(jù)時間戳作為Zset中的value值

參數(shù)介紹

KEYS[1] = prefix : ? : className : methodName

KEYS[2] = 當(dāng)前時間

ARGV = [次數(shù)，單位時間，次數(shù)，單位時間, 次數(shù), 單位時間 ...]

根據(jù)時間進(jìn)行生成value值，考慮同一毫秒添加相同時間值問題

以下為第二種實(shí)現(xiàn)方式，在并發(fā)高的情況下效率低，value是通過時間戳進(jìn)行添加，但是訪問量大的話會使得一直在調(diào)用 redis.call('ZADD', key, currentTime, currentTime)，但是在不沖突value的情況下，會比生成 UUID 好

--1.獲取參數(shù)
localkey=KEYS[1]
localcurrentTime=KEYS[2]
--2.以數(shù)組最大值為ttl最大值
localexpireTime=-1;
--3.遍歷數(shù)組查看是否越界
fori=1,#ARGV,2do
localrateRuleCount=tonumber(ARGV[i])
localrateRuleTime=tonumber(ARGV[i+1])
--3.1判斷在單位時間內(nèi)訪問次數(shù)
localcount=redis.call('ZCOUNT',key,currentTime-rateRuleTime,currentTime)
--3.2判斷是否超過規(guī)定次數(shù)
iftonumber(count)>=rateRuleCountthen
returntrue
end
--3.3判斷元素最大值，設(shè)置為最終過期時間
ifrateRuleTime>expireTimethen
expireTime=rateRuleTime
end
end
--4.更新緩存過期時間
redis.call('PEXPIRE',key,expireTime)
--5.刪除最大時間限度之前的數(shù)據(jù)，防止數(shù)據(jù)過多
redis.call('ZREMRANGEBYSCORE',key,0,currentTime-expireTime)
--6.redis中添加當(dāng)前時間(解決多個線程在同一毫秒添加相同value導(dǎo)致Redis漏記的問題)
--6.1maxRetries最大重試次數(shù)retries重試次數(shù)
localmaxRetries=5
localretries=0
whiletruedo
localresult=redis.call('ZADD',key,currentTime,currentTime)
ifresult==1then
--6.2添加成功則跳出循環(huán)
break
else
--6.3未添加成功則value+1再次進(jìn)行嘗試
retries=retries+1
ifretries>=maxRetriesthen
--6.4超過最大嘗試次數(shù)采用添加隨機(jī)數(shù)策略
localrandom_value=math.random(1,1000)
currentTime=currentTime+random_value
else
currentTime=currentTime+1
end
end
end

returnfalse

編寫 AOP 攔截

@Autowired
privateRedisTemplateredisTemplate;

@Autowired
privateRedisScriptlimitScript;

/**
*限流
*XXX對限流要求比較高，可以使用在Redis中對規(guī)則進(jìn)行存儲校驗(yàn)或者使用中間件
*
*@paramjoinPointjoinPoint
*@paramrateLimiter限流注解
*/
@Before(value="@annotation(rateLimiter)")
publicvoidboBefore(JoinPointjoinPoint,RateLimiterrateLimiter){
//1.生成key
Stringkey=getCombineKey(rateLimiter,joinPoint);
try{
//2.執(zhí)行腳本返回是否限流
Booleanflag=redisTemplate.execute(limitScript,
ListUtil.of(key,String.valueOf(System.currentTimeMillis())),
(Object[])getRules(rateLimiter));
//3.判斷是否限流
if(Boolean.TRUE.equals(flag)){
log.error("ip:'{}'攔截到一個請求RedisKey:'{}'",
IpUtil.getIpAddr(((ServletRequestAttributes)Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest()),
key);
thrownewServiceException(rateLimiter.message());
}
}catch(ServiceExceptione){
throwe;
}catch(Exceptione){
e.printStackTrace();
}
}

/**
*獲取規(guī)則
*
*@paramrateLimiter獲取其中規(guī)則信息
*@return
*/
privateLong[]getRules(RateLimiterrateLimiter){
intcapacity=rateLimiter.rules().length<

	

	


	審核編輯：劉清