一、背景

1.為什么要做風(fēng)控?

這不得拜產(chǎn)品大佬所賜

目前我們業(yè)務(wù)有使用到非常多的AI能力,如ocr識(shí)別、語(yǔ)音測(cè)評(píng)等,這些能力往往都比較費(fèi)錢或者費(fèi)資源,所以在產(chǎn)品層面也希望我們對(duì)用戶的能力使用次數(shù)做一定的限制,因此風(fēng)控是必須的!

2.為什么要自己寫風(fēng)控?

那么多開(kāi)源的風(fēng)控組件,為什么還要寫呢?是不是想重復(fù)發(fā)明輪子呀.

要想回答這個(gè)問(wèn)題,需要先解釋下我們業(yè)務(wù)需要用到的風(fēng)控(簡(jiǎn)稱業(yè)務(wù)風(fēng)控),與開(kāi)源常見(jiàn)的風(fēng)控(簡(jiǎn)稱普通風(fēng)控)有何區(qū)別:

因此,直接使用開(kāi)源的普通風(fēng)控,一般情況下是無(wú)法滿足需求的

3.其它要求

支持實(shí)時(shí)調(diào)整限制

很多限制值在首次設(shè)置的時(shí)候,基本上都是拍定的一個(gè)值,后續(xù)需要調(diào)整的可能性是比較大的,因此可調(diào)整并實(shí)時(shí)生效是必須的

基于 Spring Boot + MyBatis Plus + Vue & Element 實(shí)現(xiàn)的后臺(tái)管理系統(tǒng) + 用戶小程序，支持 RBAC 動(dòng)態(tài)權(quán)限、多租戶、數(shù)據(jù)權(quán)限、工作流、三方登錄、支付、短信、商城等功能

• 項(xiàng)目地址：https://github.com/YunaiV/ruoyi-vue-pro
• 視頻教程：https://doc.iocoder.cn/video/

二、思路

要實(shí)現(xiàn)一個(gè)簡(jiǎn)單的業(yè)務(wù)風(fēng)控組件,要做什么工作呢?

1.風(fēng)控規(guī)則的實(shí)現(xiàn)

a.需要實(shí)現(xiàn)的規(guī)則:

• 自然日計(jì)數(shù)
• 自然小時(shí)計(jì)數(shù)
• 自然日+自然小時(shí)計(jì)數(shù)

自然日+自然小時(shí)計(jì)數(shù) 這里并不能單純地串聯(lián)兩個(gè)判斷,因?yàn)槿绻匀蝗盏呐卸ㄍㄟ^(guò),而自然小時(shí)的判定不通過(guò)的時(shí)候,需要回退,自然日跟自然小時(shí)都不能計(jì)入本次調(diào)用!

b.計(jì)數(shù)方式的選擇:

目前能想到的會(huì)有:

• mysql+db事務(wù) 持久化、記錄可溯源、實(shí)現(xiàn)起來(lái)比較麻煩,稍微“重”了一點(diǎn)
• redis+lua 實(shí)現(xiàn)簡(jiǎn)單,redis的可執(zhí)行l(wèi)ua腳本的特性也能滿足對(duì)“事務(wù)”的要求
• mysql/redis+分布式事務(wù) 需要上鎖,實(shí)現(xiàn)復(fù)雜,能做到比較精確的計(jì)數(shù),也就是真正等到代碼塊執(zhí)行成功之后,再去操作計(jì)數(shù)

目前沒(méi)有很精確技術(shù)的要求,代價(jià)太大,也沒(méi)有持久化的需求,因此選用 redis+lua 即可

2.調(diào)用方式的實(shí)現(xiàn)

a.常見(jiàn)的做法 先定義一個(gè)通用的入口

//簡(jiǎn)化版代碼

@Component
classDetectManager{
funmatchExceptionally(eventId:String,content:String){
//調(diào)用規(guī)則匹配
valrt=ruleService.match(eventId,content)
if(!rt){
throwBaseException(ErrorCode.OPERATION_TOO_FREQUENT)
}
}
}

在service中調(diào)用該方法

//簡(jiǎn)化版代碼

@Service
classOcrServiceImpl:OcrService{

@Autowired
privatelateinitvardetectManager:DetectManager

/**
*提交ocr任務(wù)
*需要根據(jù)用戶id來(lái)做次數(shù)限制
*/
overridefunsubmitOcrTask(userId:String,imageUrl:String):String{
detectManager.matchExceptionally("ocr",userId)
//doocr
}

}

有沒(méi)有更優(yōu)雅一點(diǎn)的方法呢? 用注解可能會(huì)更好一點(diǎn)(也比較有爭(zhēng)議其實(shí),這邊先支持實(shí)現(xiàn))

由于傳入的 content 是跟業(yè)務(wù)關(guān)聯(lián)的,所以需要通過(guò)Spel來(lái)將參數(shù)構(gòu)成對(duì)應(yīng)的content

基于 Spring Cloud Alibaba + Gateway + Nacos + RocketMQ + Vue & Element 實(shí)現(xiàn)的后臺(tái)管理系統(tǒng) + 用戶小程序，支持 RBAC 動(dòng)態(tài)權(quán)限、多租戶、數(shù)據(jù)權(quán)限、工作流、三方登錄、支付、短信、商城等功能

• 項(xiàng)目地址：https://github.com/YunaiV/yudao-cloud
• 視頻教程：https://doc.iocoder.cn/video/

三、具體實(shí)現(xiàn)

1.風(fēng)控計(jì)數(shù)規(guī)則實(shí)現(xiàn)

a.自然日/自然小時(shí)

