1. /**
    2.  * Menu
    3.  * @author lcry
    4.  */
    5. @Data
    6. @Builder
    7. public class Menu {
    8.     /**
    9.      * id
    10.      */
    11.     public Integer id;
    12.     /**
    13.      * 名称
    14.      */
    15.     public String name;
    16.     /**
    17.      * 父id ，根节点为0
    18.      */
    19.     public Integer parentId;
    20.     /**
    21.      * 子节点信息
    22.      */
    23.     public List<Menu> childList;
    26.     public Menu(Integer id, String name, Integer parentId) {
    27.         this.id = id;
    28.         this.name = name;
    29.         this.parentId = parentId;
    30.     }
    32.     public Menu(Integer id, String name, Integer parentId, List<Menu> childList) {
    33.         this.id = id;
    34.         this.name = name;
    35.         this.parentId = parentId;
    36.         this.childList = childList;
    37.     }
    39. }
    1.  @Test
    2.     public void testtree(){
    3.         //模拟从数据库查询出来
    4.         List<Menu> menus = Arrays.asList(
    5.                 new Menu(1,"根节点",0),
    6.                 new Menu(2,"子节点1",1),
    7.                 new Menu(3,"子节点1.1",2),
    8.                 new Menu(4,"子节点1.2",2),
    9.                 new Menu(5,"根节点1.3",2),
    10.                 new Menu(6,"根节点2",1),
    11.                 new Menu(7,"根节点2.1",6),
    12.                 new Menu(8,"根节点2.2",6),
    13.                 new Menu(9,"根节点2.2.1",7),
    14.                 new Menu(10,"根节点2.2.2",7),
    15.                 new Menu(11,"根节点3",1),
    16.                 new Menu(12,"根节点3.1",11)
    17.         );
    19.         //获取父节点
    20.         List<Menu> collect = menus.stream().filter(m -> m.getParentId() == 0).map(
    21.                 (m) -> {
    22.                     m.setChildList(getChildrens(m, menus));
    23.                     return m;
    24.                 }
    25.         ).collect(Collectors.toList());
    26.         System.out.println("-------转json输出结果-------");
    27.         System.out.println(JSON.toJSON(collect));
    28.     }
    30.     /**
    31.      * 递归查询子节点
    32.      * @param root  根节点
    33.      * @param all   所有节点
    34.      * @return 根节点信息
    35.      */
    36.     private List<Menu> getChildrens(Menu root, List<Menu> all) {
    37.         List<Menu> children = all.stream().filter(m -> {
    38.             return Objects.equals(m.getParentId(), root.getId());
    39.         }).map(
    40.                 (m) -> {
    41.                     m.setChildList(getChildrens(m, all));
    42.                     return m;
    43.                 }
    44.         ).collect(Collectors.toList());
    45.         return children;
    46.     }