基于Java編寫一個(gè)簡單的風(fēng)控組件
一、背景
1.為什么要做風(fēng)控
這不得拜產(chǎn)品大佬所賜
目前我們業(yè)務(wù)有使用到非常多的AI能力,如ocr識別、語音測評等,這些能力往往都比較費(fèi)錢或者費(fèi)資源,所以在產(chǎn)品層面也希望我們對用戶的能力使用次數(shù)做一定的限制,因此風(fēng)控是必須的!
2.為什么要自己寫風(fēng)控
那么多開源的風(fēng)控組件,為什么還要寫呢?是不是想重復(fù)發(fā)明輪子呀.
要想回答這個(gè)問題,需要先解釋下我們業(yè)務(wù)需要用到的風(fēng)控(簡稱業(yè)務(wù)風(fēng)控),與開源常見的風(fēng)控(簡稱普通風(fēng)控)有何區(qū)別:
| 風(fēng)控類型 | 目的 | 對象 | 規(guī)則 |
|---|---|---|---|
| 業(yè)務(wù)風(fēng)控 | 實(shí)現(xiàn)產(chǎn)品定義的一些限制,達(dá)到限制時(shí),有具體的業(yè)務(wù)流程,如充值vip等 | 比較復(fù)雜多變的,例如針對用戶進(jìn)行風(fēng)控,也能針對用戶+年級進(jìn)行風(fēng)控 | 自然日、自然小時(shí)等 |
| 普通風(fēng)控 | 保護(hù)服務(wù)或數(shù)據(jù),攔截異常請求等 | 接口、部分可以加上簡單參數(shù) | 一般用得更多的是滑動窗口 |
因此,直接使用開源的普通風(fēng)控,一般情況下是無法滿足需求的
3.其它要求
支持實(shí)時(shí)調(diào)整限制
很多限制值在首次設(shè)置的時(shí)候,基本上都是拍定的一個(gè)值,后續(xù)需要調(diào)整的可能性是比較大的,因此可調(diào)整并實(shí)時(shí)生效是必須的
二、思路
要實(shí)現(xiàn)一個(gè)簡單的業(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)槿绻匀蝗盏呐卸ㄍㄟ^,而自然小時(shí)的判定不通過的時(shí)候,需要回退,自然日跟自然小時(shí)都不能計(jì)入本次調(diào)用!
b.計(jì)數(shù)方式的選擇
目前能想到的會有:
mysql+db事務(wù)
- 持久化、記錄可溯源、實(shí)現(xiàn)起來比較麻煩,稍微“重”了一點(diǎn)
- redis+lua
- 實(shí)現(xiàn)簡單,redis的可執(zhí)行l(wèi)ua腳本的特性也能滿足對“事務(wù)”的要求
- mysql/redis+分布式事務(wù)
- 需要上鎖,實(shí)現(xiàn)復(fù)雜,能做到比較精確的計(jì)數(shù),也就是真正等到代碼塊執(zhí)行成功之后,再去操作計(jì)數(shù)
目前沒有很精確技術(shù)的要求,代價(jià)太大,也沒有持久化的需求,因此選用 redis+lua 即可
2.調(diào)用方式的實(shí)現(xiàn)
a.常見的做法
先定義一個(gè)通用的入口
//簡化版代碼
@Component
class DetectManager {
fun matchExceptionally(eventId: String, content: String){
//調(diào)用規(guī)則匹配
val rt = ruleService.match(eventId,content)
if (!rt) {
throw BaseException(ErrorCode.OPERATION_TOO_FREQUENT)
}
}
}在service中調(diào)用該方法
//簡化版代碼
@Service
class OcrServiceImpl : OcrService {
@Autowired
private lateinit var detectManager: DetectManager
/**
* 提交ocr任務(wù)
* 需要根據(jù)用戶id來做次數(shù)限制
*/
override fun submitOcrTask(userId: String, imageUrl: String): String {
detectManager.matchExceptionally("ocr", userId)
//do ocr
}
}有沒有更優(yōu)雅一點(diǎn)的方法呢? 用注解可能會更好一點(diǎn)(也比較有爭議其實(shí),這邊先支持實(shí)現(xiàn))
由于傳入的 content 是跟業(yè)務(wù)關(guān)聯(lián)的,所以需要通過Spel來將參數(shù)構(gòu)成對應(yīng)的content
三、具體實(shí)現(xiàn)
1.風(fēng)控計(jì)數(shù)規(guī)則實(shí)現(xiàn)
a.自然日/自然小時(shí)
自然日/自然小時(shí)可以共用一套lua腳本,因?yàn)樗鼈冎挥?code>key不同,腳本如下:
//lua腳本
local currentValue = redis.call('get', KEYS[1]);
if currentValue ~= false then
if tonumber(currentValue) < tonumber(ARGV[1]) then
return redis.call('INCR', KEYS[1]);
else
return tonumber(currentValue) + 1;
end;
else
redis.call('set', KEYS[1], 1, 'px', ARGV[2]);
return 1;
end;其中 KEYS[1] 是日/小時(shí)關(guān)聯(lián)的key,ARGV[1]是上限值,ARGV[2]是過期時(shí)間,返回值則是當(dāng)前計(jì)數(shù)值+1后的結(jié)果,(如果已經(jīng)達(dá)到上限,則實(shí)際上不會計(jì)數(shù))
b.自然日+自然小時(shí)
如前文提到的,兩個(gè)的結(jié)合實(shí)際上并不是單純的拼湊,需要處理回退邏輯
//lua腳本
local dayValue = 0;
local hourValue = 0;
local dayPass = true;
local hourPass = true;
local dayCurrentValue = redis.call('get', KEYS[1]);
if dayCurrentValue ~= false then
if tonumber(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;
local hourCurrentValue = redis.call('get', KEYS[2]);
if hourCurrentValue ~= false then
if tonumber(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 (not dayPass) and hourPass then
hourValue = redis.call('DECR', KEYS[2]);
end;
if dayPass and (not hourPass) then
dayValue = redis.call('DECR', KEYS[1]);
end;
local pair = {};
pair[1] = dayValue;
pair[2] = hourValue;
return pair;其中 KEYS[1] 是天關(guān)聯(lián)生成的key, KEYS[2] 是小時(shí)關(guān)聯(lián)生成的key,ARGV[1]是天的上限值,ARGV[2]是小時(shí)的上限值,ARGV[3]是天的過期時(shí)間,ARGV[4]是小時(shí)的過期時(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)
annotation class Detect(
/**
* 事件id
*/
val eventId: String = "",
/**
* content的表達(dá)式
*/
val contentSpel: String = ""
)其中content是需要經(jīng)過表達(dá)式解析出來的,所以接受的是個(gè)String
b.定義@Detect注解的處理類
@Aspect
@Component
class DetectHandler {
private val logger = LoggerFactory.getLogger(javaClass)
@Autowired
private lateinit var detectManager: DetectManager
@Resource(name = "detectSpelExpressionParser")
private lateinit var spelExpressionParser: SpelExpressionParser
@Bean(name = ["detectSpelExpressionParser"])
fun detectSpelExpressionParser(): SpelExpressionParser {
return SpelExpressionParser()
}
@Around(value = "@annotation(detect)")
fun operatorAnnotation(joinPoint: ProceedingJoinPoint, detect: Detect): Any? {
if (detect.eventId.isBlank() || detect.contentSpel.isBlank()){
throw illegalArgumentExp("@Detect config is not available!")
}
//轉(zhuǎn)換表達(dá)式
val expression = spelExpressionParser.parseExpression(detect.contentSpel)
val argMap = joinPoint.args.mapIndexed { index, any ->
"arg${index+1}" to any
}.toMap()
//構(gòu)建上下文
val context = StandardEvaluationContext().apply {
if (argMap.isNotEmpty()) this.setVariables(argMap)
}
//拿到結(jié)果
val content = expression.getValue(context)
detectManager.matchExceptionally(detect.eventId, content)
return joinPoint.proceed()
}
}需要將參數(shù)放入到上下文中,并起名為arg1、arg2....
四、測試一下
1.寫法
使用注解之后的寫法:
//簡化版代碼
@Service
class OcrServiceImpl : OcrService {
@Autowired
private lateinit var detectManager: DetectManager
/**
* 提交ocr任務(wù)
* 需要根據(jù)用戶id來做次數(shù)限制
*/
@Detect(eventId = "ocr", contentSpel = "#arg1")
override fun submitOcrTask(userId: String, imageUrl: String): String {
//do ocr
}
}2.Debug看看
- 注解值獲取成功
- 表達(dá)式解析成功
以上就是基于Java編寫一個(gè)簡單的風(fēng)控組件的詳細(xì)內(nèi)容，更多關(guān)于Java風(fēng)控組件的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
這篇文章主要為大家介紹了七段小代碼解決Java程序常見的崩潰場景，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-06-06
本文主要介紹了如何使用MyBatis快速生成代碼的幾種方法,文中通過示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2023-12-12
springBoot+mybaties后端多層架構(gòu)的實(shí)現(xiàn)示例
本文主要介紹了springBoot+mybaties后端多層架構(gòu)的實(shí)現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07
java操作mongodb之多表聯(lián)查的實(shí)現(xiàn)($lookup)
這篇文章主要介紹了java操作mongodb之多表聯(lián)查的實(shí)現(xiàn)($lookup)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
這篇文章主要介紹了java Random.nextInt()方法的具體使用，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
java實(shí)現(xiàn)在線預(yù)覽--poi實(shí)現(xiàn)word、excel、ppt轉(zhuǎn)html的方法
這篇文章主要介紹了java實(shí)現(xiàn)在線預(yù)覽--poi實(shí)現(xiàn)word、excel、ppt轉(zhuǎn)html的方法，本文需要引入poi的jar包給大家介紹的非常詳細(xì),需要的朋友可以參考下2019-09-09
關(guān)于springboot集成swagger3時(shí)spring-plugin-core報(bào)錯(cuò)的問題
這篇文章主要介紹了關(guān)于springboot集成swagger3時(shí)spring-plugin-core報(bào)錯(cuò)的問題,本文給大家分享解決方法，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-09-09
redisson特性及優(yōu)雅實(shí)現(xiàn)示例
這篇文章主要為大家介紹了redisson特性及優(yōu)雅實(shí)現(xiàn)示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-11-11
Fluent Mybatis,原生Mybatis,Mybatis Plus三者功能對比
本文主要介紹了Fluent Mybatis,原生Mybatis,Mybatis Plus三者功能對比，分享給大家，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-08-08
這篇文章主要為大家詳細(xì)介紹了Java實(shí)現(xiàn)考試系統(tǒng)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-09-09