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



文章目录


实体类
@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;
}