Java查询返回值为树形结构

最新推荐文章于 2024-03-20 22:39:31 发布

于优秀~

最新推荐文章于 2024-03-20 22:39:31 发布

阅读量522

点赞数 1

原文链接：https://blog.csdn.net/coding_lin/article/details/113447422

版权

数据库结构
在这里插入图片描述

返回值DTO

package cn.yibiao163.unit.api.dto;

import cn.yibiao163.common.web.dto.AbstractResultDTO;
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.util.List;

/**
 * 国民经济行业分类与代码表
 *
 * @author yyh
 * @date 2021-07-30 15:06
 */
@Data
@EqualsAndHashCode(callSuper = true)
public class ApiUnitIndustryCategoryResponseDTO extends AbstractResultDTO {
    private static final long serialVersionUID = 1L;

    /**
     * 国民经济行业分类与代码表主键
     */
    private String id;

    /**
     * 级别
     */
    private Integer level;

    /**
     * 行业代码
     */
    private String industryCode;

    /**
     * 行业名称
     */
    private String industryName;

    /**
     * 父ID
     */
    private String parentId;

    /**
     * 子节点
     */
    private List<ApiUnitIndustryCategoryResponseDTO> children;

}

controller层

package cn.yibiao163.unit.api.controller;

import cn.yibiao163.common.web.base.AbstractBaseController;
import cn.yibiao163.common.web.dto.ResultDTO;
import cn.yibiao163.unit.api.dto.ApiUnitIndustryCategoryResponseDTO;
import cn.yibiao163.unit.service.UnitIndustryCategoryService;
import lombok.AllArgsConstructor;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

/**
 * 国民经济行业分类与代码表
 */
@RestController
@RequestMapping("/api/unit/v1/unitIndustryCategory")
@AllArgsConstructor
public class ApiUnitIndustryCategoryController extends AbstractBaseController {
    private final UnitIndustryCategoryService unitIndustryCategoryService;

    /**
     * 查询所有行业分类列表
     *
     * @return 行业分类树形结构
     */
    @GetMapping("/dep")
    public ResultDTO<List<ApiUnitIndustryCategoryResponseDTO>> getUnitIndustryCategoryTree() {
        return ResultDTO.success(unitIndustryCategoryService.getUnitIndustryCategoryTree());
    }
}

service层

/**
     * 查询所有行业分类列表
     *
     */
    public List<ApiUnitIndustryCategoryResponseDTO> getUnitIndustryCategoryTree() {
        List<UnitIndustryCategory> entities = list();
        //把UnitIndustryCategory的数组entities包装成ApiUnitIndustryCategoryResponseDTO数组
        List<ApiUnitIndustryCategoryResponseDTO> responseDTOList = entities.stream().map(unitIndustryCategory -> {
            ApiUnitIndustryCategoryResponseDTO responseVO = new ApiUnitIndustryCategoryResponseDTO();
            BeanUtils.copyProperties(unitIndustryCategory, responseVO);
            return responseVO;
        }).collect(Collectors.toList());
         //filter是java8的写法，取得所有parentId为0的数据，也就是一级目录
         //map也是java8的写法，用于封装数据，取得他的孩子（也就是下级目录）的数据
        List<ApiUnitIndustryCategoryResponseDTO> list = responseDTOList.stream().filter(responseVO ->
                "0".equals(responseVO.getParentId())
        ).map((menu) -> {
            menu.setChildren(getChildrenData(menu, responseDTOList));
            return menu;
        }).collect(Collectors.toList());
        return list;
    }


    /**
     * 获取孩子（下级目录）的方法，递归实现
     */
    private List<ApiUnitIndustryCategoryResponseDTO> getChildrenData(ApiUnitIndustryCategoryResponseDTO root, List<ApiUnitIndustryCategoryResponseDTO> all) {
        List<ApiUnitIndustryCategoryResponseDTO> children = all.stream().filter(responseVO ->
                responseVO.getParentId().equals(root.getId())
        ).map(responseVO -> {
            responseVO.setChildren(getChildrenData(responseVO, all));
            return responseVO;
        }).collect(Collectors.toList());
        return children;
    }