1、静态动态规则分页查询条件新增审核状态

This commit is contained in:
PushM
2024-04-29 19:49:50 +08:00
parent da76e5a590
commit dcf77ffc27
14 changed files with 56 additions and 30 deletions

View File

@@ -108,6 +108,7 @@ public class DynamicRuleController implements DynamicRuleControllerApi {
@RequestParam(value = "id", required = false) Integer dynamicRuleId, @RequestParam(value = "id", required = false) Integer dynamicRuleId,
@RequestParam(value = "source_system", required = false) String protectObjectSourceSystem, @RequestParam(value = "source_system", required = false) String protectObjectSourceSystem,
@RequestParam(value = "creator", required = false) String creator, @RequestParam(value = "creator", required = false) String creator,
@RequestParam(value = "audit_status", required = false) Integer auditStatus,
@RequestParam(value = "page", defaultValue = "1") Integer page, @RequestParam(value = "page", defaultValue = "1") Integer page,
@RequestParam(value = "page_size", defaultValue = "10") Integer pageSize) { @RequestParam(value = "page_size", defaultValue = "10") Integer pageSize) {
log.info("分页查询动态规则: {}:{}:{}:{}", dynamicRuleName, dynamicRuleId, page, pageSize); log.info("分页查询动态规则: {}:{}:{}:{}", dynamicRuleName, dynamicRuleId, page, pageSize);
@@ -115,9 +116,9 @@ public class DynamicRuleController implements DynamicRuleControllerApi {
return ResponseResult.ok() return ResponseResult.ok()
.setData("success", true) .setData("success", true)
.setData("dynamic_rule_list", dynamicRuleService.queryDynamicRuleObject(dynamicRuleName, dynamicRuleId, .setData("dynamic_rule_list", dynamicRuleService.queryDynamicRuleObject(dynamicRuleName, dynamicRuleId,
protectObjectSourceSystem, creator, page, pageSize)) protectObjectSourceSystem, creator, auditStatus, page, pageSize))
.setData("total_num",dynamicRuleService.queryDynamicRuleTotalNum(dynamicRuleName, dynamicRuleId, .setData("total_num",dynamicRuleService.queryDynamicRuleTotalNum(dynamicRuleName, dynamicRuleId,
protectObjectSourceSystem, creator)); protectObjectSourceSystem, creator, auditStatus));
} }
//详情查看?? 就是按id查询吧 //详情查看?? 就是按id查询吧
@@ -170,7 +171,7 @@ public class DynamicRuleController implements DynamicRuleControllerApi {
public ResponseResult getStaticRuleStatisticsData(){ public ResponseResult getStaticRuleStatisticsData(){
return ResponseResult.ok() return ResponseResult.ok()
.setData("dynamic_rule_num", dynamicRuleService.queryDynamicRuleTotalNum(null, null, .setData("dynamic_rule_num", dynamicRuleService.queryDynamicRuleTotalNum(null,null, null,
null, null)) null, null))
.setData("dynamic_rule_used_num", dynamicRuleService.queryAuditDynamicRuleTotalNum( .setData("dynamic_rule_used_num", dynamicRuleService.queryAuditDynamicRuleTotalNum(
AuditStatusEnum.getNumByState(AuditStatusEnum.USING.getState()) AuditStatusEnum.getNumByState(AuditStatusEnum.USING.getState())

View File

@@ -347,6 +347,7 @@ public interface DynamicRuleControllerApi {
@Parameter(name = "id", description = "动态规则ID", example = "2"), @Parameter(name = "id", description = "动态规则ID", example = "2"),
@Parameter(name = "source_system", description = "来源你系统", example = "BW系统"), @Parameter(name = "source_system", description = "来源你系统", example = "BW系统"),
@Parameter(name = "creator", description = "创建人名称"), @Parameter(name = "creator", description = "创建人名称"),
@Parameter(name = "audit_status", description = "审核状态", example = "1"),
@Parameter(name = "page", description = "页码", example = "1"), @Parameter(name = "page", description = "页码", example = "1"),
@Parameter(name = "page_size", description = "每页大小", example = "10") @Parameter(name = "page_size", description = "每页大小", example = "10")
} }
@@ -355,6 +356,7 @@ public interface DynamicRuleControllerApi {
@RequestParam(value = "id", required = false) Integer dynamicRuleId, @RequestParam(value = "id", required = false) Integer dynamicRuleId,
@RequestParam(value = "source_system", required = false) String protectObjectSourceSystem, @RequestParam(value = "source_system", required = false) String protectObjectSourceSystem,
@RequestParam(value = "creator", required = false) String creator, @RequestParam(value = "creator", required = false) String creator,
@RequestParam(value = "audit_status", required = false) Integer auditStatus,
@RequestParam(value = "page", defaultValue = "1") Integer page, @RequestParam(value = "page", defaultValue = "1") Integer page,
@RequestParam(value = "page_size", defaultValue = "10") Integer pageSize); @RequestParam(value = "page_size", defaultValue = "10") Integer pageSize);

View File

@@ -31,7 +31,7 @@ public interface DynamicRuleMapper {
void deleteDynamicRules(List<Integer> dynamicRuleIds); void deleteDynamicRules(List<Integer> dynamicRuleIds);
List<DynamicRuleObject> queryDynamicRuleObject(String dynamicRuleName, Integer dynamicRuleId, List<DynamicRuleObject> queryDynamicRuleObject(String dynamicRuleName, Integer dynamicRuleId,
String sourceSystem, String creator, String sourceSystem, String creator, Integer auditStatus,
Integer page, Integer pageSize); Integer page, Integer pageSize);
void deleteDynamicRuleProtectObjectConcat(Integer dynamicRuleId); void deleteDynamicRuleProtectObjectConcat(Integer dynamicRuleId);
@@ -43,7 +43,7 @@ public interface DynamicRuleMapper {
boolean queryProtectObjectById(Integer protectObjectId); boolean queryProtectObjectById(Integer protectObjectId);
Integer queryDynamicRuleTotalNum(String dynamicRuleName, Integer dynamicRuleId, Integer queryDynamicRuleTotalNum(String dynamicRuleName, Integer dynamicRuleId,
String sourceSystem, String creator); String sourceSystem, String creator, Integer auditStatus);
List<DynamicRuleObject> queryDynamicRuleByIds(List<Integer> ids); List<DynamicRuleObject> queryDynamicRuleByIds(List<Integer> ids);

View File

@@ -171,16 +171,17 @@ public class DynamicRuleService {
//分页查询基础的动态规则,暂时不返回关联的保护对象 //分页查询基础的动态规则,暂时不返回关联的保护对象
public List<DynamicRuleObject> queryDynamicRuleObject(String dynamicRuleName, Integer dynamicRuleId, public List<DynamicRuleObject> queryDynamicRuleObject(String dynamicRuleName, Integer dynamicRuleId,
String templateSourceSystem,String creator, String templateSourceSystem,String creator,Integer auditStatus,
Integer page, Integer pageSize) { Integer page, Integer pageSize) {
return dynamicRuleMapper.queryDynamicRuleObject(dynamicRuleName, dynamicRuleId, return dynamicRuleMapper.queryDynamicRuleObject(dynamicRuleName, dynamicRuleId,
templateSourceSystem, creator, page, pageSize); templateSourceSystem, creator, auditStatus, page, pageSize);
} }
public Integer queryDynamicRuleTotalNum(String dynamicRuleName, Integer dynamicRuleId, public Integer queryDynamicRuleTotalNum(String dynamicRuleName, Integer dynamicRuleId,
String protectObjectSourceSystem, String creator) { String protectObjectSourceSystem, String creator,
Integer auditStatus) {
return dynamicRuleMapper.queryDynamicRuleTotalNum(dynamicRuleName, dynamicRuleId, return dynamicRuleMapper.queryDynamicRuleTotalNum(dynamicRuleName, dynamicRuleId,
protectObjectSourceSystem, creator); protectObjectSourceSystem, creator, auditStatus);
} }
public List<DynamicRuleObject> queryDynamicRuleByIds(List<Integer> ids) { public List<DynamicRuleObject> queryDynamicRuleByIds(List<Integer> ids) {

View File

@@ -134,6 +134,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
@RequestParam(name = "id", required = false) Integer static_rule_id, @RequestParam(name = "id", required = false) Integer static_rule_id,
@RequestParam(name = "creator", required = false) String static_rule_creator, @RequestParam(name = "creator", required = false) String static_rule_creator,
@RequestParam(name = "ip", required = false) String static_rule_ip, @RequestParam(name = "ip", required = false) String static_rule_ip,
@RequestParam(name = "auditStatus", required = false) Integer auditStatus,
@RequestParam(defaultValue = "1") Integer page, @RequestParam(defaultValue = "1") Integer page,
@RequestParam(defaultValue = "10") Integer pageSize @RequestParam(defaultValue = "10") Integer pageSize
) { ) {
@@ -141,11 +142,11 @@ public class StaticRuleController implements StaticRuleControllerApi {
//调用service新增 //调用service新增
List<StaticRuleObject> pageResult = staticRuleService.queryStaticRule( List<StaticRuleObject> pageResult = staticRuleService.queryStaticRule(
static_rule_name, static_rule_id, static_rule_name, static_rule_id,
static_rule_creator, static_rule_ip, page, pageSize); static_rule_creator, static_rule_ip, auditStatus, page, pageSize);
return ResponseResult.ok() return ResponseResult.ok()
.setData("static_rule_list", pageResult) .setData("static_rule_list", pageResult)
.setData("total_num",staticRuleService.queryStaticRuleTotalNum(static_rule_name, static_rule_id, .setData("total_num",staticRuleService.queryStaticRuleTotalNum(static_rule_name, static_rule_id,
static_rule_creator, static_rule_ip)); static_rule_creator, static_rule_ip, auditStatus));
} }
@Override @Override
@@ -208,7 +209,7 @@ public class StaticRuleController implements StaticRuleControllerApi {
public ResponseResult getStaticRuleStatisticsData(){ public ResponseResult getStaticRuleStatisticsData(){
return ResponseResult.ok() return ResponseResult.ok()
.setData("static_rule_num", staticRuleService.queryStaticRuleTotalNum(null, null, null, null)) .setData("static_rule_num", staticRuleService.queryStaticRuleTotalNum(null, null, null, null, null))
.setData("static_rule_used_num", staticRuleService.queryAuditStaticRuleTotalNum( .setData("static_rule_used_num", staticRuleService.queryAuditStaticRuleTotalNum(
AuditStatusEnum.getNumByState(AuditStatusEnum.USING.getState()) AuditStatusEnum.getNumByState(AuditStatusEnum.USING.getState())
)) ))

View File

@@ -204,6 +204,7 @@ public interface StaticRuleControllerApi {
@Parameter(name = "id", description = "静态规则id"), @Parameter(name = "id", description = "静态规则id"),
@Parameter(name = "creator", description = "静态规则创建人"), @Parameter(name = "creator", description = "静态规则创建人"),
@Parameter(name = "ip", description = "静态规则ip匹配所有源和目的ip"), @Parameter(name = "ip", description = "静态规则ip匹配所有源和目的ip"),
@Parameter(name = "auditStatus", description = "静态规则审核状态"),
@Parameter(name = "page", description = "页码"), @Parameter(name = "page", description = "页码"),
@Parameter(name = "pageSize", description = "每页条数") @Parameter(name = "pageSize", description = "每页条数")
} }
@@ -212,6 +213,7 @@ public interface StaticRuleControllerApi {
@RequestParam(required = false) Integer static_rule_id, @RequestParam(required = false) Integer static_rule_id,
@RequestParam(required = false) String static_rule_creator, @RequestParam(required = false) String static_rule_creator,
@RequestParam(required = false) String static_rule_ip, @RequestParam(required = false) String static_rule_ip,
@RequestParam(required = false) Integer auditStatus,
@RequestParam(defaultValue = "1") Integer page, @RequestParam(defaultValue = "1") Integer page,
@RequestParam(defaultValue = "10") Integer pageSize @RequestParam(defaultValue = "10") Integer pageSize
); );

View File

@@ -28,7 +28,7 @@ public interface StaticRuleMapper {
//多页查询静态规则 //多页查询静态规则
List<StaticRuleObject> queryStaticRule(String static_rule_name, Integer static_rule_id, List<StaticRuleObject> queryStaticRule(String static_rule_name, Integer static_rule_id,
String static_rule_create_username, String ip, String static_rule_create_username, String ip, Integer auditStatus,
Integer page, Integer pageSize); Integer page, Integer pageSize);
@@ -41,7 +41,7 @@ public interface StaticRuleMapper {
Boolean updateAuditStatusById(Integer id, Integer auditStatus); Boolean updateAuditStatusById(Integer id, Integer auditStatus);
Integer queryStaticRuleTotalNum(String static_rule_name, Integer static_rule_id, Integer queryStaticRuleTotalNum(String static_rule_name, Integer static_rule_id,
String static_rule_create_username, String ip); String static_rule_create_username, String ip, Integer auditStatus);
List<StaticRuleObject> queryStaticRuleByIds(List<Integer> ids); List<StaticRuleObject> queryStaticRuleByIds(List<Integer> ids);

View File

@@ -166,11 +166,11 @@ public class StaticRuleService {
*/ */
public List<StaticRuleObject> queryStaticRule( public List<StaticRuleObject> queryStaticRule(
String static_rule_name, Integer static_rule_id, String static_rule_name, Integer static_rule_id,
String creator, String ip, String creator, String ip, Integer auditStatus,
Integer page, Integer pageSize) { Integer page, Integer pageSize) {
return staticRuleMapper.queryStaticRule(static_rule_name, static_rule_id, return staticRuleMapper.queryStaticRule(static_rule_name, static_rule_id,
creator, ip, page, pageSize); creator, ip, auditStatus, page, pageSize);
} }
public Boolean newStaticRuleObjects(List<StaticRuleObject> staticRuleList) { public Boolean newStaticRuleObjects(List<StaticRuleObject> staticRuleList) {
@@ -233,9 +233,9 @@ public class StaticRuleService {
} }
public Integer queryStaticRuleTotalNum(String static_rule_name, Integer static_rule_id, public Integer queryStaticRuleTotalNum(String static_rule_name, Integer static_rule_id,
String static_rule_creator, String static_rule_ip) { String static_rule_creator, String static_rule_ip, Integer auditStatus) {
return staticRuleMapper.queryStaticRuleTotalNum(static_rule_name, static_rule_id, return staticRuleMapper.queryStaticRuleTotalNum(static_rule_name, static_rule_id,
static_rule_creator, static_rule_ip); static_rule_creator, static_rule_ip, auditStatus);
} }
public List<StaticRuleObject> queryStaticRuleByIds(List<Integer> ids) { public List<StaticRuleObject> queryStaticRuleByIds(List<Integer> ids) {

View File

@@ -190,18 +190,21 @@
left join t_strategy_template left join t_strategy_template
on t_dynamic_rule.template_id = t_strategy_template.strategy_template_id on t_dynamic_rule.template_id = t_strategy_template.strategy_template_id
<where> <where>
<if test="dynamicRuleName != null"> <if test="dynamicRuleName != null and dynamicRuleName !=''" >
t_dynamic_rule.dynamic_rule_name like concat('%', #{dynamicRuleName}, '%') t_dynamic_rule.dynamic_rule_name like concat('%', #{dynamicRuleName}, '%')
</if> </if>
<if test="dynamicRuleId != null"> <if test="dynamicRuleId != null">
and t_dynamic_rule.dynamic_rule_id = #{dynamicRuleId} and t_dynamic_rule.dynamic_rule_id = #{dynamicRuleId}
</if> </if>
<if test="sourceSystem != null"> <if test="sourceSystem != null and sourceSystem !=''">
and t_strategy_template.strategy_template_source_system = #{sourceSystem} and t_strategy_template.strategy_template_source_system = #{sourceSystem}
</if> </if>
<if test="creator != null"> <if test="creator != null and creator !=''">
and t_dynamic_rule.dynamic_rule_create_username = #{creator} and t_dynamic_rule.dynamic_rule_create_username = #{creator}
</if> </if>
<if test="auditStatus != null">
and t_dynamic_rule.audit_status = #{auditStatus}
</if>
</where> </where>
LIMIT ${(page - 1) * pageSize}, #{pageSize} LIMIT ${(page - 1) * pageSize}, #{pageSize}
</select> </select>
@@ -244,6 +247,9 @@
<if test="creator != null"> <if test="creator != null">
and t_dynamic_rule.dynamic_rule_create_username = #{creator} and t_dynamic_rule.dynamic_rule_create_username = #{creator}
</if> </if>
<if test="auditStatus != null">
and t_dynamic_rule.audit_status = #{auditStatus}
</if>
</where> </where>
</select> </select>
<select id="queryDynamicRuleByIds" <select id="queryDynamicRuleByIds"

View File

@@ -156,6 +156,9 @@
AND (static_rule_sip = INET_ATON(#{ip}) or static_rule_msip = INET_ATON(#{ip}) AND (static_rule_sip = INET_ATON(#{ip}) or static_rule_msip = INET_ATON(#{ip})
or static_rule_dip = INET_ATON(#{ip}) or static_rule_mdip = INET_ATON(#{ip})) or static_rule_dip = INET_ATON(#{ip}) or static_rule_mdip = INET_ATON(#{ip}))
</if> </if>
<if test="auditStatus != null">
AND static_rule_audit_status = #{auditStatus}
</if>
</where> </where>
LIMIT ${(page - 1) * pageSize}, #{pageSize} LIMIT ${(page - 1) * pageSize}, #{pageSize}
</select> </select>
@@ -188,6 +191,9 @@
AND (static_rule_sip = INET_ATON(#{ip}) or static_rule_msip = INET_ATON(#{ip}) AND (static_rule_sip = INET_ATON(#{ip}) or static_rule_msip = INET_ATON(#{ip})
or static_rule_dip = INET_ATON(#{ip}) or static_rule_mdip = INET_ATON(#{ip})) or static_rule_dip = INET_ATON(#{ip}) or static_rule_mdip = INET_ATON(#{ip}))
</if> </if>
<if test="auditStatus != null">
AND static_rule_audit_status = #{auditStatus}
</if>
</where> </where>
</select> </select>
<select id="queryStaticRuleByIds" <select id="queryStaticRuleByIds"

View File

@@ -129,13 +129,13 @@ public class DynamicRuleServiceTest extends ProtectionApplicationTests {
@Test @Test
void testQueryDynamicRuleObject() { void testQueryDynamicRuleObject() {
List<DynamicRuleObject> objects = dynamicRuleService.queryDynamicRuleObject( List<DynamicRuleObject> objects = dynamicRuleService.queryDynamicRuleObject(
null,null,null, null,1, 10); null,null,null, null,2,1, 10);
System.out.println(objects); System.out.println(objects);
} }
@Test @Test
void testQueryDynamicRuleTotalNum(){ void testQueryDynamicRuleTotalNum(){
Integer num = dynamicRuleService.queryDynamicRuleTotalNum( Integer num = dynamicRuleService.queryDynamicRuleTotalNum(
null,null,null,null); null,null,null,null,null);
System.out.println(num); System.out.println(num);
} }

View File

@@ -99,7 +99,7 @@ public class StaticRuleServiceTest extends ProtectionApplicationTests {
@Test @Test
void testQueryStaticRules(){ void testQueryStaticRules(){
List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule(null, null, null, null, 1, 10); List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule(null, null, null, null,2, 1, 10);
System.out.println(staticRuleObjects); System.out.println(staticRuleObjects);
} }
@Test @Test

View File

@@ -73,14 +73,14 @@ class TaskServiceTest extends ProtectionApplicationTests {
void testNewTaskSuccess() { void testNewTaskSuccess() {
for (int i = 1; i < 10; i++) { for (int i = 1; i < 10; i++) {
List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule( List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule(
null, null, null, null, i, 2); null, null, null, null,2, i, 2);
List<Integer> staticRuleIds = new ArrayList<>(); List<Integer> staticRuleIds = new ArrayList<>();
staticRuleObjects.forEach(staticRuleObject -> staticRuleObjects.forEach(staticRuleObject ->
staticRuleIds.add(staticRuleObject.getStaticRuleId())); staticRuleIds.add(staticRuleObject.getStaticRuleId()));
task.setStaticRuleIds(staticRuleIds); task.setStaticRuleIds(staticRuleIds);
List<DynamicRuleObject> dynamicRuleObjects = dynamicRuleService.queryDynamicRuleObject( List<DynamicRuleObject> dynamicRuleObjects = dynamicRuleService.queryDynamicRuleObject(
null, null, null, null, i, 2 null, null, null, null,null, i, 2
); );
List<Integer> dynamicRuleIds = new ArrayList<>(); List<Integer> dynamicRuleIds = new ArrayList<>();
dynamicRuleObjects.forEach(dynamicRuleObject -> dynamicRuleObjects.forEach(dynamicRuleObject ->
@@ -104,6 +104,13 @@ class TaskServiceTest extends ProtectionApplicationTests {
}); });
} }
@Test
void testPageQueryTask(){
List<Task> tasks = taskService.queryTasks(null, null, null, null, null, 1, 10);
System.out.println(tasks);
}
@Test @Test
void testQueryTasks() { void testQueryTasks() {
String testName = "test query"; String testName = "test query";
@@ -113,14 +120,14 @@ class TaskServiceTest extends ProtectionApplicationTests {
task.setTaskName(testName); task.setTaskName(testName);
task.setTaskCreateUsername(testCreateName); task.setTaskCreateUsername(testCreateName);
List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule( List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule(
null, null, null, null, 1, 5); null, null, null, null, null,1, 5);
List<Integer> staticRuleIds = new ArrayList<>(); List<Integer> staticRuleIds = new ArrayList<>();
staticRuleObjects.forEach(staticRuleObject -> staticRuleObjects.forEach(staticRuleObject ->
staticRuleIds.add(staticRuleObject.getStaticRuleId())); staticRuleIds.add(staticRuleObject.getStaticRuleId()));
task.setStaticRuleIds(staticRuleIds); task.setStaticRuleIds(staticRuleIds);
List<DynamicRuleObject> dynamicRuleObjects = dynamicRuleService.queryDynamicRuleObject( List<DynamicRuleObject> dynamicRuleObjects = dynamicRuleService.queryDynamicRuleObject(
null, null, null, null, 1, 5 null, null, null, null, null,1, 5
); );
List<Integer> dynamicRuleIds = new ArrayList<>(); List<Integer> dynamicRuleIds = new ArrayList<>();
dynamicRuleObjects.forEach(dynamicRuleObject -> dynamicRuleObjects.forEach(dynamicRuleObject ->
@@ -157,7 +164,7 @@ class TaskServiceTest extends ProtectionApplicationTests {
.get(0); .get(0);
List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule( List<StaticRuleObject> staticRuleObjects = staticRuleService.queryStaticRule(
null, null, null, null, 1, 4 null, null, null, null, null,1, 4
); );
List<Integer> staticRuleIds = new ArrayList<>(); List<Integer> staticRuleIds = new ArrayList<>();
staticRuleObjects.forEach(staticRuleObject -> staticRuleIds.add(staticRuleObject.getStaticRuleId())); staticRuleObjects.forEach(staticRuleObject -> staticRuleIds.add(staticRuleObject.getStaticRuleId()));

View File

@@ -114,7 +114,7 @@ class WhiteListServiceTest extends ProtectionApplicationTests {
Integer id = staticRuleService.newStaticRuleObject(staticRuleTest); Integer id = staticRuleService.newStaticRuleObject(staticRuleTest);
List<Integer> ruleIds = new ArrayList<>(List.of(id)); List<Integer> ruleIds = new ArrayList<>(List.of(id));
List<StaticRuleObject> staticRules= staticRuleService.queryStaticRule(null,null,null,null,1,2); List<StaticRuleObject> staticRules= staticRuleService.queryStaticRule(null,null,null,null,null,1,2);
for (StaticRuleObject staticRule : staticRules) { for (StaticRuleObject staticRule : staticRules) {
ruleIds.add(staticRule.getStaticRuleId()); ruleIds.add(staticRule.getStaticRuleId());
} }