diff --git a/src/main/java/com/realtime/protection/configuration/entity/defense/template/ProtectLevel.java b/src/main/java/com/realtime/protection/configuration/entity/defense/template/ProtectLevel.java index 8e954b0..b54ee71 100644 --- a/src/main/java/com/realtime/protection/configuration/entity/defense/template/ProtectLevel.java +++ b/src/main/java/com/realtime/protection/configuration/entity/defense/template/ProtectLevel.java @@ -31,4 +31,7 @@ public class ProtectLevel { @Schema(description = "该防护等级是处置防护对象的全流量or单向流量") private Boolean isFullFlow = false; + + @Schema(description = "该防护等级防护对象ip地址是否作为源ip地址") + private Boolean isProtectObjectIPSrc = false; } diff --git a/src/main/java/com/realtime/protection/server/defense/template/TemplateController.java b/src/main/java/com/realtime/protection/server/defense/template/TemplateController.java index ff6ff38..437bfa7 100644 --- a/src/main/java/com/realtime/protection/server/defense/template/TemplateController.java +++ b/src/main/java/com/realtime/protection/server/defense/template/TemplateController.java @@ -174,6 +174,9 @@ public class TemplateController implements TemplateControllerApi { .setData("success", templateService.updateAuditStatusBatch(idsWithAuditStatusMap)); } + /* + 修改审批信息 + */ @Override @PostMapping("/auditInfo/{ids}") public ResponseResult updateAuditInfo(@PathVariable List ids, diff --git a/src/main/java/com/realtime/protection/server/defense/template/TemplateControllerApi.java b/src/main/java/com/realtime/protection/server/defense/template/TemplateControllerApi.java index a1f2243..0f8dcad 100644 --- a/src/main/java/com/realtime/protection/server/defense/template/TemplateControllerApi.java +++ b/src/main/java/com/realtime/protection/server/defense/template/TemplateControllerApi.java @@ -48,7 +48,10 @@ public interface TemplateControllerApi { ) ) }, - requestBody = @io.swagger.v3.oas.annotations.parameters.RequestBody(description = "防御策略模板信息") + requestBody = @io.swagger.v3.oas.annotations.parameters.RequestBody( + + description = "防御策略模板信息") + ) ResponseResult newTemplate(@RequestBody @Valid Template template); diff --git a/src/main/resources/mappers/TemplateMapper.xml b/src/main/resources/mappers/TemplateMapper.xml index 914aa4a..ad7dc8c 100644 --- a/src/main/resources/mappers/TemplateMapper.xml +++ b/src/main/resources/mappers/TemplateMapper.xml @@ -20,10 +20,10 @@ INSERT INTO t_protect_level(has_protect_object_ip, has_protect_object_port, has_protocol, has_url, has_dns, - has_peer_ip, has_peer_port, is_full_flow) + has_peer_ip, has_peer_port, is_full_flow, is_protect_object_dst) VALUE (#{level.hasProtectObjectIP}, #{level.hasProtectObjectPort}, #{level.hasProtocol}, #{level.hasURL}, #{level.hasDNS}, - #{level.hasPeerIP}, #{level.hasPeerPort}, #{level.isFullFlow}) + #{level.hasPeerIP}, #{level.hasPeerPort}, #{level.isFullFlow}, #{level.isProtectObjectDst}) @@ -51,6 +51,7 @@ + + + @@ -91,6 +94,7 @@ +