#

luxiaotao1123

2024-02-20 7e7dac82d82f3273cb709e404aa179787d561011

 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java

@@ -11,16 +11,14 @@
import com.zy.asrs.wcs.system.entity.Menu;
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 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")
@@ -46,7 +44,8 @@
    @PreAuthorize("hasAuthority('system:menu:list')")
    @PostMapping("/menu/tree")
    public R tree(@RequestBody Map<String, Object> map) {
        List<Menu> menus = menuService.list();
        PageParam<Menu, BaseParam> param = new PageParam<>(buildParam(map, BaseParam.class), Menu.class);
        List<Menu> menus = menuService.list(param.buildWrapper(true));
        return R.ok().add(Utils.toTreeData(menus, 0L, Menu::getParentId, Menu::getId, Menu::setChildren));
    }

@@ -60,6 +59,24 @@
    @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("添加失败");
        }
@@ -70,6 +87,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("修改失败");
        }