菜单树tree---

@Controller
@RequestMapping("/test")
public class TestTree{
    /*
     * 排序,根据order排序
     */
    public static Comparator<Menu> order() {
        Comparator<Menu> comparator = new Comparator<Menu>() {
            @Override
            public int compare(Menu o1, Menu o2) {
                if (o1.getOrder() != o2.getOrder()) {
                    return o1.getOrder() - o2.getOrder();
                }
                return 0;
            }
        };
        return comparator;
    }
@PostMapping("/findTree")
    @ResponseBody
    public static Map<String, Object> findTree() {
        Menu menu = new Menu();
        menu.setId("000");
        menu.setIcon("icon");
        menu.setName("一级菜单");
        menu.setOrder(1);
        menu.setParentId("0");
        menu.setUrl("url");

        Menu menu1 = new Menu();
        menu1.setId("000001");
        menu1.setIcon("icon");
        menu1.setName("二级菜单");
        menu1.setOrder(2);
        menu1.setParentId("000");
        menu1.setUrl("url");

        Menu menu2 = new Menu();
        menu2.setId("000002");
        menu2.setIcon("icon");
        menu2.setName("二级菜单2");
        menu2.setOrder(3);
        menu2.setParentId("000");
        menu2.setUrl("url2");

        Menu menu4 = new Menu();
        menu4.setId("000001002");
        menu4.setIcon("icon");
        menu4.setName("三级菜单2");
        menu4.setOrder(3);
        menu4.setParentId("000001");
        menu4.setUrl("url2");
Menu menu3 = new Menu();
        menu3.setId("000001001");
        menu3.setIcon("icon");
        menu3.setName("三级菜单2");
        menu3.setOrder(3);
        menu3.setParentId("000001");
        menu3.setUrl("url2");
        List<Menu> allMenu = new ArrayList<Menu>();
        allMenu.add(menu);
        allMenu.add(menu1);
        allMenu.add(menu2);
        allMenu.add(menu3);
        allMenu.add(menu4);
        Map<String, Object> data = new HashMap<String, Object>();
        try { //查询所有菜单
            //根节点
            List<Menu> rootMenu = new ArrayList<Menu>();
            for (Menu nav : allMenu) {
                if (nav.getParentId().equals("0")&&nav.getParentId()!=null) { //父节点是0的,为根节点。
                    rootMenu.add(nav);
                }
            }
            /* 根据Menu类的order排序 */
            Collections.sort(rootMenu, order());
            //为根菜单设置子菜单,getClild是递归调用的
            for (Menu nav : rootMenu) {
                /* 获取根节点下的所有子节点 使用getChild方法*/
                List<Menu> childList = getChild(nav.getId(), allMenu);
                nav.setChildren(childList);//给根节点设置子节点
            }
            data.put("list", rootMenu);
            return data;
        } catch (Exception e) {
            return data;
        }
    }

    /**
     * 获取子节点
     *
     * @param id      父节点id
     * @param allMenu 所有菜单列表
     * @return 每个根节点下,所有子菜单列表
     */
    public static List<Menu> getChild(String id, List<Menu> allMenu) {
        //子菜单
        List<Menu> childList = new ArrayList<Menu>();
        for (Menu nav : allMenu) {
            // 遍历所有节点,将所有菜单的父id与传过来的根节点的id比较
            //相等说明:为该根节点的子节点。
            if (nav.getParentId().equals(id)) {
                childList.add(nav);
            }
        }
        //递归
        for (Menu nav : childList) {
            nav.setChildren(getChild(nav.getId(), allMenu));
        }
        Collections.sort(childList, order()); //排序
        //如果节点下没有子节点,返回一个空List(递归退出)
        if (childList.size() == 0) {
            return new ArrayList<Menu>();
        }
        return childList;
    }
}
class Menu {
    // 菜单id
    private String id;
    // 菜单名称
    private String name;
    // 父菜单id
    private String parentId;
    // 菜单url
    private String url;
    // 菜单图标
    private String icon;
    // 菜单顺序
    private int order;
    // 子菜单
    private List<Menu> children;

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一叶飘零晋

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值