diff --git a/src/main/java/com/gunshi/project/hsz/controller/StZqrlBController.java b/src/main/java/com/gunshi/project/hsz/controller/StZqrlBController.java index 1d7e5e5..441fd7d 100644 --- a/src/main/java/com/gunshi/project/hsz/controller/StZqrlBController.java +++ b/src/main/java/com/gunshi/project/hsz/controller/StZqrlBController.java @@ -74,15 +74,15 @@ public class StZqrlBController { @Operation(summary = "新增") @PostMapping("/insert") public R insert(@Validated(Insert.class) @RequestBody StZqrlB dto) { -// if (StringUtils.isNotBlank(dto.getResCode())){ -// AttResBase one = resBaseService.lambdaQuery().eq(AttResBase::getResCode, dto.getResCode()).one(); + if (StringUtils.isNotBlank(dto.getResCode())){ + AttResBase one = resBaseService.list().get(0); // if (Objects.isNull(one)){ // throw new IllegalArgumentException("水库编码不存在"); // } -// if (StringUtils.isBlank(dto.getStcd())){ -// dto.setStcd(one.getStcd()); -// } -// } + if (StringUtils.isBlank(dto.getStcd())){ + dto.setStcd(one.getStcd()); + } + } if (StringUtils.isNotBlank(dto.getStcd()) && Objects.isNull(stStbprpBService.getById(dto.getStcd()))){ throw new IllegalArgumentException("测站编码不存在"); }