java 返回树状类型数据

最近新来了个前端，菜单等相关功能不想使用前端的数标签，希望后端组装好数结构返回。于是写了个通用型的工具类，供以后遇见类似的问题时使用。

实体类

@Data
public class TreeTest{
    private String id;
    private String parentId;
    private List<TreeTest> children=new ArrayList<>();
    public TreeTest(){};
    public TreeTest(String id){
        this.id=id;
    }
    public TreeTest(String id,String parentId){
        this.id=id;
        this.parentId=parentId;
    }
}

上面构造方法我是为了方便调试使用的，不用在意，但是children=new ArrayList()，这个如果要使用我的工具类的话这里初始化必须写上。

具体实现方法

public static <T,R> List<T> parTree(List<T> list, Function<T, ? extends R> id,
                                          Function<? super T, ? extends R> parent,
                                          Function<? super T, ? extends R> children){
        List<T> list1=new ArrayList();
        Map<Object, T> collect = list.stream().collect(Collectors.toMap(id, t -> t));
        Iterator<Object> iter = collect.keySet().iterator();
        while (iter.hasNext()) {
            T t = collect.get(iter.next());
            Object parentObject = parent.apply(t);
            if(parentObject==null){
                list1.add(t);
            }else{
                T parentT=collect.get(parentObject);
                List<T> childrenList = (List<T>)children.apply(parentT);
                childrenList.add(t);
            }
        }
        return list1;
    }

使用函数的方式来实现，通用性更好。

调用方式如下：

Util.parTree(list, TreeTest::getId, TreeTest::getParentId,
                TreeTest::getChildren);

例如我自己的测试例子：

List<TreeTest> list=new ArrayList(){{
            add(new TreeTest("1"));
            add(new TreeTest("2"));
            add(new TreeTest("3"));
            add(new TreeTest("4","1"));
            add(new TreeTest("5","2"));
            add(new TreeTest("6","3"));
            add(new TreeTest("7","4"));
            add(new TreeTest("8","4"));
            add(new TreeTest("9","7"));
            add(new TreeTest("10","7"));
        }};

        List<TreeTest> list1 = Util.parTree(list, TreeTest::getId, TreeTest::getParentId,
                TreeTest::getChildren);
        System.out.println(list1);

打印出来结果为[TreeTest(id=1, parentId=null, children=[TreeTest(id=4, parentId=1, children=[TreeTest(id=7, parentId=4, children=[TreeTest(id=9, parentId=7, children=[]), TreeTest(id=10, parentId=7, children=[])]), TreeTest(id=8, parentId=4, children=[])])]), TreeTest(id=2, parentId=null, children=[TreeTest(id=5, parentId=2, children=[])]), TreeTest(id=3, parentId=null, children=[TreeTest(id=6, parentId=3, children=[])])] 。

主要是利用java是引用地址对象，修改对象的值，实际上其他的引用也跟着发生变化，以及熟悉了一下function的相关用法。