Java1.8递归根据父节点查询所有子节点

模型

import java.util.List;

/**
 * @Author: lc
 * @Date: 2020/6/4 10:37
 */
public class Menu {
    private String id;
    private String name;
    private String pid;
    private List<Menu> childMenu;

    //get  set 省略。。。。
}

一、递归获取某个父节点下面的所有子节点

/**
     * 递归获取某个父节点下面的所有子节点
     *
     * @param childMenu 要返回的结果
     * @param menuList  数据库查询出来的所有机构集合
     * @param pid      父id
     *                 注:本身的机构节点不会添加进去
     */
    private static void getParentNode(List<Menu> childMenu, List<Menu> menuList, String pid) {
        menuList.stream()
                //过滤出父id等于参数的id
                .filter(menu -> StringUtils.isNotBlank(menu.getPid()) && menu.getPid().equals(pid)) 
                .forEach(menu -> {
                    //递归遍历下一级
                    getParentNode(childMenu, menuList, menu.getId());
                    //添加
                    childMenu.add(menu);
                });
    }

 

1.1、测试数据

public static void main(String args[]) {
        List<Menu> menuList = new ArrayList<Menu>();
        Menu mu = new Menu();
        mu.setId("1");
        mu.setName("阿里巴巴");
        mu.setPid("0");
        Menu mu1 = new Menu();
        mu1.setId("2");
        mu1.setName("开发部");
        mu1.setPid("1");
        Menu mu2 = new Menu();
        mu2.setId("3");
        mu2.setName("业务部");
        mu2.setPid("2");
        Menu mu3 = new Menu();
        mu3.setId("4");
        mu3.setName("销售部");
        mu3.setPid("2");
        Menu mu4 = new Menu();
        mu4.setId("5");
        mu4.setName("运营部");
        mu4.setPid("1");
        Menu mu5 = new Menu();
        mu5.setId("6");
        mu5.setName("采购部");
        mu5.setPid("5");
        Menu mu6 = new Menu();
        mu6.setId("7");
        mu6.setName("保卫部");
        mu6.setPid("5");
        Menu mu7 = new Menu();
        mu7.setId("8");
        mu7.setName("人事部");
        mu7.setPid("7");
        Menu mu8 = new Menu();
        mu8.setId("9");
        mu8.setName("财务部");
        mu8.setPid("1");

        menuList.add(mu);
        menuList.add(mu1);
        menuList.add(mu2);
        menuList.add(mu3);
        menuList.add(mu4);
        menuList.add(mu5);
        menuList.add(mu6);
        menuList.add(mu7);
        menuList.add(mu8);

        List<Menu> menus = new ArrayList<>();
        getMenu(menus, menuList, "2");
        
        menus.stream()
                .map(menu -> menu.getId() + "   " + menu.getName())
                .forEach(System.out::println);
    }

1.2、测试结果

二、根据子节点查询所有父节点

/**
     * 递归获取某个子节点下面的所有父节点
     *
     * @param childMenu 要返回的结果
     * @param menuList  数据库查询出来的所有机构集合
     * @param id        节点id
     *                  TODO   注:本身的机构节点会添加进去
     */
    private static void getChildNode(List<Menu> childMenu, List<Menu> menuList, String id) {

        menuList.stream()
                .filter(menu -> StringUtils.isNotBlank(id) && id.equals(menu.getId()))
                .forEach(menu -> {
                    getChildNode(childMenu, menuList, menu.getPid());
                    childMenu.add(menu);
                });

    }