基于Java编写一个简单的风控组件
一、背景
1.为什么要做风控
这不得拜产品大佬所赐
目前我们业务有使用到非常多的AI能力,如ocr识别、语音测评等,这些能力往往都比较费钱或者费资源,所以在产品层面也希望我们对用户的能力使用次数做一定的限制,因此风控是必须的!
2.为什么要自己写风控
那么多开源的风控组件,为什么还要写呢?是不是想重复发明轮子呀.
要想回答这个问题,需要先解释下我们业务需要用到的风控(简称业务风控),与开源常见的风控(简称普通风控)有何区别:
风控类型 | 目的 | 对象 | 规则 |
---|---|---|---|
业务风控 | 实现产品定义的一些限制,达到限制时,有具体的业务流程,如充值vip等 | 比较复杂多变的,例如针对用户进行风控,也能针对用户+年级进行风控 | 自然日、自然小时等 |
普通风控 | 保护服务或数据,拦截异常请求等 | 接口、部分可以加上简单参数 | 一般用得更多的是滑动窗口 |
因此,直接使用开源的普通风控,一般情况下是无法满足需求的
3.其它要求
支持实时调整限制
很多限制值在首次设置的时候,基本上都是拍定的一个值,后续需要调整的可能性是比较大的,因此可调整并实时生效是必须的
二、思路
要实现一个简单的业务风控组件,要做什么工作呢?
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 { @Autowired private 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]); else return tonumber(currentValue) + 1; end; else redis.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]); 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]
是天关联生成的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) @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!") } //转换表达式 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 { @Autowired private lateinit var detectManager: DetectManager /** * 提交ocr任务 * 需要根据用户id来做次数限制 */ @Detect(eventId = "ocr", contentSpel = "#arg1") override fun submitOcrTask(userId: String, imageUrl: String): String { //do ocr } }
2.Debug看看
- 注解值获取成功
- 表达式解析成功
以上就是基于Java编写一个简单的风控组件的详细内容,更多关于Java风控组件的资料请关注猪先飞其它相关文章!
原文出处:https://juejin.cn/post/7182774381448282172
相关文章
- 这篇文章主要介绍了如何利用java语言实现经典《复杂迷宫》游戏,文中采用了swing技术进行了界面化处理,感兴趣的小伙伴可以动手试一试...2022-02-01
java 运行报错has been compiled by a more recent version of the Java Runtime
java 运行报错has been compiled by a more recent version of the Java Runtime (class file version 54.0)...2021-04-01- 这篇文章主要介绍了在java中获取List集合中最大的日期时间操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-08-15
- 这篇文章主要介绍了Vue组件跨层级获取组件操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-28
- 这篇文章主要介绍了教你怎么用Java获取国家法定节假日,文中有非常详细的代码示例,对正在学习java的小伙伴们有非常好的帮助,需要的朋友可以参考下...2021-04-23
- 这篇文章主要介绍了Java如何发起http请求的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-03-31
- 说起C#和Java这两门语言(语法,数据类型 等),个人以为,大概有90%以上的相似,甚至可以认为几乎一样。但是在工作中,我也发现了一些细微的差别...2020-06-25
- 这篇文章主要介绍了解决Java处理HTTP请求超时的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-03-29
- 这篇文章主要介绍了java 判断两个时间段是否重叠的案例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-08-15
- 这篇文章主要介绍了超简洁java实现双色球若干注随机号码生成(实例代码),本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-04-02
- 今天我们来给大家介绍下在Vue开发中我们经常会碰到的一种需求场景,本文主要介绍了Vue动态查询规则生成组件,需要的朋友们下面随着小编来一起学习学习吧...2021-05-27
- 这篇文章主要介绍了Java生成随机姓名、性别和年龄的实现示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-10-01
java 画pdf用itext调整表格宽度、自定义各个列宽的方法
这篇文章主要介绍了java 画pdf用itext调整表格宽度、自定义各个列宽的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-01-31- 这篇文章主要介绍了js组件SlotMachine实现图片切换效果制作抽奖系统的相关资料,需要的朋友可以参考下...2016-04-19
- 这篇文章主要介绍了java正则表达式判断前端参数修改表中另一个字段的值,需要的朋友可以参考下...2021-05-07
Java使用ScriptEngine动态执行代码(附Java几种动态执行代码比较)
这篇文章主要介绍了Java使用ScriptEngine动态执行代码,并且分享Java几种动态执行代码比较,需要的朋友可以参考下...2021-04-15- 这篇文章主要介绍了Java开发实现人机猜拳游戏,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2020-08-03
Java 8 Stream 的终极技巧——Collectors 功能与操作方法详解
这篇文章主要介绍了Java 8 Stream Collectors 功能与操作方法,结合实例形式详细分析了Java 8 Stream Collectors 功能、操作方法及相关注意事项,需要的朋友可以参考下...2020-05-20- 这篇文章主要介绍了Java List集合返回值去掉中括号('[ ]')的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-08-29
- 这篇文章主要介绍了vue中如何使用element的日历组件,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-09-30