1、重构templateNew,未完成
This commit is contained in:
Hao Miao
@@ -0,0 +1,110 @@
|
||||
package com.realtime.protection.configuration.entity.defense.template;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
import io.swagger.v3.oas.annotations.media.Schema;
|
||||
import jakarta.validation.constraints.NotNull;
|
||||
import lombok.Data;
|
||||
|
||||
@Data
|
||||
public class TemplateNew {
|
||||
@JsonProperty("template_id")
|
||||
@Schema(description = "防御策略模板ID", example = "2", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private Integer templateId;
|
||||
|
||||
@JsonProperty("template_name")
|
||||
@NotNull(message = "template_name字段不能为空。")
|
||||
@Schema(description = "防御策略模板名称(事件类型)", example = "APT攻击事件")
|
||||
private String templateName;
|
||||
|
||||
@JsonProperty("source_system")
|
||||
@NotNull(message = "source_system字段不能为空。")
|
||||
@Schema(description = "防御策略模板数据来源系统", example = "BW系统")
|
||||
private String sourceSystem;
|
||||
|
||||
@JsonProperty("event_type")
|
||||
@NotNull(message = "事件类型字段不能为空。")
|
||||
@Schema(description = "防御策略模板数据来源系统", example = "BW系统")
|
||||
private String eventType;
|
||||
|
||||
@JsonProperty("protect_level")
|
||||
@NotNull(message = "防护等级字段不能为空。")
|
||||
@Schema(description = "防御策略模板数据来源系统", example = "BW系统")
|
||||
private String protectLevel;
|
||||
|
||||
@JsonProperty("description")
|
||||
@Schema(description = "对策略模板的文字描述。方便用户使用", example = "zd防护对象的全流量")
|
||||
private String description;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取防护对象IP地址字段")
|
||||
private Boolean hasProtectObjectIP = false;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取防护对象端口字段")
|
||||
private Boolean hasProtectObjectPort = false;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取对端IP地址字段")
|
||||
private Boolean hasPeerIP = false;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取对端端口字段")
|
||||
private Boolean hasPeerPort = false;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取网络协议字段")
|
||||
private Boolean hasProtocol = false;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取URL字段")
|
||||
private Boolean hasURL = false;
|
||||
|
||||
@Schema(description = "该防护等级是否需要提取DNS")
|
||||
private Boolean hasDNS = false;
|
||||
|
||||
@Schema(description = "该防护等级是处置防护对象的全流量or单向流量")
|
||||
private Boolean isFullFlow = false;
|
||||
|
||||
@Schema(description = "该防护等级防护对象ip地址是否作为源ip地址")
|
||||
private Boolean isProtectObjectIPSrc = false;
|
||||
|
||||
@JsonProperty("template_used_times")
|
||||
@Schema(description = "防御策略模板使用次数", example = "20", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private Integer usedTimes;
|
||||
|
||||
@JsonProperty("running_tasks")
|
||||
@Schema(description = "防御策略模板已运行的任务数量", example = "30", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private Integer runningTasks;
|
||||
|
||||
@JsonProperty("create_user_id")
|
||||
@Schema(description = "防御策略模板创建人ID", example = "1", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private Integer createUserId;
|
||||
|
||||
@JsonProperty("create_user_name")
|
||||
@Schema(description = "防御策略模板创建人名称", example = "xxx", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private String createUsername;
|
||||
|
||||
@JsonProperty("create_user_depart")
|
||||
@Schema(description = "防御策略模板创建人处室", example = "xxx", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private String createDepart;
|
||||
|
||||
@JsonProperty("audit_status")
|
||||
@Schema(description = "防御策略模板审核状态（0为未审核，1为已退回，2为审核通过）", example = "1", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private String auditStatus;
|
||||
|
||||
@JsonProperty("template_display_id")
|
||||
@Schema(description = "策略模板展示id", accessMode = Schema.AccessMode.READ_ONLY)
|
||||
private String templateDisplayId;
|
||||
|
||||
@JsonProperty("audit_user_name")
|
||||
@Schema(description = "审核用户名称", example = "user11")
|
||||
private String auditUserName;
|
||||
@JsonProperty("audit_user_id")
|
||||
@Schema(description = "审核用户id", example = "11111")
|
||||
private String auditUserId;
|
||||
@JsonProperty("audit_user_depart")
|
||||
@Schema(description = "审核用户部门", example = "部门1")
|
||||
private String auditUserDepart;
|
||||
|
||||
@JsonProperty("create_time")
|
||||
@Schema(description = "新建时间")
|
||||
private String createTime;
|
||||
|
||||
@JsonProperty("update_time")
|
||||
@Schema(description = "更新时间")
|
||||
private String updateTime;
|
||||
}
|
||||
@@ -0,0 +1,208 @@
|
||||
package com.realtime.protection.server.defense.templatenew;
|
||||
|
||||
import com.realtime.protection.configuration.entity.defense.template.TemplateNew;
|
||||
import com.realtime.protection.configuration.response.ResponseResult;
|
||||
import com.realtime.protection.configuration.utils.enums.audit.AuditStatusEnum;
|
||||
import jakarta.validation.Valid;
|
||||
import jakarta.validation.constraints.Min;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
@RestController
|
||||
@RequestMapping("/templatenew/")
|
||||
public class TemplateController implements TemplateNewCpntrollerApi{
|
||||
private final TemplateService templateService;
|
||||
|
||||
public TemplateController(TemplateService templateService) {
|
||||
this.templateService = templateService;
|
||||
}
|
||||
|
||||
|
||||
// @Override
|
||||
@PostMapping("/new")
|
||||
public ResponseResult newTemplate(@RequestBody @Valid TemplateNew template) {
|
||||
|
||||
Integer templateId = templateService.newTemplate(template);
|
||||
|
||||
if (templateId > 0) {
|
||||
return ResponseResult.ok()
|
||||
.setData("template_id", templateId)
|
||||
.setData("success", true);
|
||||
}
|
||||
|
||||
return ResponseResult.error()
|
||||
.setData("template_id", null)
|
||||
.setData("success", false);
|
||||
}
|
||||
|
||||
/*
|
||||
删 改 查 审核。。。
|
||||
*/
|
||||
|
||||
@Override
|
||||
@GetMapping("/query")
|
||||
public ResponseResult queryTemplates(@RequestParam(value = "template_name", required = false) String templateName,
|
||||
@RequestParam(value = "source_system", required = false) String sourceSystem,
|
||||
@RequestParam(value = "create_user_name", required = false) String createUserName,
|
||||
@RequestParam(value = "template_id", required = false) Integer templateId,
|
||||
@RequestParam(value = "audit_status", required = false) Integer auditStatus,
|
||||
@RequestParam(value = "event_type", required = false) String eventType,
|
||||
@RequestParam(value = "protect_level", required = false) Integer protectLevel,
|
||||
|
||||
@RequestParam("page") @Min(1) Integer page,
|
||||
@RequestParam("page_size") @Min(1) Integer pageSize) {
|
||||
List<TemplateNew> templates = templateService.queryTemplates(
|
||||
templateName, sourceSystem, createUserName, templateId,
|
||||
auditStatus,eventType,protectLevel,
|
||||
page, pageSize
|
||||
);
|
||||
|
||||
return ResponseResult.ok()
|
||||
.setData("templates", templates)
|
||||
.setData("total_num", templateService.queryTemplateTotalNum(
|
||||
templateName, sourceSystem, createUserName, templateId,
|
||||
auditStatus, eventType, protectLevel));
|
||||
}
|
||||
|
||||
@Override
|
||||
@GetMapping("/{templateId}/query")
|
||||
public ResponseResult queryTemplate(@PathVariable Integer templateId) {
|
||||
TemplateNew template = templateService.queryTemplate(templateId);
|
||||
if (template == null) {
|
||||
return ResponseResult.invalid()
|
||||
.setMessage("无效的策略模板ID，也许该模板不存在？");
|
||||
}
|
||||
return ResponseResult.ok().setData("template", template);
|
||||
}
|
||||
|
||||
// @Override
|
||||
// @GetMapping("/query/level")
|
||||
// public ResponseResult queryTemplateProtectLevels() {
|
||||
// List<Template> templates = templateService.queryTemplatesShort();
|
||||
// return ResponseResult.ok().setData("templates", templates);
|
||||
// }
|
||||
|
||||
@Override
|
||||
@PostMapping("/{templateId}/update")
|
||||
public ResponseResult updateTemplate(@PathVariable @Min(1) Integer templateId,
|
||||
@RequestBody @Valid TemplateNew template) {
|
||||
Boolean success = templateService.updateTemplate(templateId, template);
|
||||
return ResponseResult.ok()
|
||||
.setData("template_id", templateId)
|
||||
.setData("success", success);
|
||||
}
|
||||
|
||||
@Override
|
||||
@DeleteMapping("/{templateId}/delete")
|
||||
public ResponseResult deleteTemplate(@PathVariable @Min(1) Integer templateId) {
|
||||
return ResponseResult.ok()
|
||||
.setData("template_id", templateId)
|
||||
.setData("success", templateService.deleteTemplate(templateId));
|
||||
}
|
||||
|
||||
@Override
|
||||
@GetMapping("/query/source_system")
|
||||
public ResponseResult querySourceSystem() {
|
||||
return ResponseResult.ok()
|
||||
.setData("source_system", templateService.querySourceSystem());
|
||||
}
|
||||
|
||||
@Override
|
||||
@GetMapping("/query/event_name/{sourceSystem}")
|
||||
public ResponseResult queryEventName(@PathVariable String sourceSystem) {
|
||||
return ResponseResult.ok()
|
||||
.setData("event_name", templateService.queryEventName(sourceSystem));
|
||||
}
|
||||
|
||||
// @Override
|
||||
// @PostMapping("/query/templateId")
|
||||
// public ResponseResult queryTemplateId(@RequestBody Map<String,String> map) {
|
||||
// String sourceSystem = map.get("source_system");
|
||||
// String eventName = map.get("event_name");
|
||||
// return ResponseResult.ok()
|
||||
// .setData("template_id", templateService.queryTemplateId(sourceSystem, eventName));
|
||||
// }
|
||||
|
||||
/*
|
||||
策略模板数据统计
|
||||
*/
|
||||
@Override
|
||||
@GetMapping("/statistics")
|
||||
public ResponseResult statisticsTemplate() {
|
||||
return ResponseResult.ok()
|
||||
.setData("template_num", templateService.queryTemplateTotalNum(null, null, null, null, null, null, null))
|
||||
.setData("template_used_num", templateService.queryUsedTemplateTotalNum())
|
||||
.setData("template_audit_num", templateService.queryAuditTemplateTotalNum(
|
||||
AuditStatusEnum.getNumByState(AuditStatusEnum.AUDITED.getState())
|
||||
))
|
||||
.setData("template_unaudit_num", templateService.queryAuditTemplateTotalNum(
|
||||
AuditStatusEnum.getNumByState(AuditStatusEnum.PENDING.getState())
|
||||
));
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 审批
|
||||
*/
|
||||
@Override
|
||||
@GetMapping("/{id}/audit/{auditStatus}")
|
||||
public ResponseResult updateTemplateAuditStatus(@PathVariable Integer id, @PathVariable Integer auditStatus) {
|
||||
if (id <= 0 || auditStatus < 0 || auditStatus > 2) {
|
||||
return new ResponseResult(400, "id or status is invalid")
|
||||
.setData("template_id", id)
|
||||
.setData("success", false);
|
||||
}
|
||||
return ResponseResult.ok()
|
||||
.addDataMap(templateService.updateAuditStatus(id, auditStatus))
|
||||
.setData("template_id", id);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 批量修改审核状态
|
||||
*/
|
||||
@Override
|
||||
@PostMapping("/auditbatch")
|
||||
public ResponseResult updateWhiteListAuditStatusBatch(@RequestBody Map<Integer, Integer> idsWithAuditStatusMap) {
|
||||
List<Integer> errorIds = new ArrayList<>();
|
||||
for (Map.Entry<Integer, Integer> entry: idsWithAuditStatusMap.entrySet()) {
|
||||
Integer id = entry.getKey();
|
||||
Integer auditStatus = entry.getValue();
|
||||
if (id <= 0 || auditStatus < 0 || auditStatus > 2) {
|
||||
errorIds.add(id);
|
||||
}
|
||||
}
|
||||
if (!errorIds.isEmpty()){
|
||||
return new ResponseResult(400, "id or status is invalid")
|
||||
.setData("id", errorIds)
|
||||
.setData("success", false);
|
||||
}
|
||||
|
||||
return ResponseResult.ok()
|
||||
.setData("success", templateService.updateAuditStatusBatch(idsWithAuditStatusMap));
|
||||
}
|
||||
|
||||
/*
|
||||
修改审批信息
|
||||
*/
|
||||
@Override
|
||||
@PostMapping("/auditInfo/{ids}")
|
||||
public ResponseResult updateAuditInfo(@PathVariable List<Integer> ids,
|
||||
@RequestBody Map<String, String> auditInfo) {
|
||||
if (auditInfo.get("auditInfo") == null || auditInfo.get("auditInfo").isEmpty()) {
|
||||
return ResponseResult.ok();
|
||||
}
|
||||
return ResponseResult.ok()
|
||||
.setData("success", templateService.updateAuditInfo(ids, auditInfo.get("auditInfo")));
|
||||
}
|
||||
|
||||
@Override
|
||||
@GetMapping("/auditInfo/{id}")
|
||||
public ResponseResult queryAuditInfo(@PathVariable Integer id) {
|
||||
return ResponseResult.ok()
|
||||
.setData("auditInfo", templateService.queryAuditInfo(id));
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,66 @@
|
||||
package com.realtime.protection.server.defense.templatenew;
|
||||
|
||||
import com.realtime.protection.configuration.entity.defense.template.ProtectLevel;
|
||||
import com.realtime.protection.configuration.entity.defense.template.Template;
|
||||
import com.realtime.protection.configuration.entity.defense.template.TemplateNew;
|
||||
import org.apache.ibatis.annotations.Delete;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
@Mapper
|
||||
public interface TemplateMapper {
|
||||
void newTemplate(TemplateNew template);
|
||||
|
||||
List<TemplateNew> queryTemplates(@Param("template_name") String templateName,
|
||||
@Param("source_system") String sourceSystem,
|
||||
@Param("create_username") String createUsername,
|
||||
@Param("template_id")Integer templateId,
|
||||
@Param("audit_status")Integer auditStatus,
|
||||
@Param("event_type")String eventType,
|
||||
@Param("protect_level")Integer protectLevel,
|
||||
@Param("page") Integer page,
|
||||
@Param("page_size") Integer pageSize);
|
||||
|
||||
|
||||
Boolean updateTemplateInformation(@Param("template") TemplateNew template);
|
||||
|
||||
TemplateNew queryTemplate(@Param("template_id") Integer templateId);
|
||||
|
||||
@Delete("DELETE FROM t_strategy_template WHERE strategy_template_id = #{template_id}")
|
||||
Boolean deleteTemplate(@Param("template_id") Integer templateId);
|
||||
|
||||
Integer queryTemplateTotalNum(@Param("template_name") String templateName,
|
||||
@Param("source_system") String sourceSystem,
|
||||
@Param("create_username") String createUsername,
|
||||
@Param("template_id")Integer templateId,
|
||||
@Param("audit_status")Integer auditStatus,
|
||||
@Param("event_type")String eventType,
|
||||
@Param("protect_level")Integer protectLevel
|
||||
);
|
||||
|
||||
List<String> querySourceSystem();
|
||||
|
||||
List<String> queryEventName(String sourceSystem);
|
||||
|
||||
// Integer queryTemplateId(String sourceSystem, String eventName);
|
||||
|
||||
Integer queryUsedTemplateTotalNum();
|
||||
|
||||
Integer queryAuditTemplateTotalNum(Integer auditState);
|
||||
|
||||
Integer queryAuditStatusById(Integer id);
|
||||
|
||||
Boolean updateAuditStatusById(Integer id, Integer auditStatus);
|
||||
|
||||
List<Integer> queryAuditStatusByIds(@Param("idsWithAuditStatusMap") Map<Integer, Integer> idsWithAuditStatusMap);
|
||||
|
||||
void updateAuditStatusByIdBatch(@Param("idWithAuditStatusBatch") Map<Integer, Integer> idWithAuditStatusBatch);
|
||||
|
||||
Boolean updateAuditInfo(List<Integer> ids, String auditInfo);
|
||||
|
||||
String queryAuditInfo(Integer id);
|
||||
|
||||
}
|
||||
@@ -0,0 +1,67 @@
|
||||
package com.realtime.protection.server.defense.templatenew;
|
||||
|
||||
import com.realtime.protection.configuration.entity.defense.template.Template;
|
||||
import com.realtime.protection.configuration.entity.defense.template.TemplateNew;
|
||||
import com.realtime.protection.configuration.response.ResponseResult;
|
||||
import jakarta.validation.Valid;
|
||||
import jakarta.validation.constraints.Min;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
public interface TemplateNewCpntrollerApi {
|
||||
|
||||
|
||||
@GetMapping("/query")
|
||||
ResponseResult queryTemplates(@RequestParam(value = "template_name", required = false) String templateName,
|
||||
@RequestParam(value = "source_system", required = false) String sourceSystem,
|
||||
@RequestParam(value = "create_user_name", required = false) String createUserName,
|
||||
@RequestParam(value = "template_id", required = false) Integer templateId,
|
||||
@RequestParam(value = "audit_status", required = false) Integer auditStatus,
|
||||
@RequestParam(value = "event_type", required = false) String eventType,
|
||||
@RequestParam(value = "protect_level", required = false) Integer protectLevel,
|
||||
|
||||
@RequestParam("page") @Min(1) Integer page,
|
||||
@RequestParam("page_size") @Min(1) Integer pageSize);
|
||||
|
||||
@GetMapping("/{templateId}/query")
|
||||
ResponseResult queryTemplate(@PathVariable Integer templateId);
|
||||
|
||||
|
||||
|
||||
@PostMapping("/{templateId}/update")
|
||||
ResponseResult updateTemplate(@PathVariable @Min(1) Integer templateId,
|
||||
@RequestBody @Valid TemplateNew template);
|
||||
|
||||
@DeleteMapping("/{templateId}/delete")
|
||||
ResponseResult deleteTemplate(@PathVariable @Min(1) Integer templateId);
|
||||
|
||||
@GetMapping("/query/source_system")
|
||||
ResponseResult querySourceSystem();
|
||||
|
||||
@GetMapping("/query/event_name/{sourceSystem}")
|
||||
ResponseResult queryEventName(@PathVariable String sourceSystem);
|
||||
|
||||
/*
|
||||
策略模板数据统计
|
||||
*/
|
||||
@GetMapping("/statistics")
|
||||
ResponseResult statisticsTemplate();
|
||||
|
||||
@GetMapping("/{id}/audit/{auditStatus}")
|
||||
ResponseResult updateTemplateAuditStatus(@PathVariable Integer id, @PathVariable Integer auditStatus);
|
||||
|
||||
@PostMapping("/auditbatch")
|
||||
ResponseResult updateWhiteListAuditStatusBatch(@RequestBody Map<Integer, Integer> idsWithAuditStatusMap);
|
||||
|
||||
/*
|
||||
修改审批信息
|
||||
*/
|
||||
@PostMapping("/auditInfo/{ids}")
|
||||
ResponseResult updateAuditInfo(@PathVariable List<Integer> ids,
|
||||
@RequestBody Map<String, String> auditInfo);
|
||||
|
||||
@GetMapping("/auditInfo/{id}")
|
||||
ResponseResult queryAuditInfo(@PathVariable Integer id);
|
||||
}
|
||||
@@ -0,0 +1,169 @@
|
||||
package com.realtime.protection.server.defense.templatenew;
|
||||
|
||||
import com.realtime.protection.configuration.entity.defense.template.Template;
|
||||
import com.realtime.protection.configuration.entity.defense.template.TemplateNew;
|
||||
import com.realtime.protection.configuration.utils.Counter;
|
||||
import com.realtime.protection.configuration.utils.SqlSessionWrapper;
|
||||
import com.realtime.protection.configuration.utils.enums.audit.AuditStatusValidator;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.function.Function;
|
||||
|
||||
@Service
|
||||
public class TemplateService {
|
||||
|
||||
private final TemplateMapper templateMapper;
|
||||
private final Counter counter;
|
||||
private final SqlSessionWrapper sqlSessionWrapper;
|
||||
|
||||
public TemplateService(TemplateMapper templateMapper, Counter counter, SqlSessionWrapper sqlSessionWrapper) {
|
||||
this.templateMapper = templateMapper;
|
||||
this.counter = counter;
|
||||
this.sqlSessionWrapper = sqlSessionWrapper;
|
||||
}
|
||||
|
||||
|
||||
public Integer newTemplate( TemplateNew template) {
|
||||
template.setTemplateDisplayId(
|
||||
"CLMB-"
|
||||
+ LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"))
|
||||
+ "-"
|
||||
+ String.format("%06d", counter.generateId("strategy_template"))
|
||||
);
|
||||
|
||||
templateMapper.newTemplate(template);
|
||||
|
||||
if (template.getTemplateId() == null) {
|
||||
return 0;
|
||||
}
|
||||
return template.getTemplateId();
|
||||
}
|
||||
|
||||
public List<TemplateNew> queryTemplates(String templateName, String sourceSystem, String createUserName,
|
||||
Integer templateId, Integer auditStatus,
|
||||
String eventType, Integer protectLevel,
|
||||
Integer page, Integer pageSize) {
|
||||
return templateMapper.queryTemplates(templateName, sourceSystem, createUserName,
|
||||
templateId, auditStatus,eventType, protectLevel, page, pageSize);
|
||||
}
|
||||
|
||||
public Boolean updateTemplate(Integer templateId, TemplateNew template) {
|
||||
template.setTemplateId(templateId);
|
||||
|
||||
return templateMapper.updateTemplateInformation(template);
|
||||
}
|
||||
|
||||
public TemplateNew queryTemplate(Integer templateId) {
|
||||
return templateMapper.queryTemplate(templateId);
|
||||
}
|
||||
|
||||
|
||||
|
||||
public Boolean deleteTemplate(Integer templateId) {
|
||||
return templateMapper.deleteTemplate(templateId);
|
||||
}
|
||||
|
||||
public Integer queryTemplateTotalNum(
|
||||
String templateName, String sourceSystem, String createUserName,
|
||||
Integer templateId, Integer auditStatus,
|
||||
String eventType, Integer protectLevel) {
|
||||
return templateMapper.queryTemplateTotalNum(templateName, sourceSystem, createUserName,
|
||||
templateId,auditStatus,eventType, protectLevel );
|
||||
}
|
||||
|
||||
public List<String> querySourceSystem() {
|
||||
return templateMapper.querySourceSystem();
|
||||
}
|
||||
|
||||
public List<String> queryEventName(String sourceSystem) {
|
||||
return templateMapper.queryEventName(sourceSystem);
|
||||
}
|
||||
|
||||
|
||||
|
||||
public Integer queryUsedTemplateTotalNum() {
|
||||
return templateMapper.queryUsedTemplateTotalNum();
|
||||
}
|
||||
|
||||
public Integer queryAuditTemplateTotalNum(Integer auditState) {
|
||||
return templateMapper.queryAuditTemplateTotalNum(auditState);
|
||||
}
|
||||
|
||||
public Map<String, Object> updateAuditStatus(Integer id, Integer auditStatus) {
|
||||
Integer originalAuditStatus = templateMapper.queryAuditStatusById(id);
|
||||
if (originalAuditStatus == null) {
|
||||
throw new IllegalArgumentException("cannot find audit status of static rule " + id + ", maybe static rule doesn't exist?");
|
||||
}
|
||||
if (!AuditStatusValidator.setOriginal(originalAuditStatus).checkValidate(auditStatus)) {
|
||||
throw new IllegalArgumentException("invalid audit status");
|
||||
}
|
||||
Boolean success = templateMapper.updateAuditStatusById(id, auditStatus);
|
||||
|
||||
Map<String, Object> resultMap = new HashMap<>();
|
||||
resultMap.put("success", success);
|
||||
resultMap.put("audit_status", auditStatus);
|
||||
return resultMap;
|
||||
}
|
||||
|
||||
public Boolean updateAuditStatusBatch(Map<Integer, Integer> idsWithAuditStatusMap) {
|
||||
//校验id和status是否合法
|
||||
List<Integer> originalAuditStatusList = templateMapper.queryAuditStatusByIds(idsWithAuditStatusMap);
|
||||
if (originalAuditStatusList == null || originalAuditStatusList.size() != idsWithAuditStatusMap.size()) {
|
||||
throw new IllegalArgumentException("策略模板部分不存在");
|
||||
}
|
||||
int index = 0;
|
||||
List<Integer> errorIds = new ArrayList<>();
|
||||
for(Map.Entry<Integer, Integer> entry: idsWithAuditStatusMap.entrySet()) {
|
||||
Integer id = entry.getKey();
|
||||
Integer auditStatus = entry.getValue();
|
||||
Integer originalAuditStatus = originalAuditStatusList.get(index);
|
||||
index++;
|
||||
if (!AuditStatusValidator.setOriginal(originalAuditStatus).checkValidate(auditStatus)) {
|
||||
errorIds.add(id);
|
||||
}
|
||||
}
|
||||
if (!errorIds.isEmpty()){
|
||||
throw new IllegalArgumentException("策略模板无法修改为对应审核状态, 错误id: " + errorIds);
|
||||
}
|
||||
|
||||
Function<com.realtime.protection.server.defense.template.TemplateMapper, Function<Map<Integer, Integer>, Boolean>> updateTemplateAuditStatusFunction =
|
||||
mapper -> map -> {
|
||||
if (map == null || map.isEmpty()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
Map<Integer, Integer> idWithAuditStatusBatch = new HashMap<>();
|
||||
for (Map.Entry<Integer, Integer> item : map.entrySet()) {
|
||||
idWithAuditStatusBatch.put(item.getKey(), item.getValue());
|
||||
if (idWithAuditStatusBatch.size() < 100) {
|
||||
continue;
|
||||
}
|
||||
//mapper指的就是外层函数输入的参数，也就是WhiteListMapper
|
||||
mapper.updateAuditStatusByIdBatch(idWithAuditStatusBatch);
|
||||
idWithAuditStatusBatch.clear();
|
||||
}
|
||||
if (!idWithAuditStatusBatch.isEmpty()) {
|
||||
mapper.updateAuditStatusByIdBatch(idWithAuditStatusBatch);
|
||||
}
|
||||
return true;
|
||||
};
|
||||
//实现事务操作
|
||||
return sqlSessionWrapper.startBatchSession(com.realtime.protection.server.defense.template.TemplateMapper.class, updateTemplateAuditStatusFunction, idsWithAuditStatusMap);
|
||||
|
||||
}
|
||||
|
||||
|
||||
public Boolean updateAuditInfo(List<Integer> ids, String auditInfo) {
|
||||
return templateMapper.updateAuditInfo(ids, auditInfo);
|
||||
}
|
||||
|
||||
public String queryAuditInfo(Integer id) {
|
||||
return templateMapper.queryAuditInfo(id);
|
||||
}
|
||||
}
|
||||
145
src/main/resources/mappers/TemplateNewMapper.xml
Normal file
145
src/main/resources/mappers/TemplateNewMapper.xml
Normal file
@@ -0,0 +1,145 @@
|
||||
<?xml version="1.0" encoding="UTF-8" ?>
|
||||
<!DOCTYPE mapper
|
||||
PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
|
||||
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
||||
<mapper namespace="com.realtime.protection.server.defense.templatenew.TemplateMapper">
|
||||
<insert id="newTemplate" useGeneratedKeys="true" keyProperty="templateId">
|
||||
INSERT INTO t_strategy_template_new(strategy_template_name, strategy_template_source_system,
|
||||
has_protect_object_ip, has_protect_object_port,
|
||||
has_protocol, has_url, has_dns,
|
||||
has_peer_ip, has_peer_port, is_full_flow, is_protect_object_src,
|
||||
strategy_template_create_user_id, strategy_template_create_username,
|
||||
strategy_template_create_depart,
|
||||
strategy_template_description, strategy_template_display_id)
|
||||
VALUE (#{template.templateName}, #{template.sourceSystem},
|
||||
#{template.hasProtectObjectIp}, #{template.hasProtectObjectPort},
|
||||
#{template.hasProtocol}, #{template.hasUrl}, #{template.hasDns},
|
||||
#{template.hasPeerIp}, #{template.hasPeerPort}, #{template.isFullFlow},
|
||||
#{template.isProtectObjectSrc},
|
||||
#{template.createUserId}, #{template.createUsername}, #{template.createDepart},
|
||||
#{template.description}, #{template.templateDisplayId})
|
||||
</insert>
|
||||
|
||||
<update id="updateTemplateInformation">
|
||||
UPDATE t_strategy_template_new
|
||||
<set>
|
||||
<if test="template.templateName != null">strategy_template_name = #{template.templateName},</if>
|
||||
<if test="template.sourceSystem != null">strategy_template_source_system = #{template.sourceSystem},</if>
|
||||
<if test="template.template.hasProtectObjectIp != null">has_protect_object_ip = #{template.hasProtectObjectIp},</if>
|
||||
<if test="template.template.hasProtectObjectPort != null">has_protect_object_port = #{template.hasProtectObjectPort},</if>
|
||||
<if test="template.template.hasProtocol != null">has_protocol = #{template.hasProtocol},</if>
|
||||
<if test="template.template.hasUrl != null">has_url = #{template.hasUrl},</if>
|
||||
<if test="template.template.hasDns != null">has_dns = #{template.hasDns},</if>
|
||||
<if test="template.template.hasPeerIp != null">has_peer_ip = #{template.hasPeerIp},</if>
|
||||
<if test="template.template.hasPeerPort != null">has_peer_port = #{template.hasPeerPort},</if>
|
||||
<if test="template.template.isFullFlow != null">is_full_flow = #{template.isFullFlow},</if>
|
||||
<if test="template.template.isProtectObjectSrc != null">is_protect_object_src = #{template.isProtectObjectSrc},</if>
|
||||
|
||||
modify_time = NOW()
|
||||
</set>
|
||||
<where>
|
||||
AND strategy_template_id = #{template.templateId}
|
||||
</where>
|
||||
</update>
|
||||
|
||||
|
||||
|
||||
<resultMap id="templateMap" type="com.realtime.protection.configuration.entity.defense.template.TemplateNew">
|
||||
<id column="strategy_template_id" property="templateId"/>
|
||||
<result column="strategy_template_name" property="templateName"/>
|
||||
|
||||
<result column="strategy_template_used_times" property="usedTimes"/>
|
||||
<result column="strategy_template_running_tasks" property="runningTasks"/>
|
||||
<result column="audit_status" property="auditStatus"/>
|
||||
<result column="audit_user_name" property="auditUserName"/>
|
||||
<result column="audit_user_depart" property="auditUserDepart"/>
|
||||
|
||||
<result column="strategy_template_display_id" property="templateDisplayId"/>
|
||||
|
||||
<result column="strategy_template_create_user_id" property="createUserId"/>
|
||||
<result column="strategy_template_create_username" property="createUsername"/>
|
||||
<result column="strategy_template_create_depart" property="createDepart"/>
|
||||
<result column="create_time" property="createTime"/>
|
||||
<result column="update_time" property="updateTime"/>
|
||||
|
||||
<result column="strategy_template_source_system" property="sourceSystem"/>
|
||||
<result column="event_type" property="eventType"/>
|
||||
<result column="protect_level" property="protectLevel"/>
|
||||
<result column="strategy_template_description" property="description"/>
|
||||
|
||||
<result column="has_protect_object_ip" property="hasProtectObjectIP"/>
|
||||
<result column="has_protect_object_port" property="hasProtectObjectPort"/>
|
||||
<result column="has_protocol" property="hasProtocol"/>
|
||||
<result column="has_url" property="hasURL"/>
|
||||
<result column="has_dns" property="hasDNS"/>
|
||||
<result column="has_peer_ip" property="hasPeerIP"/>
|
||||
<result column="has_peer_port" property="hasPeerPort"/>
|
||||
<result column="is_full_flow" property="isFullFlow"/>
|
||||
<result column="is_protect_object_src" property="isProtectObjectIPSrc"/>
|
||||
|
||||
</resultMap>
|
||||
|
||||
<select id="queryTemplates" resultMap="templateMap">
|
||||
SELECT * FROM t_strategy_template_new
|
||||
<where>
|
||||
<if test="source_system != null and source_system != '' ">
|
||||
AND strategy_template_source_system = #{source_system}
|
||||
</if>
|
||||
<if test="create_username != null and create_username != '' ">
|
||||
AND strategy_template_create_username = #{create_username}
|
||||
</if>
|
||||
<if test="template_name != null and template_name != '' ">
|
||||
AND strategy_template_name LIKE CONCAT('%', #{template_name}, '%')
|
||||
</if>
|
||||
<if test="template_id != null">
|
||||
AND strategy_template_id = #{template_id}
|
||||
</if>
|
||||
<if test="audit_status != null">
|
||||
AND audit_status = #{audit_status}
|
||||
</if>
|
||||
<if test="event_type != null and event_type != '' ">
|
||||
AND event_type LIKE CONCAT('%', #{event_type}, '%')
|
||||
</if>
|
||||
<if test="protect_level != null and protect_level != '' ">
|
||||
AND protect_level LIKE CONCAT('%', #{protect_level}, '%')
|
||||
</if>
|
||||
</where>
|
||||
LIMIT ${(page - 1) * page_size}, #{page_size}
|
||||
</select>
|
||||
|
||||
<select id="queryTemplate" resultMap="templateMap">
|
||||
SELECT * FROM t_strategy_template_new
|
||||
<where>
|
||||
strategy_template_id = #{template_id}
|
||||
</where>
|
||||
</select>
|
||||
|
||||
<select id="queryTemplateTotalNum" resultType="java.lang.Integer">
|
||||
SELECT count(*) FROM t_strategy_template_new
|
||||
<where>
|
||||
<if test="source_system != null and source_system != '' ">
|
||||
AND strategy_template_source_system = #{source_system}
|
||||
</if>
|
||||
<if test="create_username != null and create_username != '' ">
|
||||
AND strategy_template_create_username = #{create_username}
|
||||
</if>
|
||||
<if test="template_name != null and template_name != '' ">
|
||||
AND strategy_template_name LIKE CONCAT('%', #{template_name}, '%')
|
||||
</if>
|
||||
<if test="template_id != null">
|
||||
AND strategy_template_id = #{template_id}
|
||||
</if>
|
||||
<if test="audit_status != null">
|
||||
AND audit_status = #{audit_status}
|
||||
</if>
|
||||
<if test="event_type != null and event_type != '' ">
|
||||
AND event_type LIKE CONCAT('%', #{event_type}, '%')
|
||||
</if>
|
||||
<if test="protect_level != null and protect_level != '' ">
|
||||
AND protect_level LIKE CONCAT('%', #{protect_level}, '%')
|
||||
</if>
|
||||
</where>
|
||||
</select>
|
||||
|
||||
|
||||
</mapper>
|
||||
@@ -96,6 +96,16 @@ class TemplateServiceTest extends ProtectionApplicationTests {
|
||||
// }
|
||||
}
|
||||
|
||||
@Test
|
||||
void testQueryTemplateShort() {
|
||||
List<Template> templates = templateService.queryTemplatesShort(
|
||||
);
|
||||
// assertEquals(5, templates.size());
|
||||
for (Template template : templates) {
|
||||
System.out.println(template);
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
void testUpdateTemplateSuccess() {
|
||||
List<Template> templates = templateService.queryTemplates(
|
||||
|
||||
Reference in New Issue
Block a user