使用HutoUtil 中TreeUtil進行樹形結構操作(可直接運行)

pom.xml

<dependency>    <groupId>cn.hutool</groupId>    <artifactId>hutool-all</artifactId>    <version>5.4.6</version></dependency>

實體類

@Data@Accessors(chain = true)public class TreeDemo {    private Integer id;    private Integer parentId;    private Integer level;    private String  levelName;    private Integer isEfficient;    private Integer isShow;    private String createdBy;    private Date createdDate;    private String updatedBy;    private Date updatedDate;}

測試類

import cn.hutool.core.lang.tree.Tree;import cn.hutool.core.lang.tree.TreeNodeConfig;import cn.hutool.core.lang.tree.TreeUtil;import cn.hutool.json.JSONUtil;import java.util.*;import java.util.function.Function;import java.util.stream.Collectors;/** * TODO 樹形結構測試 * @date 2021/12/22 11:41 */public class TestDemo {    public static void main(String[] args) {        List<TreeDemo> treeDemoList = new ArrayList<>();        treeDemoList=initTreeDemo();        // 1、將源數據轉化為樹        List<Tree<Integer>> build = transform(treeDemoList);        //2、通過子節點查找父節點        List<TreeDemo> parentList = childToParent(treeDemoList,build);        //3、通過父節點查找子節點        Tree<Integer> tree= parentToChile(build);    }    /**     *  1:初始化數據     */    public static List<TreeDemo> initTreeDemo(){        List<TreeDemo> treeDemoList = new ArrayList<>();        TreeDemo treeDemo  = new TreeDemo();        treeDemo.setId(1).setParentId(0).setLevel(1).setLevelName("A").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo1  = new TreeDemo();        treeDemo1.setId(2).setParentId(1).setLevel(2).setLevelName("B").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo2  = new TreeDemo();        treeDemo2.setId(3).setParentId(2).setLevel(3).setLevelName("C").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo3  = new TreeDemo();        treeDemo3.setId(4).setParentId(3).setLevel(4).setLevelName("D").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo4  = new TreeDemo();        treeDemo4.setId(5).setParentId(4).setLevel(5).setLevelName("E").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo5  = new TreeDemo();        treeDemo5.setId(6).setParentId(0).setLevel(1).setLevelName("A1").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo6  = new TreeDemo();        treeDemo6.setId(7).setParentId(6).setLevel(2).setLevelName("B1").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo7  = new TreeDemo();        treeDemo7.setId(8).setParentId(7).setLevel(3).setLevelName("C1").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo8  = new TreeDemo();        treeDemo8.setId(9).setParentId(8).setLevel(4).setLevelName("D1").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo9  = new TreeDemo();        treeDemo9.setId(10).setParentId(9).setLevel(5).setLevelName("E1").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo10  = new TreeDemo();        treeDemo10.setId(11).setParentId(7).setLevel(3).setLevelName("C2").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo11  = new TreeDemo();        treeDemo11.setId(12).setParentId(8).setLevel(4).setLevelName("D2").setIsEfficient(1).setIsShow(1);        TreeDemo treeDemo12  = new TreeDemo();        treeDemo12.setId(13).setParentId(9).setLevel(5).setLevelName("E2").setIsEfficient(1).setIsShow(1);        treeDemoList.add(treeDemo);        treeDemoList.add(treeDemo1);        treeDemoList.add(treeDemo2);        treeDemoList.add(treeDemo3);        treeDemoList.add(treeDemo4);        treeDemoList.add(treeDemo5);        treeDemoList.add(treeDemo6);        treeDemoList.add(treeDemo7);        treeDemoList.add(treeDemo8);        treeDemoList.add(treeDemo9);        treeDemoList.add(treeDemo10);        treeDemoList.add(treeDemo11);        treeDemoList.add(treeDemo12);        return treeDemoList;    }    /**     *  1.2:將源數據轉化為樹結構數據     */    public static List<Tree<Integer>> transform(List<TreeDemo> treeDemoList){        treeDemoList=initTreeDemo();        // 2.配置        TreeNodeConfig config = new TreeNodeConfig();        //默認為id可以不設置        config.setIdKey("id");        //默認為parentId可以不設置        config.setParentIdKey("parentId");        //最大遞歸深度        config.setDeep(5);        //排序字段        config.setWeightKey("level");        // 3.轉樹,Tree<>里面泛型為id的類型        List<Tree<Integer>> build = TreeUtil.build(treeDemoList, 0, config, (object, tree) -> {            // 也可以使用 tree.setId(object.getId());等一些默認值            tree.putExtra("id", object.getId());            tree.putExtra("parentId", object.getParentId());            tree.putExtra("level", object.getLevel());            tree.putExtra("levelName", object.getLevelName());            tree.putExtra("isEfficient", object.getIsEfficient());            tree.putExtra("isShow", object.getIsShow());        });        System.out.println("將源數據轉化為樹:"+JSONUtil.toJsonStr(build));        return build;    }    /**     * @desc  2.1 子節點查找父節點(此功能自己實現可以用treeUtil里面的getParentsName)     *     *  實現思路:1、獲取指定節點     *          2、將全部節點轉化為map,id做key     *          3、遞歸循環向上匹配 知道對象為空     */    public static List<TreeDemo> childToParent(List<TreeDemo> treeDemoList, List<Tree<Integer>> build){//定義指定節點        TreeDemo treeDemo9  = new TreeDemo();        treeDemo9.setId(10).setParentId(9).setLevel(5).setLevelName("E1").setIsEfficient(1).setIsShow(1);//通過set添加查找后的實體        final Set<TreeDemo> nodeList = new HashSet<>();//將源數據轉化為map        final Map<Integer, TreeDemo> allMap = treeDemoList.stream().collect(Collectors.toMap(TreeDemo::getId, Function.identity(), (v1, v2) -> v2));//        build.forEach(node -> {        findParent(nodeList, treeDemo9, allMap);//        });        List<TreeDemo> nodes = new ArrayList<>(nodeList);        System.out.println("子節點找父節點:"+JSONUtil.toJsonStr(nodes));        return nodes;    }    /**     * :2.2遞歸查找父節點     */    private static void findParent(Set<TreeDemo> nodeList, TreeDemo current, Map<Integer, TreeDemo> allMap) {        if (Objects.nonNull(current)) {            nodeList.add(current);        }        if (Objects.nonNull(current)) {            TreeDemo parent = allMap.get(current.getParentId());            findParent(nodeList, parent, allMap);        }    }    /**     * @desc  通過指定3:節點查找子節點     */    public static Tree<Integer> parentToChile( List<Tree<Integer>> build){        Tree<Integer> childNode = new Tree<>();        for(Tree node : build){            childNode=TreeUtil.getNode(node,6);            if(Objects.nonNull(childNode)){                break;            }        }        System.out.println("查找父節點及其子節點:"+JSONUtil.toJsonStr(childNode));        return childNode;    }