根据数据中父节点id遍历出树状结构

业务中有需求，每个分类都有上一级id（父节点id），根据这个id，组装成树状结构

下面People是我的实体类，Model是树状结构的类

转成json用到了fastjson，主要是可以去json格式转换器那里看具体的结构

		<dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.41</version>
        </dependency>
        <!-- 实体类中我用的lombok -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>

lombok需要下载lombok插件

这个是People实体类，需要组建成树状结构的类

import lombok.Data;
@Data
public class People {
   
    private String id;
    private String name;
    private String pid;
}

Model树状结构的类

import lombok.Data;
import lombok.Getter;
import lombok.Setter;
import java.util.List;
@Data
@Getter
@Setter
public class Model {
   
    private String value;
    private String label;
    private List<Model> children;
}

实际的操作的类

package com.example.demo.gong.tree;
import com.alibaba.fastjson.JSONObject;
import org.apache.commons.lang3.StringUtils;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class TreeUtil {
   
    public static void main(String[] args) {
   
        List<People> peopleList = getList();
        Model parentModel = new Model();
        parentModel.setLabel("开始节点");
        parentModel.setValue("");
        parentModel.setChildren(new ArrayList<>());

        Map<String, People> map = new HashMap<>();

        Map<String, List<Model>> modelMap = new HashMap<>();
        modelMap.put("", new ArrayList<>());

        for (People people : peopleList) {
   
            map.put(people.getId(), people);
        }
        for (People people : peopleList) {
   
            Model model = new Model();
            if (map.get(people.getId()) != null) {
   
                model.setValue(people.getId());
                model.setLabel(people.getName());
                model.setChildren(new ArrayList<>());
            } else {
   
                model.setValue("");
                model.setLabel("")