From 956c1548d91132d5270a573315dc16acb5227f96 Mon Sep 17 00:00:00 2001
From: wany <13995595726@qq.com>
Date: Fri, 16 Aug 2024 13:43:18 +0800
Subject: [PATCH] =?UTF-8?q?=E9=87=8F=E6=B0=B4=E5=A0=B0=E5=92=8C=E9=97=B8?=
 =?UTF-8?q?=E9=98=80=E6=96=B0=E5=A2=9E=E4=B8=BB=E9=94=AE=E8=B5=8B=E5=80=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../project/xyt/controller/AttGateValveController.java       | 5 ++---
 .../gunshi/project/xyt/controller/AttMeaWeirController.java  | 5 ++---
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/gunshi/project/xyt/controller/AttGateValveController.java b/src/main/java/com/gunshi/project/xyt/controller/AttGateValveController.java
index 9600b23..bcfc7c6 100644
--- a/src/main/java/com/gunshi/project/xyt/controller/AttGateValveController.java
+++ b/src/main/java/com/gunshi/project/xyt/controller/AttGateValveController.java
@@ -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);
     }
diff --git a/src/main/java/com/gunshi/project/xyt/controller/AttMeaWeirController.java b/src/main/java/com/gunshi/project/xyt/controller/AttMeaWeirController.java
index 62709ae..6768ae0 100644
--- a/src/main/java/com/gunshi/project/xyt/controller/AttMeaWeirController.java
+++ b/src/main/java/com/gunshi/project/xyt/controller/AttMeaWeirController.java
@@ -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("当前行政区划不存在");
         }