工具类TreeBuild:
public class TreeBuild {
// 保存参与构建树形的所有数据(通常数据库查询结果)
public List<TreeNodeDTO> nodeList = new ArrayList<>();
/**
* 构造方法
* @param nodeList 将数据集合赋值给nodeList,即所有数据作为所有节点。
*/
public TreeBuild(List<TreeNodeDTO> nodeList){
this.nodeList = nodeList;
}
/**
* 获取需构建的所有根节点(顶级节点) "0"
* @return 所有根节点List集合
*/
public List<TreeNodeDTO> getRootNode(){
// 保存所有根节点(所有根节点的数据)
List<TreeNodeDTO> rootNodeList = new ArrayList<>();
// treeNode:查询出的每一条数据(节点)
for (TreeNodeDTO treeNode : nodeList){
// 判断当前节点是否为根节点,此处注意:若parentId类型是String,则要采用equals()方法判断。
if ("0".equals(treeNode.getParentId())) {
// 是,添加
root