package com.project.common.utils;
|
|
import com.project.common.core.absclasses.TreeNodeManager;
|
import com.project.common.core.interfaces.ITreeNode;
|
|
import java.util.List;
|
|
public class TreeManager extends TreeNodeManager
|
{
|
|
public TreeManager(List nodes) {
|
super(nodes);
|
}
|
|
public static <T extends ITreeNode<T>> List<T> merge(List<T> items) {
|
TreeNodeManager<T> forestNodeManager = new TreeNodeManager(items);
|
items.forEach((forestNode) -> {
|
if (forestNode.getParentId() != 0L) {
|
ITreeNode<T> node = forestNodeManager.getTreeNodeAt(forestNode.getParentId());
|
if (node != null) {
|
node.getChildren().add(forestNode);
|
} else {
|
forestNodeManager.addParentId(forestNode.getId());
|
}
|
}
|
|
});
|
return forestNodeManager.getRoot();
|
}
|
}
|