使用Map和List构建树形结构数据--(适合数据量不大时使用)

bean对象

@ApiModel(value = "AdminMenuTreeVO", description = "树形菜单")
public class AdminMenuTreeVO {

  @ApiModelProperty(value = "ID")
  private String id;

  @ApiModelProperty(value = "父级ID")
  private String parentId;

  @ApiModelProperty(value = "名称")
  private String name;


  @ApiModelProperty(value = "菜单唯一编码")
  private String code;

  @ApiModelProperty(value = "菜单路径")
  private String menuUrl;


  @ApiModelProperty(value = "菜单类型")
  private String menuType;

  @ApiModelProperty(value = "菜单排序")
  private String sortNum;


  @ApiModelProperty(value = "菜单等级")
  private String level;


  @ApiModelProperty(value = "状态 0:不可用 1:可用")
  private String status;

  @ApiModelProperty(value = "描述")
  private String description;

  @ApiModelProperty(value = "子节点")
  private List<AdminMenuTreeVO> childrenList;
  
  // 其余代码......
}

构建树

  public List<AdminMenuTreeVO> getMenuTree(List<AdminMenu> adminMenus) {
    List<AdminMenuTreeVO> list = new ArrayList<>();

    Map<Long, AdminMenuTreeVO> map = new HashMap<>(); // <id,菜单对象>
    if (adminMenus != null && adminMenus.size() > 0) {
      adminMenus.forEach(t -> map.put(t.getId(), new AdminMenuTreeVO(t)));

      // 建树
      for (AdminMenu menu : adminMenus) {
        AdminMenuTreeVO p_menu = map.get(menu.getParentId()); // 父级菜单
        AdminMenuTreeVO c_menu = map.get(menu.getId()); // 当前菜单

        if (p_menu != null) { // 有父级
          List<AdminMenuTreeVO> childrenList = p_menu.getChildrenList(); // 取子
          childrenList.add(c_menu); // 当前添加到父级孩子中

        } else if (menu.getParentId() != null && menu.getParentId() > 0) { // 有父级，但是父级不在map中：即父级被停用或删除了
          map.remove(map.get(menu.getId())); // 父级不在map，则当前对象从map移除

        } else { // 没有父级，当前菜单为顶级菜单
          list.add(c_menu); // 顶级菜单带着子菜单加入list
        }
      }
    }

    return list;
  }