原文：Java8新特性-使用Stream流遞歸實(shí)現(xiàn)遍歷樹形結(jié)構(gòu)

import com.alibaba.fastjson.JSONObject;
import lombok.Data;
import org.junit.Test;

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

public class Test2Class {

    @Data
    public class Menu {
        private Integer id;
        private String name;
        private Integer parentId;
        private List<Menu> childList;

        public Menu(Integer id, String name, Integer parentId) {
            this.id = id;
            this.name = name;
            this.parentId = parentId;
        }

    }



    @Test
    public void test(){
        List<Menu> menus = Arrays.asList(
                new Menu(1,"根節(jié)點(diǎn)1",0),
                new Menu(2,"子節(jié)點(diǎn)1.1",1),
                new Menu(3,"子節(jié)點(diǎn)1.2",1),
                new Menu(4,"子節(jié)點(diǎn)1.3",1),
                new Menu(5,"子節(jié)點(diǎn)1.4",1),
                new Menu(6,"根節(jié)點(diǎn)2",0),
                new Menu(7,"子節(jié)點(diǎn)2.1",6),
                new Menu(8,"子節(jié)點(diǎn)2.2",6),
                new Menu(9,"子節(jié)點(diǎn)2.2.1",8),
                new Menu(10,"子節(jié)點(diǎn)2.2.2",8),
                new Menu(11,"根節(jié)點(diǎn)3",0),
                new Menu(12,"子節(jié)點(diǎn)3.1",11)
        );

        //獲取父節(jié)點(diǎn)
        List<Menu> collect = menus.stream().filter(m-> m.getParentId()==0).map(
                m->{
                    m.setChildList(getChildrens(m,menus));
                    return m;
                }
        ).collect(Collectors.toList());
        System.out.println(JSONObject.toJSON(collect));
    }



    public List<Menu> getChildrens(Menu root,List<Menu> all){
        List<Menu> children = all.stream().filter(m-> Objects.equals(m.getParentId(),root.getId())).map(
                m->{
                    m.setChildList(getChildrens(m,all));
                    return m;
                }
        ).collect(Collectors.toList());
        return children;
    }

}

getChildrens也可以這么寫，寫在方法內(nèi)：

BiFunction<Menu,List<Menu>,List<Menu>> getChildrens =new BiFunction<Menu, List<Menu>, List<Menu>>() {
    @Override
    public List<Menu> apply(Menu root, List<Menu> all) {
        List<Menu> children = all.stream().filter(m-> Objects.equals(m.getParentId(),root.getId())).peek(
            m-> m.setChildList(this.apply(m,all))
        ).collect(Collectors.toList());
        return children;
    }
};