From f1f506172a3edff8d31e4db77b8940f115660919 Mon Sep 17 00:00:00 2001
From: songhuangfeng123 <shf18767906695@163.com>
Date: 星期二, 20 九月 2022 16:25:43 +0800
Subject: [PATCH] fix

---
 emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/service/impl/EmergencyPlanServiceImpl.java |   39 +++++++++++++++++++++++++++++++++++++++
 1 files changed, 39 insertions(+), 0 deletions(-)

diff --git a/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/service/impl/EmergencyPlanServiceImpl.java b/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/service/impl/EmergencyPlanServiceImpl.java
index 01766ef..6f4a884 100644
--- a/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/service/impl/EmergencyPlanServiceImpl.java
+++ b/emergency/emergency-service/src/main/java/com/gkhy/safePlatform/emergency/service/impl/EmergencyPlanServiceImpl.java
@@ -6,10 +6,12 @@
 import com.gkhy.safePlatform.commons.enums.ResultCodes;
 import com.gkhy.safePlatform.commons.query.PageQuery;
 import com.gkhy.safePlatform.commons.utils.BeanCopyUtils;
+import com.gkhy.safePlatform.commons.utils.StringUtils;
 import com.gkhy.safePlatform.commons.vo.ResultVO;
 import com.gkhy.safePlatform.commons.vo.SearchResultVO;
 import com.gkhy.safePlatform.emergency.entity.*;
 import com.gkhy.safePlatform.emergency.enums.ApproveStatus;
+import com.gkhy.safePlatform.emergency.enums.EmergencyPlanTypeEnum;
 import com.gkhy.safePlatform.emergency.enums.EmergencyResultCodes;
 import com.gkhy.safePlatform.emergency.excepiton.EmergencyException;
 import com.gkhy.safePlatform.emergency.model.dto.req.*;
@@ -95,6 +97,13 @@
 
     @Override
     public ResultVO addEmergencyPlan(Long uid, EmergencyPlanReqDTO emergencyPlanReqDTO) {
+        // 验证必填
+        checkRequired(emergencyPlanReqDTO);
+        // 验证枚举
+        checkEnum(emergencyPlanReqDTO);
+
+
+
         Date nowDate = new Date();
         // 新增应急预案
         EmergencyPlanInfo emergencyPlanInfo = new EmergencyPlanInfo();
@@ -102,6 +111,7 @@
         emergencyPlanInfo.setDelFlag(false);
         emergencyPlanInfo.setCreateUid(uid);
         emergencyPlanInfo.setGmtCreate(nowDate);
+        emergencyPlanInfo.setAbolishStatus(false);
         emergencyPlanInfoService.addEmergencyPlan(emergencyPlanInfo);
         // 新增应急预案区域表
         if (!CollectionUtils.isEmpty(emergencyPlanReqDTO.getAreaList())) {
@@ -231,6 +241,11 @@
 
     @Override
     public ResultVO updateEmergencyPlan(Long uid, EmergencyPlanReqDTO emergencyPlanReqDTO) {
+        // 验证必填
+        checkRequired(emergencyPlanReqDTO);
+        // 验证枚举
+        checkEnum(emergencyPlanReqDTO);
+
         Date nowDate = new Date();
         // 查询是否存在
         EmergencyPlanInfoDetailDO emergencyPlanInfoDetailDO = emergencyPlanInfoService.selectEmergencyPlanById(emergencyPlanReqDTO.getId());
@@ -461,4 +476,28 @@
             emergencyPlanTeamInfoService.deleteEmergencyPlanTeamByPlanId(PlanId);
         }
     }
+
+    /**
+     * 验证必填项
+     * @return
+     */
+    private void checkRequired(EmergencyPlanReqDTO emergencyPlanReqDTO) {
+        //名称
+        if (StringUtils.isBlank(emergencyPlanReqDTO.getName())) {
+            throw new EmergencyException(EmergencyResultCodes.PLAN_NAME_NULL);
+        }
+    }
+
+    /**
+     * 验证枚举
+     */
+    private void checkEnum(EmergencyPlanReqDTO emergencyPlanReqDTO){
+        if (EmergencyPlanTypeEnum.getByCode(emergencyPlanReqDTO.getType())==null){
+            throw new EmergencyException(EmergencyResultCodes.PLAN_ENUM_TYPE_NOT_EXIST);
+        }
+        if (EmergencyPlanTypeEnum.getByCode(emergencyPlanReqDTO.getLevel())==null){
+            throw new EmergencyException(EmergencyResultCodes.PLAN_ENUM_LEVEL_NOT_EXIST);
+        }
+    }
+
 }

--
Gitblit v1.9.2