gfwleak/enderbyendera-realtime-protection
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1、AlertMessage删除is_distribute属性

Browse Source 
2、修改为total_num
3、StateEnum新增getStateEnumByNum方法
This commit is contained in:
Hao Miao
2024-01-22 15:05:15 +08:00
parent 63e7270c92
commit 193862ff3e
7 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/realtime/protection/configuration/entity/rule/dynamicrule/AlertMessage.java
View File

@@ -18,8 +18,8 @@ public class AlertMessage {
@JsonProperty("five_tuple_with_mask")
private FiveTupleWithMask fiveTupleWithMask;
@JsonProperty("is_distribute")
private Boolean isDistribute;//待删除
// @JsonProperty("is_distribute")
// private Boolean isDistribute;//待删除
@JsonProperty("command_uuid")
private String commandUUID;

4
src/main/java/com/realtime/protection/configuration/utils/enums/StateEnum.java
View File

@@ -49,4 +49,8 @@ public enum StateEnum {
public static StateEnum getStateEnumByState(State state) {
return stateToStateEnumMap.get(state);
}
public static StateEnum getStateEnumByNum(Integer stateNum) {
return getStateEnumByState(getStateByNum(stateNum));
}
}

30
src/main/java/com/realtime/protection/server/alertmessage/AlertMessageService.java
View File

@@ -15,6 +15,7 @@ import com.baomidou.dynamic.datasource.annotation.DSTransactional;
import java.util.List;
@Service
public class AlertMessageService {
private final CommandService commandService;
@@ -36,12 +37,12 @@ public class AlertMessageService {
Integer taskType = dynamicTaskCommandInfo.getTaskType();
if (taskType == TaskTypeEnum.DYNAMIC.getTaskType())//动态
switch (taskStatus) {
case 2://running
insertCommandAndAlertMessage(dynamicTaskCommandInfo, true, alertMessage, true);
switch (StateEnum.getStateEnumByNum(taskStatus)) {
case RUNNING:
insertCommandAndAlertMessage(dynamicTaskCommandInfo, true, alertMessage);
break;
case 3://Paused
insertCommandAndAlertMessage(dynamicTaskCommandInfo, false, alertMessage, true);
case PAUSED:
insertCommandAndAlertMessage(dynamicTaskCommandInfo, false, alertMessage);
break;
default://主要是stop
//command不入库
@@ -50,12 +51,12 @@ public class AlertMessageService {
break;
}
else if (taskType == TaskTypeEnum.JUDGED.getTaskType())//研判后
switch (taskStatus) {
case 2://running
insertCommandAndAlertMessage(dynamicTaskCommandInfo, false, alertMessage, false);
switch (StateEnum.getStateEnumByNum(taskStatus)) {
case RUNNING:
insertCommandAndAlertMessage(dynamicTaskCommandInfo, false, alertMessage);
break;
case 3://Paused
insertCommandAndAlertMessage(dynamicTaskCommandInfo, false, alertMessage, false);
case PAUSED:
insertCommandAndAlertMessage(dynamicTaskCommandInfo, false, alertMessage);
break;
default://主要是stop
//command不入库
@@ -88,21 +89,17 @@ public class AlertMessageService {
@DSTransactional
private void insertCommandAndAlertMessage(TaskCommandInfo dynamicTaskCommandInfo, Boolean isValid,
AlertMessage alertMessage, Boolean isDistribute){
AlertMessage alertMessage){
//command入库
dynamicTaskCommandInfo.setIsValid(isValid);
String commandUUID = commandService.createCommand(dynamicTaskCommandInfo);
if (true){
throw new RuntimeException("test");
}
//alertmessage入库
alertMessage.setIsDistribute(isDistribute);
alertMessage.setCommandUUID(commandUUID);
alertMessageMapper.insertAlertMessage(alertMessage);
}
private void insertAlertMessageOnly(AlertMessage alertMessage, Boolean isDistribute){
//alertmessage入库
alertMessage.setIsDistribute(isDistribute);
alertMessage.setCommandUUID(null);
alertMessageMapper.insertAlertMessage(alertMessage);
}
@@ -141,6 +138,7 @@ public class AlertMessageService {
public List<AlertMessage> queryAlarmsByCommandId(String commandId) {
return alertMessageMapper.queryAlermsByCommandId(commandId);
}
}

4
src/main/java/com/realtime/protection/server/rule/dynamicrule/DynamicRuleController.java
View File

@@ -57,7 +57,7 @@ public class DynamicRuleController implements DynamicRuleControllerApi {
@Override
@DeleteMapping("/{ids}")
public ResponseResult deleteDynamicRuleObjects(@PathVariable List<Integer> ids) {
log.info("批量删除动态规则: {}", ids);
//log.info("批量删除动态规则: {}", ids);
//调用service删除
return ResponseResult.ok()
@@ -104,7 +104,7 @@ public class DynamicRuleController implements DynamicRuleControllerApi {
.setData("success", true)
.setData("dynamic_rule_list", dynamicRuleService.queryDynamicRuleObject(dynamicRuleName, dynamicRuleId,
protectObjectSourceSystem, creator, page, pageSize))
.setData("dynamic_rule_total_num",dynamicRuleService.queryDynamicRuleTotalNum());
.setData("total_num",dynamicRuleService.queryDynamicRuleTotalNum());
}
//详情查看？？ 就是按id查询吧

2
src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleController.java
View File

@@ -139,7 +139,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
static_rule_creator, static_rule_ip, page, pageSize);
return ResponseResult.ok()
.setData("static_rule_list", pageResult)
.setData("static_rule_total_num",staticRuleService.queryStaticRuleTotalNum());
.setData("total_num",staticRuleService.queryStaticRuleTotalNum());
}
/**

2
src/main/java/com/realtime/protection/server/whitelist/WhiteListController.java
View File

@@ -70,7 +70,7 @@ public class WhiteListController implements WhiteListControllerApi {
}
return ResponseResult.ok()
.setData("whiteobj_list", whiteListService.queryWhiteListObject(whiteListName, whiteListId, page, pageSize))
.setData("whiteobj_total_num", whiteListService.queryWhiteListTotalNum());
.setData("total_num", whiteListService.queryWhiteListTotalNum());
}
@Override