三分钟带你实现 java8 下 Stream 递归实现树形目录结构

你知道的越多，你不知道的越多
点赞再看，养成习惯
如果您有疑问或者见解，或者没有积分想获取项目和定制项目，欢迎指教：
企鹅：869192208

实体类

@Data
@ApiModel("功能菜单实体类")
public class MenuModel{

	/**
	 * 主键
	 */
	private Integer id;
	/**
	 * 标题
	 */
	private String title;
	/**
	 * 图标
	 */
	private String icon;
	/**
	 * 链接
	 */
	private String href;
	/**
	 * 目标
	 */
	private String target;
	/**
	 * 父ID
	 */
	private Integer parentId;
	/**
	 * 类型：0-管理员/宿管员;1-学生
	 */
	private Integer type;

	/**
	 * 子菜单
	 */
	private List<MenuModel> child;
}

service实现

public List<MenuModel> query(Integer userId) {
		//从数据库查出所有的功能菜单，根节点的parentId为0
        List<MenuModel> menus = menuDao.query(userId);
        // 获取所有根节点
        List<MenuModel> parentMenus = menus.stream().filter(l -> l.getParentId().equals(0)).map(m -> {
            m.setChild(getChildrens(m, menus));
            return m;
        }).collect(Collectors.toList());
        return parentMenus;
    }

    /**
     * 递归查询子节点
     * @param root  根节点
     * @param all   所有节点
     * @return 根节点信息
     */
    private List<MenuModel> getChildrens(MenuModel root, List<MenuModel> all) {
        List<MenuModel> children = all.stream().filter(m -> {
            return Objects.equals(m.getParentId(), root.getId());
        }).map(
                m -> {
                    m.setChild(getChildrens(m, all));
                    return m;
                }
        ).collect(Collectors.toList());
        return children;
    }