From 0f69561e397093b5165c4aac58530721d5c62178 Mon Sep 17 00:00:00 2001
From: zjj <3272660260@qq.com>
Date: 星期一, 25 十一月 2024 11:03:16 +0800
Subject: [PATCH] #led

---
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java |   56 +++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 47 insertions(+), 9 deletions(-)

diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java
index f4a6da7..791f880 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java
@@ -12,16 +12,13 @@
 import com.zy.asrs.wcs.system.service.MenuService;
 import com.zy.asrs.wcs.utils.ExcelUtil;
 import com.zy.asrs.wcs.utils.NodeUtils;
-import com.zy.asrs.wcs.utils.Utils;
+import com.zy.asrs.wcs.utils.CommonUtils;
 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.ArrayList;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 @RestController
 @RequestMapping("/api")
@@ -47,8 +44,17 @@
     @PreAuthorize("hasAuthority('system:menu:list')")
     @PostMapping("/menu/tree")
     public R tree(@RequestBody Map<String, Object> map) {
-        List<Menu> menus = menuService.list();
-        return R.ok().add(Utils.toTreeData(menus, 0L, Menu::getParentId, Menu::getId, Menu::setChildren));
+//        PageParam<Menu, BaseParam> param = new PageParam<>(buildParam(map, BaseParam.class), Menu.class);
+//        QueryWrapper<Menu> wrapper = param.buildWrapper(true, queryWrapper -> queryWrapper.orderByAsc("sort"));
+//        List<Menu> menus = menuService.list(wrapper);
+//        return R.ok().add(Utils.toTreeData(menus, 0L, Menu::getParentId, Menu::getId, Menu::setChildren));
+        List<Menu> menuList = menuService.list(new LambdaQueryWrapper<Menu>().orderByAsc(Menu::getSort));
+        List<Menu> treeData = CommonUtils.toTreeData(menuList, 0L, Menu::getParentId, Menu::getId, Menu::setChildren);
+        if (!Cools.isEmpty(map.get("condition"))) {
+            CommonUtils.treeRemove(treeData, String.valueOf(map.get("condition")), Menu::getName, Menu::getChildren);
+            CommonUtils.treeRemove(treeData, String.valueOf(map.get("condition")), Menu::getName, Menu::getChildren);
+        }
+        return R.ok().add(treeData);
     }
 
     @PreAuthorize("hasAuthority('system:menu:list')")
@@ -61,11 +67,27 @@
     @OperationLog("娣诲姞鑿滃崟")
     @PostMapping("/menu/save")
     public R save(@RequestBody Menu menu) {
+        if (menu.getParentId() != null && menu.getParentId() > 0) {
+            Menu parent = menuService.getById(menu.getParentId());
+            if (parent != null) {
+                menu.setParentName(parent.getName());
+            }
+        } else {
+            menu.setParentId(0L);
+        }
+
+        NodeUtils nodeUtils = new NodeUtils();
+        nodeUtils.generatePath0(item -> menuService.getById(item.getParentId()), menu, Menu::getId, Menu::getName, Menu::getParentId);
+        menu.setPath(nodeUtils.path.toString());
+        menu.setPathName(nodeUtils.pathName.toString());
+
+        menu.setCreateBy(getLoginUserId());
+        menu.setCreateTime(new Date());
+        menu.setUpdateBy(getLoginUserId());
+        menu.setUpdateTime(new Date());
         if (!menuService.save(menu)) {
             return R.error("娣诲姞澶辫触");
         }
-        NodeUtils nodeUtils = new NodeUtils();
-        nodeUtils.generatePath(Menu.class, menu, Menu::getId, Menu::getName, Menu::getParentId);
         return R.ok("娣诲姞鎴愬姛");
     }
 
@@ -73,6 +95,22 @@
     @OperationLog("淇敼鑿滃崟")
     @PostMapping("/menu/update")
     public R update(@RequestBody Menu menu) {
+        if (menu.getParentId() != null && menu.getParentId() > 0) {
+            Menu parent = menuService.getById(menu.getParentId());
+            if (parent != null) {
+                menu.setParentName(parent.getName());
+            }
+        } else {
+            menu.setParentId(0L);
+        }
+
+        NodeUtils nodeUtils = new NodeUtils();
+        nodeUtils.generatePath0(item -> menuService.getById(item.getParentId()), menu, Menu::getId, Menu::getName, Menu::getParentId);
+        menu.setPath(nodeUtils.path.toString());
+        menu.setPathName(nodeUtils.pathName.toString());
+
+        menu.setUpdateBy(getLoginUserId());
+        menu.setUpdateTime(new Date());
         if (!menuService.updateById(menu)) {
             return R.error("淇敼澶辫触");
         }

--
Gitblit v1.9.1