1、whitelist查询返回审核状态字段,task查询返回动作字段

2、解决alertmessageTest出错
This commit is contained in:
Hao Miao
2024-01-24 14:06:49 +08:00
parent 7cb8666867
commit 02dd20743f
5 changed files with 7 additions and 4 deletions

View File

@@ -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(),

View File

@@ -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,

View File

@@ -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"/>

View File

@@ -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">

View File

@@ -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++) {