shiyue před 1 rokem
rodič
revize
756f1361ae

+ 7 - 0
energy-manage-service/src/main/java/com/energy/manage/service/controller/windenginegroup/WindEngineGroupController.java

@@ -2,6 +2,7 @@ package com.energy.manage.service.controller.windenginegroup;
 
 
 import com.energy.manage.common.base.Page;
+import com.energy.manage.common.enums.IdPrefixEnum;
 import com.energy.manage.common.reponse.ResultResp;
 import com.energy.manage.service.config.annotations.UserLoginToken;
 import com.energy.manage.service.domain.dto.windenginegroup.*;
@@ -37,6 +38,9 @@ public class WindEngineGroupController extends BaseServiceController {
     @PostMapping(value = "/createWindEngineGroup")
     @ApiOperation(value = "创建风机机组")
     public ResultResp createWindEngineGroup(@RequestBody @Valid WindEngineGroupCreateDto windEngineGroupCreateDto) {
+        if(!windEngineGroupCreateDto.getFieldCode().contains(IdPrefixEnum.WIND_FIELD_NUMBER.getCode())){
+            return ResultResp.FAIL("未绑定风场信息!");
+        }
         windEngineGroupCreateDto.setCreateBy(getUserId());
         boolean flg = windEngineGroupService.createWindEngineGroup(windEngineGroupCreateDto);
         return flg ? ResultResp.SUCCESS() : ResultResp.FAIL();
@@ -47,6 +51,9 @@ public class WindEngineGroupController extends BaseServiceController {
     @PostMapping(value = "/windEngineGroupImportData")
     @ApiOperation(value = "导入风机信息")
     public ResultResp windEngineGroupImportData(@RequestPart MultipartFile file, @RequestParam("fieldCode") String fieldCode) {
+        if(!fieldCode.contains(IdPrefixEnum.WIND_FIELD_NUMBER.getCode())){
+            return ResultResp.FAIL("未绑定风场信息!");
+        }
         return  windEngineGroupService.windEngineGroupImportData(file, fieldCode);
     }