|
@@ -200,29 +200,21 @@ public class SceneEvaluationRuleService {
|
|
|
SceneComplexityPO sceneComplexityPO = new SceneComplexityPO();
|
|
|
sceneComplexityPO.setMinComplexity(param.getMinComplexity());
|
|
|
sceneComplexityPO.setMaxComplexity(param.getMaxComplexity());
|
|
|
- SceneEvaluationEnum comEnums = SceneEvaluationEnum.matchLevelEnumByLevel(param.getComplexityLevel());
|
|
|
- sceneComplexityPO.setComplexityLevel(comEnums.getLevelValue());
|
|
|
+ sceneComplexityPO.setComplexityLevel(SceneEvaluationEnum.matchLevelEnumByLevel(param.getComplexityLevel()));
|
|
|
sceneComplexityPO.setSceneIdList(param.getSceneIdList());
|
|
|
// 查询复杂度的符合条件的第一条数据
|
|
|
List<SceneComplexityPO> sceneReferenceLibCSPOS = sceneComplexityMapper.selectSceneIdsByEvaluation(sceneComplexityPO);
|
|
|
- sceneReferenceLibCSPOS.forEach(po -> {
|
|
|
- SceneEvaluationEnum sceneEvaluationEnum = SceneEvaluationEnum.matchLevelEnumByLevelValue(po.getComplexityLevel());
|
|
|
- po.setComplexityLevel(sceneEvaluationEnum.getLevel());
|
|
|
- });
|
|
|
+ sceneReferenceLibCSPOS.forEach(po -> po.setComplexityLevel(SceneEvaluationEnum.matchLevelEnumByLevelValue(po.getComplexityLevel())));
|
|
|
|
|
|
Map<String, SceneComplexityPO> complexityPOMap = sceneReferenceLibCSPOS.stream().collect(Collectors.toMap(SceneComplexityPO::getSceneId, Function.identity()));
|
|
|
SceneRiskPO sceneRiskPO = new SceneRiskPO();
|
|
|
sceneRiskPO.setMaxRisk(param.getMaxRisk());
|
|
|
sceneRiskPO.setMinRisk(param.getMinRisk());
|
|
|
- SceneEvaluationEnum riskEnums = SceneEvaluationEnum.matchLevelEnumByLevel(param.getRiskLevel());
|
|
|
- sceneRiskPO.setRiskLevel(riskEnums.getLevelValue());
|
|
|
+ sceneRiskPO.setRiskLevel(SceneEvaluationEnum.matchLevelEnumByLevel(param.getRiskLevel()));
|
|
|
sceneRiskPO.setSceneIdList(param.getSceneIdList());
|
|
|
// 查询复杂度的符合条件的第一条数据
|
|
|
List<SceneRiskPO> sceneReferenceLibCRPOS = sceneRiskMapper.selectSceneIdsByEvaluation(sceneRiskPO);
|
|
|
- sceneReferenceLibCRPOS.forEach(po -> {
|
|
|
- SceneEvaluationEnum sceneEvaluationEnum = SceneEvaluationEnum.matchLevelEnumByLevelValue(po.getRiskLevel());
|
|
|
- po.setRiskLevel(sceneEvaluationEnum.getLevel());
|
|
|
- });
|
|
|
+ sceneReferenceLibCRPOS.forEach(po -> po.setRiskLevel(SceneEvaluationEnum.matchLevelEnumByLevelValue(po.getRiskLevel())));
|
|
|
Map<String, SceneRiskPO> riskPOMap = sceneReferenceLibCRPOS.stream().collect(Collectors.toMap(SceneRiskPO::getSceneId, Function.identity()));
|
|
|
return SceneEvaluationOperatePO.builder().sceneComplexityPOMap(complexityPOMap).sceneRiskPOMap(riskPOMap).build();
|
|
|
}
|
|
@@ -345,7 +337,6 @@ public class SceneEvaluationRuleService {
|
|
|
if (StringUtils.equals(sceneEvaluationComputeParam.getComputeType(), DictConstants.COMPLEXITY)) {
|
|
|
String complexity = rootNode.path("复杂度").asText();
|
|
|
String complexityLevel = rootNode.path("复杂度等级").asText();
|
|
|
- SceneEvaluationEnum sceneEvaluationEnum = matchLevelEnumByLevel(complexityLevel);
|
|
|
SceneComplexityPO sceneComplexityPO = new SceneComplexityPO();
|
|
|
sceneComplexityPO.setSceneId(sceneEvaluationComputeParam.getSceneId());
|
|
|
sceneComplexityPO.setComplexityId(StringUtil.getRandomUUID());
|
|
@@ -353,7 +344,7 @@ public class SceneEvaluationRuleService {
|
|
|
sceneComplexityPO.setRuleId(ruleId);
|
|
|
sceneComplexityPO.setTaskId(sceneEvaluationComputeParam.getTaskId());
|
|
|
sceneComplexityPO.setComplexity(complexity);
|
|
|
- sceneComplexityPO.setComplexityLevel(sceneEvaluationEnum.getLevelValue());
|
|
|
+ sceneComplexityPO.setComplexityLevel(matchLevelEnumByLevel(complexityLevel));
|
|
|
sceneComplexityPO.setIsDeleted(DictConstants.IS_NOT_DELETED);
|
|
|
sceneComplexityPO.setCreateUserId(AuthUtil.getCurrentUserId());
|
|
|
sceneComplexityPO.setCreateTime(TimeUtil.getNowForMysql());
|
|
@@ -361,7 +352,6 @@ public class SceneEvaluationRuleService {
|
|
|
} else {
|
|
|
String risk = rootNode.path("危险度").asText();
|
|
|
String riskLevel = rootNode.path("危险度等级").asText();
|
|
|
- SceneEvaluationEnum sceneEvaluationEnum = matchLevelEnumByLevel(riskLevel);
|
|
|
SceneRiskPO sceneRiskPO = new SceneRiskPO();
|
|
|
sceneRiskPO.setSceneId(sceneEvaluationComputeParam.getSceneId());
|
|
|
sceneRiskPO.setRuleId(StringUtil.getRandomUUID());
|
|
@@ -369,7 +359,7 @@ public class SceneEvaluationRuleService {
|
|
|
sceneRiskPO.setRuleId(ruleId);
|
|
|
sceneRiskPO.setTaskId(sceneEvaluationComputeParam.getTaskId());
|
|
|
sceneRiskPO.setRisk(risk);
|
|
|
- sceneRiskPO.setRiskLevel(sceneEvaluationEnum.getLevelValue());
|
|
|
+ sceneRiskPO.setRiskLevel(matchLevelEnumByLevel(riskLevel));
|
|
|
sceneRiskPO.setIsDeleted(DictConstants.IS_NOT_DELETED);
|
|
|
sceneRiskPO.setCreateUserId(AuthUtil.getCurrentUserId());
|
|
|
sceneRiskPO.setCreateTime(TimeUtil.getNowForMysql());
|