java树结构的构建方式

1.Service层方法

    public List<DsModelType> getModelTypeTree() {
        // 一次性查询所有数据
        List<DsModelType> allModelTypes = dsModelTypeMapper.selectDsModelTypeList(new DsModelType());

        // 使用 Map 来构建模型类型的快速索引
        Map<Long, DsModelType> modelTypeMap = allModelTypes.stream()
                .collect(Collectors.toMap(DsModelType::getModelTypeId, type -> type));


        // 构建树状结构
        List<DsModelType> tree = new ArrayList<>();
        for (DsModelType type : allModelTypes) {
            if (Objects.isNull(type.getParentId()) || type.getParentId().equals(ROOT_DIR_ID)) {
                // 顶层节点
                tree.add(type);
            } else {
                // 子节点添加到父节点的 children 列表中
                DsModelType parent = modelTypeMap.get(type.getParentId());
                if (parent != null) {
                    if (parent.getChildren() == null) {
                        parent.setChildren(new ArrayList<>());
                    }
                    parent.getChildren().add(type);
                }
            }
        }

        return tree;
    }

2. 实体类:

@Data
@NoArgsConstructor
@AllArgsConstructor
@ApiModel(value = "数据集模型类型")
public class DsModelType extends BaseEntity {

    private static final long serialVersionUID = 1L;

    /**
     * 主键
     */
    private Long modelTypeId;

    /**
     * 模型名称
     */
    @Excel(name = "模型名称")
    @ApiModelProperty("模型名称")
    @NotEmpty(message = "模型名称不能为空")
    private String name;

    /**
     * 父模型
     */
    @Excel(name = "父模型")
    @ApiModelProperty("父模型")
    private Long parentId;

    /**
     * 级别（一般两级，虫害下的棉铃虫）
     */
    @ApiModelProperty("级别")
    @Excel(name = "级别", readConverterExp = "一=般两级，虫害下的棉铃虫")
    private Long level;


    @ApiModelProperty("子模型")
    private List<DsModelType> children;

}