From bcd9cb83cbbb36217cb99ced7bf1fa3cb26bd53c Mon Sep 17 00:00:00 2001
From: zjj <3272660260@qq.com>
Date: 星期一, 28 四月 2025 16:59:51 +0800
Subject: [PATCH] #wms端生成任务优化

---
 rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/MatnrGroupController.java |   47 ++++++++++++++++++++++++++++-------------------
 1 files changed, 28 insertions(+), 19 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/MatnrGroupController.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/MatnrGroupController.java
index fbc7197..ff0cb65 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/MatnrGroupController.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/MatnrGroupController.java
@@ -17,12 +17,14 @@
 import com.vincent.rsf.server.manager.service.MatnrGroupService;
 import com.vincent.rsf.server.system.controller.BaseController;
 import io.swagger.annotations.Api;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletResponse;
 import java.util.*;
+import java.util.stream.Collectors;
 
 @Api(tags = "鐗╂枡鍒嗙粍")
 @RestController
@@ -63,25 +65,32 @@
     @PostMapping("/matnrGroup/save")
     public R save(@RequestBody MatnrGroup matnrGroup) {
         if (Objects.isNull(matnrGroup)) {
-            throw new CoolException("鍙傛暟涓嶈兘涓虹┖锛侊紒");
+            return R.error("鍙傛暟涓嶈兘涓虹┖锛侊紒");
         }
+
         if (Objects.isNull(matnrGroup.getName())) {
-            throw new CoolException("鍒嗙粍鍚嶇О涓嶈兘涓虹┖锛侊紒");
+            return R.error("鍒嗙粍鍚嶇О涓嶈兘涓虹┖锛侊紒");
         }
         if (Objects.isNull(matnrGroup.getCode())) {
-            throw new CoolException("鍒嗙粍缂栫爜涓嶈兘涓虹┖锛侊紒");
+            return R.error("鍒嗙粍缂栫爜涓嶈兘涓虹┖锛侊紒");
         }
         matnrGroup.setCreateBy(getLoginUserId());
         matnrGroup.setUpdateBy(getLoginUserId());
+
         List<MatnrGroup> list = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>()
+                .eq(!Objects.isNull(matnrGroup.getParCode()), MatnrGroup::getParCode, matnrGroup.getParCode())
                 .eq(MatnrGroup::getCode, matnrGroup.getCode()));
         if (!list.isEmpty()) {
-            throw new CoolException("鐗╂枡鍒嗙粍缂栫爜涓嶈兘閲嶅锛侊紒");
+            return R.error("鐗╂枡鍒嗙粍缂栫爜涓嶈兘閲嶅锛侊紒");
         }
         List<MatnrGroup> groups = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>()
                 .eq(MatnrGroup::getName, matnrGroup.getName()));
         if (!groups.isEmpty()) {
-            throw new CoolException("鍒嗙粍宸插瓨鍦紝璇峰嬁閲嶅娣诲姞锛侊紒");
+            return R.error("鍒嗙粍宸插瓨鍦紝璇峰嬁閲嶅娣诲姞锛侊紒");
+        }
+        //鍒ゆ柇涓婄骇缂栫爜鏄惁涓虹┖
+        if (!Objects.isNull(matnrGroup.getParCode()) && !StringUtils.isBlank(matnrGroup.getParCode())) {
+            matnrGroup.setParCode(matnrGroup.getParCode() + matnrGroup.getCode());
         }
         if (!matnrGroupService.save(matnrGroup)) {
             return R.error("Save Fail");
@@ -93,18 +102,24 @@
     @OperationLog("Update 鐗╂枡鍒嗙被琛�")
     @PostMapping("/matnrGroup/update")
     public R update(@RequestBody MatnrGroup matnrGroup) {
+
         matnrGroup.setUpdateBy(getLoginUserId());
         MatnrGroup matGroup = matnrGroupService.getById(matnrGroup.getId());
+        if (Objects.isNull(matGroup)) {
+            return R.error("鐗╂枡鍒嗙粍涓嶅瓨鍦紒锛�");
+        }
         if (!matnrGroup.getName().equals(matGroup.getName())) {
             List<MatnrGroup> areasList = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>().eq(MatnrGroup::getName, matnrGroup.getName()));
             if (!areasList.isEmpty()) {
-                throw new CoolException("浠撳簱鍚嶅凡瀛樺湪锛侊紒");
+                return R.error("浠撳簱鍚嶅凡瀛樺湪锛侊紒");
             }
         }
-        if (!matnrGroup.getCode().equals(matGroup.getCode())) {
-            List<MatnrGroup> areasList = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>().eq(MatnrGroup::getCode, matnrGroup.getCode()));
+        if (!Objects.isNull(matnrGroup.getCode()) && !matnrGroup.getCode().equals(matGroup.getCode())) {
+            List<MatnrGroup> areasList = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>()
+                            .eq(!Objects.isNull(matnrGroup.getParCode()), MatnrGroup::getParCode, matnrGroup.getParCode())
+                    .eq(MatnrGroup::getCode, matnrGroup.getCode()));
             if (!areasList.isEmpty()) {
-                throw new CoolException("浠撳簱缂栫爜宸插瓨鍦紒锛�");
+                return R.error("浠撳簱缂栫爜宸插瓨鍦紒锛�");
             }
         }
 
@@ -141,18 +156,12 @@
     @PreAuthorize("hasAuthority('manager:matnrGroup:list')")
     @PostMapping("/matnrGroup/tree")
     public R tree(@RequestBody(required = false) Map<String, Object> map) {
-        List<MatnrGroup> matnrs = new ArrayList<>();
         if (Objects.isNull(map)) {
-            matnrs = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>().orderByAsc(MatnrGroup::getCode));
-        } else {
-            if (Objects.isNull(map.get("condition"))) {
-                return R.ok("condition鍙傛暟涓嶈兘涓虹┖锛侊紒");
-            }
-            matnrs = matnrGroupService.list(new LambdaQueryWrapper<MatnrGroup>()
-                    .like(MatnrGroup::getName, map.get("condition")).orderByAsc(MatnrGroup::getCode));
+            return R.error("鍙傛暟涓嶈兘涓虹┖锛侊紒");
         }
-        List<MatnrGroup> treeData = Utils.toTreeData(matnrs, 0L, MatnrGroup::getParentId, MatnrGroup::getId, MatnrGroup::setChildren);
-        return R.ok().add(treeData);
+        List<MatnrGroup> matnrs = matnrGroupService.getTreeData(map);
+
+        return R.ok().add(matnrs);
     }
 
 

--
Gitblit v1.9.1