#

luxiaotao1123

2024-02-21 7d3040d4c742af5e13f7327e7397a0524c1ffdec

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

@@ -1,6 +1,7 @@
package com.zy.asrs.wcs.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.zy.asrs.framework.common.Cools;
import com.zy.asrs.framework.common.R;
@@ -11,15 +12,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")
@@ -43,6 +43,15 @@
    }

    @PreAuthorize("hasAuthority('system:menu:list')")
    @PostMapping("/menu/tree")
    public R tree(@RequestBody Map<String, Object> map) {
        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));
    }

    @PreAuthorize("hasAuthority('system:menu:list')")
    @GetMapping("/menu/{id}")
    public R get(@PathVariable("id") Long id) {
        return R.ok().add(menuService.getById(id));
@@ -52,6 +61,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("添加失败");
        }
@@ -62,6 +89,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("修改失败");
        }