修改

shuili-system/src/main/java/com/kms/yg/df/controller/BsSgcDfHzxxController.java

@@ -3,6 +3,7 @@ package com.kms.yg.df.controller;
 import java.util.Arrays;
 import java.util.List;
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.kms.system.service.SysDeptService;
 import com.kms.system.service.SysXzqhService;
 import com.kms.yg.df.domain.BsSgcDfHzxx;
@@ -105,7 +106,10 @@ public class BsSgcDfHzxxController extends BaseController
     public AjaxResult add(@RequestBody BsSgcDfHzxx bsSgcSzHzxx)
     {
         BaseEntityUtils.preInsert(bsSgcSzHzxx);
-        BsSgcDfSafeJbxx byId = bsSgcSzSafeJbxxService.getById(bsSgcSzHzxx.getDikeId());
+        String dikeCode = bsSgcSzHzxx.getDikeCode();
+        BsSgcDfSafeJbxx byId = bsSgcSzSafeJbxxService.getOne(Wrappers.lambdaQuery(BsSgcDfSafeJbxx.class)
+                .eq(BsSgcDfSafeJbxx::getDikeCode, dikeCode));
+
         byId.setHzStatus("1");
         bsSgcSzSafeJbxxService.updateById(byId);
         return toAjax(bsSgcSzHzxxService.save(bsSgcSzHzxx));
@@ -120,8 +124,9 @@ public class BsSgcDfHzxxController extends BaseController
     public AjaxResult edit(@RequestBody BsSgcDfHzxx bsSgcSzHzxx)
     {
         String status = bsSgcSzHzxx.getStatus();
-        String wagaId = bsSgcSzHzxx.getDikeId();
-        BsSgcDfSafeJbxx byId = bsSgcSzSafeJbxxService.getById(wagaId);
+        String dikeCode = bsSgcSzHzxx.getDikeCode();
+        BsSgcDfSafeJbxx byId = bsSgcSzSafeJbxxService.getOne(Wrappers.lambdaQuery(BsSgcDfSafeJbxx.class)
+                .eq(BsSgcDfSafeJbxx::getDikeCode, dikeCode));
         byId.setHzStatus(status);
         if (status.equals("2")) {
             byId.setSecurityCategory(bsSgcSzHzxx.getSecurityCategoryNow());