From 646ec3901cba867fd539146b26980c2ad510560c Mon Sep 17 00:00:00 2001
From: wuzeyu <1223318623@qq.com>
Date: 星期五, 31 五月 2024 15:28:35 +0800
Subject: [PATCH] 田间灌溉项目管理  新增 修改轮罐组绑定的灌溉单元接口

---
 pipIrr-platform/pipIrr-web/pipIrr-web-irrigate/src/main/java/com/dy/pipIrrIrrigate/irrigateGroup/IrrigateGroupCtrl.java |  131 ++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 116 insertions(+), 15 deletions(-)

diff --git a/pipIrr-platform/pipIrr-web/pipIrr-web-irrigate/src/main/java/com/dy/pipIrrIrrigate/irrigateGroup/IrrigateGroupCtrl.java b/pipIrr-platform/pipIrr-web/pipIrr-web-irrigate/src/main/java/com/dy/pipIrrIrrigate/irrigateGroup/IrrigateGroupCtrl.java
index e79df64..afb4ec0 100644
--- a/pipIrr-platform/pipIrr-web/pipIrr-web-irrigate/src/main/java/com/dy/pipIrrIrrigate/irrigateGroup/IrrigateGroupCtrl.java
+++ b/pipIrr-platform/pipIrr-web/pipIrr-web-irrigate/src/main/java/com/dy/pipIrrIrrigate/irrigateGroup/IrrigateGroupCtrl.java
@@ -31,10 +31,7 @@
 import org.springframework.validation.BindingResult;
 import org.springframework.web.bind.annotation.*;
 
-import java.util.List;
-import java.util.Map;
-import java.util.Objects;
-import java.util.Optional;
+import java.util.*;
 
 /**
  * @author :WuZeYu
@@ -250,14 +247,24 @@
     @PostMapping(path = "add_unit", consumes = MediaType.APPLICATION_JSON_VALUE)
     @Transactional(rollbackFor = Exception.class)
     @SsoAop()
-    public BaseResponse<Boolean> addGroupUnit(@RequestBody @Valid IrGroupUnit po, BindingResult bindingResult) {
+    public BaseResponse<Boolean> addGroupUnit(@RequestBody @Valid Map<String,Object> po, BindingResult bindingResult) {
         if (bindingResult != null && bindingResult.hasErrors()) {
             return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
         }
+        Long groupId = Long.parseLong(po.get("groupId").toString());
+        Long operator = Long.parseLong(po.get("operator").toString());
+        List<Long> unitIds = (List<Long>) po.get("unitIds");
         try {
-            Integer rec = Optional.ofNullable(irrigateGroupSv.addGroupUnit(po)).orElse(0);
-            if (rec == 0) {
-                return BaseResponseUtils.buildFail(IrrigateResultCode.ADD_PROJECT_FAIL.getMessage());
+            for (int i = 0; i < unitIds.size(); i++) {
+                Long unitId = unitIds.get(i);
+                IrGroupUnit irGroupUnit = new IrGroupUnit();
+                irGroupUnit.setGroupId(groupId);
+                irGroupUnit.setUnitId(unitId);
+                irGroupUnit.setOperator(operator);
+                Integer rec = Optional.ofNullable(irrigateGroupSv.addGroupUnit(irGroupUnit)).orElse(0);
+                if (rec == 0) {
+                    return BaseResponseUtils.buildFail(IrrigateResultCode.ADD_PROJECT_FAIL.getMessage());
+                }
             }
         } catch (Exception e) {
             log.error("杞亴缁勬坊鍔犵亴婧夊崟鍏冨紓甯�", e);
@@ -267,7 +274,7 @@
     }
 
     /**
-     * 閫昏緫鍒犻櫎 杞亴缁勭Щ闄ょ亴婧夊崟鍏�
+     *  杞亴缁勭Щ闄ょ亴婧夊崟鍏�
      * @param
      * @return
      */
@@ -283,15 +290,26 @@
     @PostMapping(path = "remove_unit")
     @Transactional(rollbackFor = Exception.class)
     @SsoAop()
