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

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

2. 定义数据返回模型：

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;
}

3. 初始化一个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;
    }

4. 将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;
    }

5. 树结构的简单快速做法，只需要把对象换成自己的对象即可。一次通过。仅供参考。