使用Java代码返回一个树形结构

Java怎么写

已于 2023-02-27 00:26:00 修改

阅读量525

点赞数

分类专栏：代码专栏文章标签： java Powered by 金山文档

于 2023-02-27 00:13:55 首次发布

本文链接：https://blog.csdn.net/qq_42437161/article/details/129233878

版权

代码专栏专栏收录该内容

1 篇文章 0 订阅

订阅专栏

文章讲述了在Java后端如何构建一个返回公司组织机构的树形结构接口数据。首先定义了TreeDTO数据模型，包含编码、名称、上级编码和子级列表等属性。接着初始化了一个平铺的List数据，然后通过遍历和映射方法将该列表转换成树形结构，以便前端处理。此方法适用于将对象列表转化为树结构的场景。

摘要由CSDN通过智能技术生成

场景：Java后端给前端返回一个公司组织机构的接口数据，要求是树形结构。

定义数据返回模型：

import lombok.Data;

import java.io.Serializable;
import java.util.List;

@Data
public class TreeDTO implements Serializable {

    /**
     * 编码
     */
    private String code;
    /**
     * 名称
     */
    private String name;
    /**
     * 上级编码
     */
    private String parentCode;
    /**
     * 上级名称
     */
    private String parentName;
    /**
     * 子级
     */
    private List<TreeDTO> childTreeDTOList;
}

初始化一个List数据：（具体场景是仓储或者实现层返回list集合数据，此时还是平铺的数据。）

private static List<TreeDTO> listInit() {
        List<TreeDTO> treeDTOList = new ArrayList<TreeDTO>();
        TreeDTO treeDTO1 = new TreeDTO();
        treeDTO1.setCode("code1");
        treeDTO1.setName("name1");
        treeDTO1.setParentCode("0");
        treeDTO1.setParentName("0");

        TreeDTO treeDTO2 = new TreeDTO();
        treeDTO2.setCode("code2");
        treeDTO2.setName("name2");
        treeDTO2.setParentCode("code1");
        treeDTO2.setParentName("name1");

        TreeDTO treeDTO3 = new TreeDTO();
        treeDTO3.setCode("code3");
        treeDTO3.setName("name3");
        treeDTO3.setParentCode("code2");
        treeDTO3.setParentName("name2");

        TreeDTO treeDTO4 = new TreeDTO();
        treeDTO4.setCode("code4");
        treeDTO4.setName("name4");
        treeDTO4.setParentCode("code3");
        treeDTO4.setParentName("name3");

        treeDTOList.add(treeDTO4);
        treeDTOList.add(treeDTO3);
        treeDTOList.add(treeDTO2);
        treeDTOList.add(treeDTO1);
        return treeDTOList;
    }

将List转换为树形结构：（根据parentCode查找子级的数据，组装树结构）

private static List<TreeDTO> changeTree(List<TreeDTO> treeDTOList) {
        Map<String, TreeDTO> treeDTOMap = treeDTOList.stream().collect(Collectors.toMap(treeDTO -> treeDTO.getCode(), treeDTO -> treeDTO));
        List<TreeDTO> targetTreeDTOList = new ArrayList<TreeDTO>();
        Iterator<TreeDTO> iterator = treeDTOList.iterator();
        while (iterator.hasNext()) {
            TreeDTO it = iterator.next();
            if (treeDTOMap.containsKey(it.getParentCode())) {
                TreeDTO innerDTO = treeDTOMap.get(it.getParentCode());
                List<TreeDTO> child = innerDTO.getChildTreeDTOList();
                if (CollectionUtils.isEmpty(child)) {
                    child = new ArrayList<TreeDTO>();
                    child.add(it);
                    innerDTO.setChildTreeDTOList(child);
                } else {
                    child.add(it);
                }
            } else {
                targetTreeDTOList.add(it);
            }
        }
        return targetTreeDTOList;
    }