springboot项目中构建树结构数据返回

需要构建一个树结构返回（ID,PID），具体示例如下

1、创建实体数据表

-- 测试省份数据表
DROP TABLE IF EXISTS test_province;
CREATE TABLE IF NOT EXISTS test_province (
    id                  BIGINT(32) NOT NULL AUTO_INCREMENT PRIMARY KEY COMMENT '表ID',
    pid                 BIGINT(32) COMMENT '父级ID',
    name                 VARCHAR(255) COMMENT '名称',
    code                 VARCHAR(255) COMMENT '邮政编号',
    type                 INTEGER      COMMENT '类型(1=省, 2=市, 3=区/县)',
    abbreviation         VARCHAR(255) COMMENT '简称',

    status        VARCHAR(10) COMMENT '状态（0=正常）' DEFAULT '0',
    remarks       VARCHAR(255) COMMENT '备注',
    create_by     VARCHAR(50) COMMENT '创建人',
    create_time   TIMESTAMP DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
    update_by     VARCHAR(50) COMMENT '更新人',
    update_time   TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间'
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT '测试省份数据表';

-- 修改字段名
ALTER TABLE test_province CHANGE code postal_code INT DEFAULT 000000 COMMENT '邮政编号';
ALTER TABLE test_province MODIFY COLUMN postal_code VARCHAR(255);
-- 如果是PostgreSQL，语法是：
--  ALTER TABLE test_province ALTER COLUMN postal_code TYPE VARCHAR(255);
-- 添加字段名
ALTER TABLE test_province ADD COLUMN area DECIMAL(10,2) DEFAULT NULL COMMENT '面积（保留2位小数）';
ALTER TABLE test_province ADD COLUMN area_code VARCHAR(255) DEFAULT NULL COMMENT '区号';

添加数据

INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (0, '广东省', '0', 1, null, null, '020');
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (0, '北京市', '0', 1, null, null, '010');
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (1, '广州市', '0', 2, null, null, null);
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (1, '深圳市', '0', 2, null, null, '0755');
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (2, '朝阳区', '0', 3, null, null, null);
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (2, '海淀区', '0', 3, null, null, null);
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (3, '天河区', '0', 3, null, null, null);
INSERT INTO test_province (pid, name, postal_code, type, abbreviation, area, area_code) VALUES (4, '福田区', '0', 3, null, null, null);

2、创建一个实体

import lombok.Data;

import java.util.List;

@Data
public class Province {
    private Integer id;
    private Integer pid;
    private String name;
    private String postalCode;
    private Integer type;
    private String abbreviation;
    private Double area;
    private String areaCode;
    // 子级数据
    private List children;
}

3、编写代码（返回树结构方法）

/**
     * 构建父子关系的数据列表（树形结构）
     *
     * @param allData 已查询的所有数据列表
     * @param  数据类型
     * @return 父子关系结构化数据列表
     */
    public static  List buildHierarchicalData(List allData) {
        // 存储每个对象的ID到对象的映射
        Map dataMap = new HashMap<>();
        // 存储顶级节点
        List rootData = new ArrayList<>();
        // 遍历所有数据并放入 dataMap 中
        for (T item : allData) {
            try {
                // 通过反射获取 item 的 parentId 和 id
                Integer parentId = (Integer) item.getClass().getMethod("getPid").invoke(item);
                Integer id = (Integer) item.getClass().getMethod("getId").invoke(item);

                dataMap.put(id, item);

                // 如果是顶级节点，添加到 rootData 中
                if (parentId == 0) {
                    rootData.add(item);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        // 构建父子关系
        for (T item : allData) {
            try {
                Integer parentId = (Integer) item.getClass().getMethod("getPid").invoke(item);
                if (parentId != 0) {
                    T parent = dataMap.get(parentId);
                    if (parent != null) {
                        // 获取父节点的 children 列表并添加当前节点
                        List children = (List) parent.getClass().getMethod("getChildren").invoke(parent);
                        if (children == null) {
                            children = new ArrayList<>();
                            parent.getClass().getMethod("setChildren", List.class).invoke(parent, children);
                        }
                        children.add(item);
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        // 返回顶级节点列表
        return rootData;
    }

4、编写接口

    // controller
    @GetMapping("/getProvinceListInfo")
    public ResponseEntity