自然日/自然小時(shí)可以共用一套lua腳本,因?yàn)樗鼈冎挥?code style="font-size:14px;padding:2px 4px;margin-right:2px;margin-left:2px;color:rgb(30,107,184);background-color:rgba(27,31,35,.05);font-family:'Operator Mono', Consolas, Monaco, Menlo, monospace;">key不同,腳本如下:

//lua腳本
localcurrentValue=redis.call('get',KEYS[1]);
ifcurrentValue~=falsethen
iftonumber(currentValue)tonumber(ARGV[1])then
returnredis.call('INCR',KEYS[1]);
else
returntonumber(currentValue)+1;
end;
else
redis.call('set',KEYS[1],1,'px',ARGV[2]);
return1;
end;

其中 KEYS[1] 是日/小時(shí)關(guān)聯(lián)的key,ARGV[1]是上限值,ARGV[2]是過(guò)期時(shí)間,返回值則是當(dāng)前計(jì)數(shù)值+1后的結(jié)果,(如果已經(jīng)達(dá)到上限,則實(shí)際上不會(huì)計(jì)數(shù))

b.自然日+自然小時(shí) 如前文提到的,兩個(gè)的結(jié)合實(shí)際上并不是單純的拼湊,需要處理回退邏輯

//lua腳本
localdayValue=0;
localhourValue=0;
localdayPass=true;
localhourPass=true;
localdayCurrentValue=redis.call('get',KEYS[1]);
ifdayCurrentValue~=falsethen
iftonumber(dayCurrentValue)tonumber(ARGV[1])then
dayValue=redis.call('INCR',KEYS[1]);
else
dayPass=false;
dayValue=tonumber(dayCurrentValue)+1;
end;
else
redis.call('set',KEYS[1],1,'px',ARGV[3]);
dayValue=1;
end;

localhourCurrentValue=redis.call('get',KEYS[2]);
ifhourCurrentValue~=falsethen
iftonumber(hourCurrentValue)tonumber(ARGV[2])then
hourValue=redis.call('INCR',KEYS[2]);
else
hourPass=false;
hourValue=tonumber(hourCurrentValue)+1;
end;
else
redis.call('set',KEYS[2],1,'px',ARGV[4]);
hourValue=1;
end;

if(notdayPass)andhourPassthen
hourValue=redis.call('DECR',KEYS[2]);
end;

ifdayPassand(nothourPass)then
dayValue=redis.call('DECR',KEYS[1]);
end;

localpair={};
pair[1]=dayValue;
pair[2]=hourValue;
returnpair;

其中 KEYS[1] 是天關(guān)聯(lián)生成的key, KEYS[2] 是小時(shí)關(guān)聯(lián)生成的key,ARGV[1]是天的上限值,ARGV[2]是小時(shí)的上限值,ARGV[3]是天的過(guò)期時(shí)間,ARGV[4]是小時(shí)的過(guò)期時(shí)間,返回值同上

這里給的是比較粗糙的寫法,主要需要表達(dá)的就是,進(jìn)行兩個(gè)條件判斷時(shí),有其中一個(gè)不滿足,另一個(gè)都需要進(jìn)行回退.

2.注解的實(shí)現(xiàn)

a.定義一個(gè)@Detect注解

@Retention(AnnotationRetention.RUNTIME)
@Target(AnnotationTarget.FUNCTION,AnnotationTarget.CLASS)
annotationclassDetect(

/**
*事件id
*/
valeventId:String="",

/**
*content的表達(dá)式
*/
valcontentSpel:String=""

)

其中content是需要經(jīng)過(guò)表達(dá)式解析出來(lái)的,所以接受的是個(gè)String

b.定義@Detect注解的處理類

@Aspect
@Component
classDetectHandler{

privatevallogger=LoggerFactory.getLogger(javaClass)

@Autowired
privatelateinitvardetectManager:DetectManager

@Resource(name="detectSpelExpressionParser")
privatelateinitvarspelExpressionParser:SpelExpressionParser

@Bean(name=["detectSpelExpressionParser"])
fundetectSpelExpressionParser():SpelExpressionParser{
returnSpelExpressionParser()
}

@Around(value="@annotation(detect)")
funoperatorAnnotation(joinPoint:ProceedingJoinPoint,detect:Detect):Any?{
if(detect.eventId.isBlank()||detect.contentSpel.isBlank()){
throwillegalArgumentExp("@Detectconfigisnotavailable!")
}
//轉(zhuǎn)換表達(dá)式
valexpression=spelExpressionParser.parseExpression(detect.contentSpel)
valargMap=joinPoint.args.mapIndexed{index,any->
"arg${index+1}"toany
}.toMap()
//構(gòu)建上下文
valcontext=StandardEvaluationContext().apply{
if(argMap.isNotEmpty())this.setVariables(argMap)
}
//拿到結(jié)果
valcontent=expression.getValue(context)

detectManager.matchExceptionally(detect.eventId,content)
returnjoinPoint.proceed()
}
}

需要將參數(shù)放入到上下文中,并起名為arg1、arg2....

四、測(cè)試一下

1.寫法

使用注解之后的寫法:

//簡(jiǎn)化版代碼

@Service
classOcrServiceImpl:OcrService{

@Autowired
privatelateinitvardetectManager:DetectManager

/**
*提交ocr任務(wù)
*需要根據(jù)用戶id來(lái)做次數(shù)限制
*/
@Detect(eventId="ocr",contentSpel="#arg1")
overridefunsubmitOcrTask(userId:String,imageUrl:String):String{
//doocr
}

}

2.Debug看看

• 注解值獲取成功
• 表達(dá)式解析成功

審核編輯 ：李倩