java返回树型结构数据给前端

在实体类添加一个list集合字段

 private List<User> childList; //提供get，set方法

查询出所有数据

List<User> users = userMapper.selectList(null); 

通过流去读取数据

List<User> collect = users.stream()
                .filter(o -> o.getPid() == 0)
                // 给每个一级分类加子分类
                .peek(o -> o.setChildList(getChildCategoryList(o, users)))
                // 排序
                .sorted(Comparator.comparingInt(User::getId))
                // 收集
                .collect(Collectors.toList());

 // 根据当前分类 找出子类， 并通过递归找出子类的子类
    private List<User> getChildCategoryList(User currMenu, List<User> categories) {
        return categories.stream().filter(o -> o.getPid().equals(currMenu.getId()))
                .peek(o -> o.setChildList(getChildCategoryList(o, categories)))
                .sorted(Comparator.comparingInt(User::getId))
                .collect(Collectors.toList());
    }