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("当前行政区划不存在");
         }