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

1、静态规则增加重复性检测

Browse Source
This commit is contained in:
PushM
2024-06-10 04:34:54 +08:00
parent 274eecfa79
commit a0e9515d3b
4 changed files with 59 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -52,6 +52,10 @@ public class StaticRuleController implements StaticRuleControllerApi {
object.setStaticRuleCreateUserId(Integer.valueOf(user.uid));
object.setStaticRuleCreateDepart(user.getOrgName());
}
Integer duplicateRuleNum = staticRuleService.queryDuplicateStaticRule(object);
if (duplicateRuleNum > 0) {
return ResponseResult.invalid().setMessage("规则重复");
}
//调用service新增
staticRuleService.newStaticRuleObject(object);
@@ -126,7 +130,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
//调用service修改
Boolean updateValid = staticRuleService.updateStaticRule(id, object);
if (!updateValid) {
return ResponseResult.invalid().setMessage("id is invalid");
return ResponseResult.invalid().setMessage("无效的规则");
}
return ResponseResult.ok().setData("success",updateValid);
}
@@ -141,7 +145,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
log.info("根据id查询静态规则:{}", id);
StaticRuleObject object = staticRuleService.queryStaticRuleById(id);
if (object == null) {
return ResponseResult.invalid().setMessage("id is invalid");
return ResponseResult.invalid().setMessage("无效的规则");
}
return ResponseResult.ok().setData("static_rule", object);
}
@@ -199,7 +203,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
public ResponseResult updateStaticRuleAuditStatus(@PathVariable Integer id, @PathVariable Integer auditStatus,
@Autowired HttpServletRequest request) {
if (id <= 0 || auditStatus < 0 || auditStatus > 2) {
return new ResponseResult(400, "id or status is invalid")
return new ResponseResult(400, "规则或状态无效")
.setData("staticRule_id", id)
.setData("success", false);
}
@@ -238,7 +242,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
}
}
if (!errorIds.isEmpty()){
return new ResponseResult(400, "id or status is invalid")
return new ResponseResult(400, "规则或状态无效")
.setData("staticRule_id", errorIds)
.setData("success", false);
}

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

@@ -81,5 +81,7 @@ public interface StaticRuleMapper {
void removeUsedTaskId(Long taskId);
Integer queryDuplicateStaticRule(StaticRuleObject object);
// boolean queryStaticRuleRepeat(StaticRuleObject object);
}

4
src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleService.java
View File

@@ -428,4 +428,8 @@ public class StaticRuleService {
public List<StaticRuleObject> queryHistory(Integer id, Integer page, Integer pageSize) {
return staticRuleMapper.queryHistory(id, page, pageSize);
}
public Integer queryDuplicateStaticRule(StaticRuleObject object) {
return staticRuleMapper.queryDuplicateStaticRule(object);
}
}