树形菜单结构

/**
 * 获取树形结构菜单
 * @return
 */
@Override
public Map<String, Object> getTreeMenu() {
    List<SysMenu> menuList = sysMenuMapper.selectAll();//所有菜单
    List<SysMenu> rootMenu = new ArrayList<>();//根节点
    for (SysMenu menu : menuList){
        if(menu.getParentId() == 0){//父节点为0的为根节点
            rootMenu.add(menu);
        }
    }
    Collections.sort(rootMenu, order());//根据menu的sort排序
    //设置子菜单，getchild是递归调用的
    for (SysMenu menu : rootMenu){
        List<SysMenu> childList = getChild(menu.getId(), menuList);
        menu.setChildren(childList); //给根节点设置子节点
    }

    //数据处理
    Map<String, Object> map = new HashMap<>();
    List<ZtreeVO> list = new ArrayList<>();
    for(SysMenu menu : rootMenu){
        ZtreeVO ztreeVO = SysMenuStrut.INSTANCE.toZtreeVO(menu);
        list.add(ztreeVO);
    }
    map.put("menus", list);
    return map;
}

//排序
private Comparator<SysMenu> order() {
    Comparator<SysMenu> comparator = new Comparator<SysMenu>() {
        @Override
        public int compare(SysMenu o1, SysMenu o2) {
            if(o1.getSort() != o2.getSort()){
                return o1.getSort() - o2.getSort();
            }
            return 0;
        }
    };
    return comparator;
}

//获取子节点
private List<SysMenu> getChild(Integer id, List<SysMenu> menuList) {
    List<SysMenu> childList = Lists.newArrayList();//子菜单
    for(SysMenu menu : menuList){
        //比较所有节点，相等说明为该根节点的子节点
        if(menu.getParentId().equals(id)){
            childList.add(menu);
        }
    }
    //递归
    for(SysMenu menu : childList){
        menu.setChildren(getChild(menu.getId(), menuList));
    }
    Collections.sort(childList, order());//排序
    //如果节点下没有子节点返回一个空List(递归退出)
    if(childList.size() == 0){
        return Lists.newArrayList();
    }
    return childList;
}

@Data
public class ZtreeVO {

private Integer id;

private String menuName;

private List<ZtreeVO> children;

}

@Mapper
public interface SysMenuStrut {

SysMenuStrut INSTANCE = Mappers.getMapper(SysMenuStrut.class);

@Mapping(source = "menuName", target = "menuName")
ZtreeVO toZtreeVO(SysMenu sysMenu);

}