风控系统设计

一、思路

要实现一个简单的业务风控组件,要做什么工作呢?

1.风控规则的实现

a.需要实现的规则:

自然日计数
自然小时计数
自然日+自然小时计数

自然日+自然小时计数这里并不能单纯地串联两个判断,因为如果自然日的判定通过,而自然小时的判定不通过的时候,需要回退,自然日跟自然小时都不能计入本次调用!

b.计数方式的选择:

目前能想到的会有:

mysql+db事务持久化、记录可溯源、实现起来比较麻烦,稍微“重”了一点
redis+lua 实现简单,redis的可执行lua脚本的特性也能满足对“事务”的要求
mysql/redis+分布式事务需要上锁,实现复杂,能做到比较精确的计数,也就是真正等到代码块执行成功之后,再去操作计数

目前没有很精确技术的要求,代价太大,也没有持久化的需求,因此选用 redis+lua 即可

2.调用方式的实现

a.常见的做法 先定义一个通用的入口

//简化版代码@Component
class DetectManager {fun matchExceptionally(eventId: String, content: String){//调用规则匹配val rt = ruleService.match(eventId,content)if (!rt) {throw BaseException(ErrorCode.OPERATION_TOO_FREQUENT)}}
}

在service中调用该方法

//简化版代码@Service
class OcrServiceImpl : OcrService {@Autowiredprivate lateinit var detectManager: DetectManager/*** 提交ocr任务* 需要根据用户id来做次数限制*/override fun submitOcrTask(userId: String, imageUrl: String): String {detectManager.matchExceptionally("ocr", userId)//do ocr}}

有没有更优雅一点的方法呢? 用注解可能会更好一点(也比较有争议其实,这边先支持实现)

由于传入的 content 是跟业务关联的,所以需要通过Spel来将参数构成对应的content

二、具体实现

1.风控计数规则实现

a.自然日/自然小时

自然日/自然小时可以共用一套lua脚本,因为它们只有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]);elsereturn tonumber(currentValue) + 1;end;
elseredis.call('set', KEYS[1], 1, 'px', ARGV[2]);return 1;
end;

其中 KEYS[1] 是日/小时关联的key,ARGV[1]是上限值,ARGV[2]是过期时间,返回值则是当前计数值+1后的结果,(如果已经达到上限,则实际上不会计数)

b.自然日+自然小时 如前文提到的,两个的结合实际上并不是单纯的拼凑,需要处理回退逻辑

//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]);elsedayPass = false;dayValue = tonumber(dayCurrentValue) + 1;end;
elseredis.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]);elsehourPass = false;hourValue = tonumber(hourCurrentValue) + 1;end;
elseredis.call('set', KEYS[2], 1, 'px', ARGV[4]);hourValue = 1;
end;if (not dayPass) and hourPass thenhourValue = redis.call('DECR', KEYS[2]);
end;if dayPass and (not hourPass) thendayValue = redis.call('DECR', KEYS[1]);
end;local pair = {};
pair[1] = dayValue;
pair[2] = hourValue;
return pair;

其中 KEYS[1] 是天关联生成的key, KEYS[2] 是小时关联生成的key,ARGV[1]是天的上限值,ARGV[2]是小时的上限值,ARGV[3]是天的过期时间,ARGV[4]是小时的过期时间,返回值同上

这里给的是比较粗糙的写法,主要需要表达的就是,进行两个条件判断时,有其中一个不满足,另一个都需要进行回退.

2.注解的实现

a.定义一个@Detect注解

@Retention(AnnotationRetention.RUNTIME)
@Target(AnnotationTarget.FUNCTION, AnnotationTarget.CLASS)
annotation class Detect(/*** 事件id*/val eventId: String = "",/*** content的表达式*/val contentSpel: String = "")

其中content是需要经过表达式解析出来的,所以接受的是个String

b.定义@Detect注解的处理类

@Aspect
@Component
class DetectHandler {private val logger = LoggerFactory.getLogger(javaClass)@Autowiredprivate 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!")}//转换表达式val expression = spelExpressionParser.parseExpression(detect.contentSpel)val argMap = joinPoint.args.mapIndexed { index, any ->"arg${index+1}" to any}.toMap()//构建上下文val context = StandardEvaluationContext().apply {if (argMap.isNotEmpty()) this.setVariables(argMap)}//拿到结果val content = expression.getValue(context)detectManager.matchExceptionally(detect.eventId, content)return joinPoint.proceed()}
}

需要将参数放入到上下文中,并起名为arg1、arg2....

三、测试一下

1.写法

使用注解之后的写法:

//简化版代码@Service
class OcrServiceImpl : OcrService {@Autowiredprivate lateinit var detectManager: DetectManager/*** 提交ocr任务* 需要根据用户id来做次数限制*/@Detect(eventId = "ocr", contentSpel = "#arg1")override fun submitOcrTask(userId: String, imageUrl: String): String {//do ocr}}