#

luxiaotao1123

2024-02-20 8c4fc3d632958daa7aff0fdd8b273d85bec68ead

#

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

@@ -11,6 +11,7 @@
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;
@@ -63,6 +64,8 @@
        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("添加成功");
    }


 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/utils/NodeUtils.java

New file
@@ -0,0 +1,42 @@
package com.zy.asrs.wcs.utils;


import com.baomidou.mybatisplus.extension.service.IService;
import com.zy.asrs.framework.common.SpringUtils;

import java.io.Serializable;
import java.util.function.Function;

/**
 * Created by vincent on 2021/1/19
 */
public class NodeUtils {

    public StringBuilder path = new StringBuilder();

    public StringBuilder pathName = new StringBuilder();

    public interface NodeSupport<T> {
        T query();
    }

    public <T, R extends Serializable> void generatePath(Class<T> cls
            , T t
            , Function<? super T, ? extends R> idMapper
            , Function<? super T, ? extends String> nameMapper
            , Function<? super T, ? extends R> parentIdMapper) {
        IService<T> bean = (IService<T>) SpringUtils.getBean(cls.getSimpleName() + "Service");
        T parent = bean.getById(parentIdMapper.apply(t));
        if (null != parent) {
            path.insert(0, idMapper.apply(parent)).insert(0,",");
            pathName.insert(0, nameMapper.apply(parent)).insert(0,",");
            if (parentIdMapper.apply(parent) != null) {
                generatePath(cls, parent, idMapper, nameMapper, parentIdMapper);
            } else {
                path.deleteCharAt(0);
                pathName.deleteCharAt(0);
            }
        }
    }

}