量水堰和闸阀新增主键赋值

master
wany 2024-08-16 13:43:18 +08:00
parent 0d52b07e50
commit 956c1548d9
2 changed files with 4 additions and 6 deletions

View File

@ -1,6 +1,7 @@
package com.gunshi.project.xyt.controller;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@ -38,9 +39,7 @@ public class AttGateValveController {
@Operation(summary = "新增")
@PostMapping("/insert")
public R<AttGateValve> insert(@Validated(Insert.class) @RequestBody AttGateValve dto) {
if (Objects.nonNull(service.getById(dto.getValveCode()))){
throw new IllegalArgumentException("当前闸阀编码已存在");
}
dto.setValveCode(String.valueOf(IdWorker.getId()));
boolean result = service.save(dto);
return R.ok(result ? dto : null);
}

View File

@ -1,6 +1,7 @@
package com.gunshi.project.xyt.controller;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@ -43,9 +44,7 @@ public class AttMeaWeirController {
@Operation(summary = "新增")
@PostMapping("/insert")
public R<AttMeaWeir> insert(@Validated(Insert.class) @RequestBody AttMeaWeir dto) {
if (Objects.nonNull(service.getById(dto.getWeirCode()))){
throw new IllegalArgumentException("当前闸阀编码已存在");
}
dto.setWeirCode(String.valueOf(IdWorker.getId()));
if (StringUtils.isNotBlank(dto.getAdcd()) && Objects.isNull(stAddvcdDService.getById(dto.getAdcd()))){
throw new IllegalArgumentException("当前行政区划不存在");
}