TreeUtils 树工具类_treeutil.build-CSDN博客

本文链接：https://blog.csdn.net/javaxueba/article/details/141217178

数据展示：

如图：部门树数据，树形的基础数据 id 、 parentId 、label 便可形成

嵌套对象字段如下：{id: 103, parentId: 101, label: "研发部门", weight: 1}

一、工具类

继承了 hutoo 的工具类 TreeUtil （引入hutool 依赖）

@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class TreeBuildUtils extends TreeUtil {

    /**
     * 根据前端定制差异化字段
     */
    public static final TreeNodeConfig DEFAULT_CONFIG = TreeNodeConfig.DEFAULT_CONFIG.setNameKey("label");

    /**
     * <简述>
     * <详细描述>
     * @author syf
     * @date 2024/8/15 11:27
     * @param list 提供的集合
     * @param nodeParser 节点解析器
     * @return java.util.List<cn.hutool.core.lang.tree.Tree < K>>
     */
    public static <T, K> List<Tree<K>> build(List<T> list, NodeParser<T, K> nodeParser) {
        if (CollUtil.isEmpty(list)) {
            return null;
        }
        // 获取父节点
        K k = ReflectUtils.invokeGetter(list.get(0), "parentId");
        return TreeUtil.build(list, k, DEFAULT_CONFIG, nodeParser);
    }

}

TreeNodeConfig 树形需要的字段：可以通过set字段修改

public class TreeNodeConfig implements Serializable {
    private static final long serialVersionUID = 1L;
    public static TreeNodeConfig DEFAULT_CONFIG = new TreeNodeConfig();
    private String idKey = "id";
    private String parentIdKey = "parentId";
    private String weightKey = "weight";
    private String nameKey = "name";
    private String childrenKey = "children";
    private Integer deep;

。。。。。省里get  set

TreeUtil 方法展示：树形构建已经操作节点方法

二、准备简单树形数据

@Slf4j
public class TestTreeNode {
   static List<TreeNode<Integer>> createTreeNodes(){
       List<TreeNode<Integer>> list = new ArrayList<>();
       TreeNode<Integer>  root= new TreeNode<>();
       root.setName("总公司");
       root.setId(1);
       root.setParentId(0);

       TreeNode<Integer>  node1= new TreeNode<>();
       node1.setName("分子公司1");node1.setId(2);node1.setParentId(1);


       TreeNode<Integer>  node2= new TreeNode<>();
       node2.setName("财务部");node2.setId(3);node2.setParentId(2);
       TreeNode<Integer>  node3= new TreeNode<>();
       node3.setName("开发部").setId(4).setParentId(2);

       list.add(root);
       list.add(node1);
       list.add(node2);
       list.add(node3);
       return list;
   }

    public static void main(String[] args) {
        buildSingle();
    }

    static void  buildSingle(){
        List<TreeNode<Integer>> list = createTreeNodes();
        Tree<Integer> integerTree = TreeBuildUtils.buildSingle(list);
        Console.log(integerTree);
    }


}

打印结果：

null[0]
  总公司[1]
    分子公司1[2]
      财务部[3]
      开发部[4]

三、测试所有常用树形方法，并附上DEMO演示

构建树形
获取指定节点
添加节点
移除节点
遍历
过滤
克隆

public static void main(String[] args) {
        buildSingle();
    }

    static void  buildSingle(){
        List<TreeNode<Integer>> list = createTreeNodes();

        //build 对比上面 去掉最外层 null 的父节点
        List<Tree<Integer>> build = TreeBuildUtils.build(list);
        Console.log(build);


        Console.log("-------------------------------------------测试开始------------------------------------------------");

        //原生 构建树结构
        Tree<Integer> integerTree = TreeBuildUtils.buildSingle(list);
        Console.log(integerTree);

        // 获取父节点
        Tree<Integer> parent = integerTree.getParent();
        Console.log("getParent->{}", parent);
        // 获取节点
        Tree<Integer> node = integerTree.getNode(2);
        Console.log("getNode->{}", node);
        // 获取父节点名称
        List<CharSequence> parentsName = integerTree.getParentsName(4, true);
        Console.log("getParentsName->{}", parentsName);
        // 获取配置
        TreeNodeConfig config = integerTree.getConfig();


        // 添加子节点
//        Tree<Integer> integerTree1 = integerTree.addChildren(node);
//        Console.log("addChildren->{}", integerTree1);
        // 判断是否有子节点
        boolean b = integerTree.hasChild();
        Console.log("hasChild->{}", b);
        // 获取子节点
        List<Tree<Integer>> children = integerTree.getChildren();
        Console.log("getChildren->{}", children);

        //遍历整数树结构，并打印每个节点
        integerTree.walk(new Consumer<Tree<Integer>>() {
            @Override
            public void accept(Tree<Integer> integerTree) {
                // 打印节点
                Console.log("walk->{}", integerTree);
            }
        });


        //根据是否包含特定名称（如“分子公司”）来筛选树结构数据
        Tree<Integer> filterNew = integerTree.filterNew(new Filter<Tree<Integer>>() {
            @Override
            public boolean accept(Tree<Integer> integerTree) {
                if (null != integerTree.getName() && integerTree.getName().toString().contains("分子公司")) {
                    return true;
                }
                return false;
            }
        });
        Console.log("filterNew->{}", filterNew);


        // 创建一个过滤器，用于筛选出名称中包含"分子公司"的Tree对象
        Tree<Integer> filter = filterNew.filter(new Filter<Tree<Integer>>() {
            @Override
            public boolean accept(Tree<Integer> integerTree) {
                if (null != integerTree.getName() && integerTree.getName().toString().contains("分子公司")) {
                    return true;
                }
                return false;
            }
        });
        Console.log("filter->{}", filter);


        // 克隆一个整数树结构
        Tree<Integer> integerTree2 = integerTree.cloneTree();
        Console.log("cloneTree->{}", integerTree2);


    }

测试结果：

null[0]
  总公司[1]
    分子公司1[2]
      财务部[3]
      开发部[4]

[总公司[1]
  分子公司1[2]
    财务部[3]
    开发部[4]
]
getParent->null
getNode->分子公司1[2]
  财务部[3]
  开发部[4]

getParentsName->[开发部, 分子公司1, 总公司]
hasChild->true
getChildren->[总公司[1]
  分子公司1[2]
    财务部[3]
    开发部[4]
]
walk->null[0]
  总公司[1]
    分子公司1[2]
      财务部[3]
      开发部[4]

walk->总公司[1]
  分子公司1[2]
    财务部[3]
    开发部[4]

walk->分子公司1[2]
  财务部[3]
  开发部[4]

walk->财务部[3]

walk->开发部[4]

filterNew->null[0]
  总公司[1]
    分子公司1[2]
      财务部[3]
      开发部[4]

filter->null[0]
  总公司[1]
    分子公司1[2]
      财务部[3]
      开发部[4]

cloneTree->null[0]
  总公司[1]
    分子公司1[2]
      财务部[3]
      开发部[4]


Process finished with exit code 0