1、whitelist查询返回审核状态字段,task查询返回动作字段
2、解决alertmessageTest出错
This commit is contained in:
@@ -70,7 +70,9 @@ public class AlertMessageService {
|
||||
// (1)查询生成指令所需信息:和alertMessage中的fiveTuple信息 合并成 TaskCommandInfo;
|
||||
// (2)额外信息:并额外查询templateId、protectLevel和taskStatus
|
||||
TaskCommandInfo dynamicCommandInfo = alertMessageMapper.getDynamicTaskInfos(taskId, DynamicRuleId);
|
||||
|
||||
if (dynamicCommandInfo.getTemplateId() == null){
|
||||
throw new IllegalArgumentException("taskId: " + taskId + " DynamicRuleId: " + DynamicRuleId + " 不匹配");
|
||||
}
|
||||
// 根据templateId、protectLevel获取策略模板
|
||||
ProtectLevel templateProtectLevel = alertMessageMapper.queryTemplateProtectLevel(
|
||||
dynamicCommandInfo.getTemplateId(),
|
||||
|
||||
@@ -39,7 +39,6 @@
|
||||
<result column="TASK_ID" property="taskId"/>
|
||||
<result column="DYNAMIC_RULE_ID" property="dynamicRuleId"/>
|
||||
|
||||
<result column="IS_DISTRIBUTE" property="isDistribute"/>
|
||||
<result column="COMMAND_UUID" property="commandUUID"/>
|
||||
|
||||
<result column="CREATE_TIME" property="createTime"/>
|
||||
@@ -173,7 +172,6 @@
|
||||
t_alertmessage.MASK_DST_PORT,
|
||||
t_alertmessage.MASK_PROTOCOL,
|
||||
|
||||
t_alertmessage.IS_DISTRIBUTE,
|
||||
t_alertmessage.COMMAND_UUID,
|
||||
|
||||
t_alertmessage.CREATE_TIME,
|
||||
|
||||
@@ -43,6 +43,7 @@
|
||||
<id column="task_id" property="taskId"/>
|
||||
<result column="task_name" property="taskName"/>
|
||||
<result column="task_type" property="taskType"/>
|
||||
<result column="task_act" property="taskAct"/>
|
||||
|
||||
<result column="task_status" property="taskStatus"/>
|
||||
<result column="task_audit_status" property="taskAuditStatus"/>
|
||||
|
||||
@@ -46,7 +46,7 @@
|
||||
<result column="white_list_port" property="whiteListPort"/>
|
||||
<result column="white_list_url" property="whiteListUrl"/>
|
||||
<result column="white_list_protocol" property="whiteListProtocol"/>
|
||||
<!-- <result column="white_list_audit_status" property="whiteListAuditStatus"/>-->
|
||||
<result column="white_list_audit_status" property="whiteListAuditStatus"/>
|
||||
</resultMap>
|
||||
|
||||
<select id="queryWhiteListObject" resultMap="whiteListMap">
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
package com.realtime.protection.server.alertmessage;
|
||||
|
||||
import com.github.xiaoymin.knife4j.annotations.Ignore;
|
||||
import com.realtime.protection.configuration.entity.rule.dynamicrule.AlertMessage;
|
||||
import com.realtime.protection.configuration.entity.task.FiveTupleWithMask;
|
||||
import org.junit.jupiter.api.Test;
|
||||
@@ -15,6 +16,7 @@ public class AlertMessageTest {
|
||||
this.alertMessageService = alertMessageService;
|
||||
}
|
||||
|
||||
@Ignore
|
||||
@Test
|
||||
void testReceiveAlertMessage() {
|
||||
for (int i = 1; i < 4; i++) {
|
||||
|
||||
Reference in New Issue
Block a user