菜单递归


/**
 * @ClassName: FunctionService
 * @Description: 功能菜单
 * @author: yaozhenhua
 * @date: 2018/12/26 12:29
 */
@Service
public class MenuService {
    
    @Resource
    MenuDao menuDao;
    
    /**
     *查询可用且上线的菜单
     *
     * @param
     * @author: yaozhenhua  2019/3/27 17:24
     */
    public List<MenuVO> listMenu() throws InstantiationException, IllegalAccessException, AobpException {
        //结果处理容器
        List<MenuVO> menuVOS = new ArrayList<>();
        //查询一级菜单
        List<MenuBO> menuBOS = menuDao.listParentMenu();
        if(!EmptyUtils.isEmpty(menuBOS)){
            //一级菜单排序
            Collections.sort(menuBOS, new Comparator<MenuBO>() {
                @Override
                public int compare(MenuBO o1, MenuBO o2) {
                    int ret = 0;
                    ret = o1.getParentSequence().compareTo(o2.getParentSequence());
                    return ret;
                }
            });
                
            menuVOS = BeanCopyUtils.copyList(menuBOS,MenuVO.class);
          
            for(int i=0; i<menuBOS.size(); i++){
                MenuBO menuBO = menuBOS.get(i);
                MenuVO menuVO = menuVOS.get(i);
                //查询子菜单并排序
                List<MenuVO> subMenuVOS = listSubMenu(menuBO.getId(),1);
                menuVO.setSubs(subMenuVOS);
                }
            }
        
        return menuVOS;
    }
    
    /**
     *根据父id查询子菜单
     *
     * @param parentId
     * @author: yaozhenhua  2019/3/27 17:24
     */
    public List<MenuVO> listSubMenu(Long parentId, int alarm) throws InstantiationException, IllegalAccessException, AobpException {
        　//计数器 递归20次抛出异常，菜单层级最大没有超过20层的。
　　　　　if(alarm>20){
            throw new AobpException(ExceptionConstant.INTERNAL_ERR,"递归次数超过20次异常");
        }
        //查询出此父菜单下的子菜单
        List<MenuBO>  menuBOS= menuDao.listMenuByParentId(parentId);
        //结果容器
        List<MenuVO> menuVOS = new ArrayList<>();
        //查看子菜单是不是还有下一级菜单（判定条件  用id作为父id去查 ）
        if (!EmptyUtils.isEmpty(menuBOS)){
            Collections.sort(menuBOS, new Comparator<MenuBO>() {
                @Override
                public int compare(MenuBO o1, MenuBO o2) {
                    int ret = 0;
                    ret = o1.getChildSequence().compareTo(o2.getChildSequence());
                    return ret;
                }
            });
    
            menuVOS = BeanCopyUtils.copyList(menuBOS,MenuVO.class);
    
            for (int i = 0; i < menuBOS.size(); i++) {
                MenuBO menuBO = menuBOS.get(i);
                MenuVO menuVO = menuVOS.get(i);
                List<MenuVO> nextMenuVOS = listSubMenu(menuBO.getId(),alarm+1);
                if(!EmptyUtils.isEmpty(nextMenuVOS)){
    
                    menuVO.setSubs(nextMenuVOS);
                }
            }
           
            
        }
        return menuVOS;
    }
}


另一种方式：

/**
 * @ClassName: MenuService
 * @author: yaozhenhua
 * @date: 2019/3/27 15:59
 */
public class MenuService {
    
    @Transactional(readOnly = true)
    public List<Menu> selectDocMenuList() {
        // 查询全部文档
        List<Menu> rootMenuList = docDao.selectListExpUnable();
        // 最终的菜单
        List<Menu> docList = new ArrayList<Menu>();
        // 没有父ID的文档, 作为一级菜单
        for (Menu rootMenu : rootMenuList) {
            if (rootMenu.getDocPid() == null) {
                docList.add(rootMenu);
            }
        }
        
        // 为一级菜单添加子菜单
        for (Menu menu : docList) {
            menu.setChildList(getChild(menu.getDocId(), rootMenuList));
        }
        if (docList != null && docList.size() > 0) {
            // 为list排序
            this.mySort(docList, "docOrder", null);
        }
        
        return docList;
    }
    

    private List<Menu> getChild(String docId, List<Menu> rootMenu) {
        if (StringUtils.isBlank(docId)) {
            throw new BusinessException("文档id不能为空");
        }
        // 子菜单
        List<Menu> childList = new ArrayList<>();
        for (Menu menu : rootMenu) {
            // 遍历所有节点，将父菜单id与传过来的type比较
            if (menu.getDocPid() != null) {
                if (menu.getDocPid().equals(docId)) {
                    childList.add(menu);
                }
            }
        }
        // 把子菜单的子菜单再循环一遍
        for (Menu menu : childList) {// 没有内容的子菜单还有子菜单
            try {
                if (menu.getDocContentHref() != null) {
                    menu.setDocContent(new String(menu.getDocContentHref(), "UTF-8"));
                }
                if (StringUtils.isBlank(menu.getDocContent())) {
                    // 递归
                    menu.setChildList(getChild(menu.getDocId(), rootMenu));
                }
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
        } // 递归退出条件
        if (childList != null && childList.size() > 0) {
            this.mySort(childList, "docOrder", null);
        } else {
            return null;
        }
        
        return childList;
    }
}

public 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;
  // ... 省去getter和setter方法以及toString方法
}

菜单一般需要排序，我们根据Menu的order字段进行排序：

/*
  * 排序,根据order排序
  */
 public 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;
 }

生成树的方法：

public Map<String,Object> findTree(){
  Map<String,Object> data = new HashMap<String,Object>();
    try {//查询所有菜单
      List<Menu> allMenu = menuDao.findTree();
      //根节点
      List<Menu> rootMenu = new ArrayList<Menu>();
      for (Menu nav : allMenu) {
        if(nav.getParentId().equals("0")){//父节点是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("success", "true");
      data.put("list", rootMenu);
      return data;
    } catch (Exception e) {
      data.put("success", "false");
      data.put("list", new ArrayList());
      return data;
    }
  }

获取子菜单：

 




/**
   * 获取子节点
   * @param id 父节点id
   * @param allMenu 所有菜单列表
   * @return 每个根节点下，所有子菜单列表
   */
  public List<Menu> getChild(String id,List<Menu> allMenu){
    //子菜单
    List<Menu> childList = new ArrayList<Menu>();
    for (Menu nav : allMenu) {
      // 遍历所有节点，将所有菜单的父id与传过来的根节点的id比较
      //相等说明：为该根节点的子节点。
      if(nav.ParentId().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;
  }

菜单递归

推荐阅读