Przeglądaj źródła

隐患排查调整,添加所属队伍

Zimo 6 dni temu
rodzic
commit
cacba37e00

+ 13 - 1
yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/qhse/hazard/IotHazardServiceImpl.java

@@ -49,7 +49,12 @@ public class IotHazardServiceImpl implements IotHazardService {
         iotHazard.setDeleted(false);
         //未整改
         iotHazard.setStatus("todo");
-        Long deptId = SecurityFrameworkUtils.getLoginUserDeptId();
+        Long deptId;
+        if (createReqVO.getDeptId()==null) {
+            deptId = SecurityFrameworkUtils.getLoginUserDeptId();
+        } else {
+            deptId = createReqVO.getDeptId();
+        }
         iotHazard.setDeptId(deptId);
         if (Objects.nonNull(createReqVO.getClassify())&& StringUtils.isNotBlank(createReqVO.getClassify())) {
             IotHazardTypeDO iotHazardTypeDO = iotHazardTypeMapper.selectById(Long.valueOf(createReqVO.getClassify()));
@@ -65,6 +70,13 @@ public class IotHazardServiceImpl implements IotHazardService {
 
     @Override
     public void updateIotHazard(IotHazardSaveReqVO updateReqVO) {
+        Long deptId;
+        if (updateReqVO.getDeptId()==null) {
+            deptId = SecurityFrameworkUtils.getLoginUserDeptId();
+        } else {
+            deptId = updateReqVO.getDeptId();
+        }
+        updateReqVO.setDeptId(deptId);
         // 校验存在
         validateIotHazardExists(updateReqVO.getId());
         // 更新