父子结构List(根节点可以为null)转树形结构
public List<Menu> toTree(List<Menu> list) {
Map<String, Menu> map = new HashMap<>();
List<Menu> trees = new ArrayList<>();
for (Menu item : list) {
Menu node = new Menu();
BeanUtils.copyProperties(item, node);
map.put(item.getId(), node);
}
for(Menu n : list){
String pid = n.getParentId();
String id = n.getId();
//当前节点的父节点
Menu pNode = map.get(pid);
//当前节点
Menu cNode = map.get(id);
if (StringUtils.isBlank(pid)) {
//如果父id为空,则为根节点
cNode.setParentId(CommonConstant.AUTH_ROOT_PARENT_ID);
trees.add(cNode);
}else{
List<Menu> children = pNode.getMenus();
if (children == null) {
//如果子节点为空,初始化子节点
children = new ArrayList<>();
}
//将此子节点放入对应父节点下
children.add(cNode);
pNode.setMenus(children);
}
}
return trees;
}