-    public BaseResponse<Boolean> deleteGroupUnit(@RequestBody @Valid IrGroupUnit po, BindingResult bindingResult) {
+    public BaseResponse<Boolean> deleteGroupUnit(@RequestBody @Valid Map<String,Object> po, BindingResult bindingResult) {
         if (bindingResult != null && bindingResult.hasErrors()) {
             return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
         }
+        Long groupId = Long.parseLong(po.get("groupId").toString());
+        Long operator = Long.parseLong(po.get("operator").toString());
+        List<Long> unitIds = (List<Long>) po.get("unitIds");
         try {
-            Integer rec = Optional.ofNullable(irrigateGroupSv.deleteGroupUnit(po)).orElse(0);
-            if (rec == 0) {
-                return BaseResponseUtils.buildFail(IrrigateResultCode.DELETE_PROJECT_FAIL.getMessage());
+            for (int i = 0; i < unitIds.size(); i++) {
+                Long unitId = unitIds.get(i);
+                IrGroupUnit irGroupUnit = new IrGroupUnit();
+                irGroupUnit.setGroupId(groupId);
+                irGroupUnit.setUnitId(unitId);
+                irGroupUnit.setOperator(operator);
+                Integer rec = Optional.ofNullable(irrigateGroupSv.deleteGroupUnit(irGroupUnit)).orElse(0);
+                if (rec == 0) {
+                    return BaseResponseUtils.buildFail(IrrigateResultCode.DELETE_PROJECT_FAIL.getMessage());
+                }
             }
+
         } catch (Exception e) {
             log.error("杞亴缁勭Щ闄ょ亴婧夊崟鍏冨紓甯�", e);
             return BaseResponseUtils.buildException(e.getMessage());
@@ -301,13 +319,96 @@
 
     @GetMapping(path = "bind_units")
     @SsoAop()
-    public BaseResponse<List<Long>> getGroupBindUnits(Long id) {
+    public BaseResponse<List<String>> getGroupBindUnits(Long id) {
         try {
             List<Long> res = irrigateGroupSv.getGroupBindUnits(id);
-            return BaseResponseUtils.buildSuccess(res);
+            ArrayList<Object> list = new ArrayList<>();
+            for (int i = 0; i < res.size(); i++) {
+                String aLong = res.get(i).toString();
+                list.add(aLong);
+            }
+            return BaseResponseUtils.buildSuccess(list);
         } catch (Exception e) {
             log.error("鑾峰彇涓�涓疆鐏岀粍缁戝畾鐏屾簤鍗曞厓璁板綍寮傚父", e);
             return BaseResponseUtils.buildException(e.getMessage());
         }
     }
+
+    @GetMapping(path = "notBind_units")
+    @SsoAop()
+    public BaseResponse<List<String>> getNotBindUnits() {
+        try {
+            List<Long> res = irrigateGroupSv.getNotBindUnits();
+            ArrayList<Object> list = new ArrayList<>();
+            for (int i = 0; i < res.size(); i++) {
+                String aLong = res.get(i).toString();
+                list.add(aLong);
+            }
+            return BaseResponseUtils.buildSuccess(list);
+        } catch (Exception e) {
+            log.error("鑾峰彇鏈粦瀹氳疆鐏岀粍鐨勭亴婧夊崟鍏僫d璁板綍寮傚父", e);
+            return BaseResponseUtils.buildException(e.getMessage());
+        }
+    }
+
+
+
+    /**
+     * 淇敼杞亴缁勭粦瀹氱殑鐏屾簤鍗曞厓
+     *
+     * @param po            杞亴缁勭亴婧夊崟鍏冧腑闂磋〃
+     * @param bindingResult
+     * @return 娣诲姞鏄惁鎴愬姛
+     */
+    @Operation(summary = "淇敼杞亴缁勭粦瀹氱殑鐏屾簤鍗曞厓", description = "淇敼杞亴缁勭粦瀹氱殑鐏屾簤鍗曞厓")
+    @ApiResponses(value = {
+            @ApiResponse(
+                    responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE,
+                    description = "鎿嶄綔缁撴灉锛歵rue锛氭垚鍔燂紝false锛氬け璐ワ紙BaseResponse.content锛�",
+                    content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE,
+                            schema = @Schema(implementation = Boolean.class))}
+            )
+    })
+    @PostMapping(path = "updateBindUnits", consumes = MediaType.APPLICATION_JSON_VALUE)
+    @Transactional(rollbackFor = Exception.class)
+    @SsoAop()
+    public BaseResponse<Boolean> updateBindUnits(@RequestBody @Valid Map<String,Object> po, BindingResult bindingResult){
+        if (bindingResult != null && bindingResult.hasErrors()) {
+            return BaseResponseUtils.buildFail(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
+        }
+        Long groupId = Long.parseLong(po.get("groupId").toString());
+        Long operator = Long.parseLong(po.get("operator").toString());
+        List<Long> newUnitIds = (List<Long>) po.get("unitIds");
+
+        List<Long> oldUnitIds = irrigateGroupSv.getGroupBindUnits(groupId);
+
+        try {
+            for (int j = 0; j < oldUnitIds.size(); j++) {
+                Long oldUnitId = oldUnitIds.get(j);
+                IrGroupUnit irGroupUnit1 = new IrGroupUnit();
+                irGroupUnit1.setGroupId(groupId);
+                irGroupUnit1.setUnitId(oldUnitId);
+                irGroupUnit1.setOperator(operator);
+                Integer rec = Optional.ofNullable(irrigateGroupSv.deleteGroupUnit(irGroupUnit1)).orElse(0);
+                if (rec == 0) {
+                    return BaseResponseUtils.buildFail(IrrigateResultCode.ADD_PROJECT_FAIL.getMessage());
+                }
+            }
+            for (int i = 0; i < newUnitIds.size(); i++) {
+                Long newUnitId = newUnitIds.get(i);
+                IrGroupUnit irGroupUnit2 = new IrGroupUnit();
+                irGroupUnit2.setGroupId(groupId);
+                irGroupUnit2.setUnitId(newUnitId);
+                irGroupUnit2.setOperator(operator);
+                Integer rec = Optional.ofNullable(irrigateGroupSv.addGroupUnit(irGroupUnit2)).orElse(0);
+                if (rec == 0) {
+                    return BaseResponseUtils.buildFail(IrrigateResultCode.ADD_PROJECT_FAIL.getMessage());
+                }
+            }
+        } catch (Exception e) {
+            log.error("淇敼杞亴缁勭粦瀹氱殑鐏屾簤鍗曞厓寮傚父", e);
+            return BaseResponseUtils.buildException(e.getMessage());
+        }
+        return BaseResponseUtils.buildSuccess(true);
+    }
 }

--
Gitblit v1.8.0