有没有办法在使用 springdoc-openapi-maven-plugin 生成的合约中拥有带有鉴别器部分的 CompositSchema？

Question

我有一个示例 SpringBoot API，具有以下功能：

• 1 个控制器，公开可通过 GET 请求调用的单个端点，并返回自定义类（在我的示例中为 ContainerClass）
• ContainerClass 包含一个属性列表
• ParentClass 是一个抽象类，有 2 个子类：ChildA 和 ChildB

我尝试使用 springdoc-openapi-maven-plugin 从该 API 生成 OpenApi 合约。

在我的 pom.xml 中，我有以下元素：

• SpringBoot版本：2.2.6
• org.springdoc:springdoc-openapi-ui:1.4.1
• org.springdoc:springdoc-openapi-maven-plugin:1.0

这是我生成架构的类。

import io.swagger.v3.oas.annotations.media.ArraySchema;
import io.swagger.v3.oas.annotations.media.Schema;

public class ContainerClass {

    @ArraySchema(
        arraySchema = @Schema(discriminatorProperty = "classType"), 
        schema = @Schema(implementation = ParentClass.class)
    )
    public List<ParentClass> elements;
    
    // + Getter/Setter
}

@JsonTypeInfo(
        use = JsonTypeInfo.Id.NAME,
        include = JsonTypeInfo.As.EXISTING_PROPERTY,
        property = "classType",
        defaultImpl = ParentClass.class,
        visible = true)
@JsonSubTypes({
        @JsonSubTypes.Type(value = ChildA.class, name = "CHILD_A"),
        @JsonSubTypes.Type(value = ChildB.class, name = "CHILD_B")})
@Schema(
        description = "Parent description",
        discriminatorProperty = "classType",
        discriminatorMapping = {
                @DiscriminatorMapping(value = "CHILD_A", schema = ChildA.class),
                @DiscriminatorMapping(value = "CHILD_B", schema = ChildB.class)
        }
)
public abstract class ParentClass {

    public String classType;

    // + Getter/Setter
}

@io.swagger.v3.oas.annotations.media.Schema(description = " Child A", allOf = ParentClass.class)
public class ChildA extends ParentClass{
}

@io.swagger.v3.oas.annotations.media.Schema(description = " Child B", allOf = ParentClass.class)
public class ChildB extends ParentClass{
}

当我运行 springdoc-openapi-maven-plugin 时，我得到以下合同文件。

openapi: 3.0.1
info:
  title: OpenAPI definition
  version: v0
servers:
- url: http://localhost:8080
  description: Generated server url
paths:
  /container:
    get:
      tags:
      - hello-controller
      operationId: listElements
      responses:
        "200":
          description: OK
          content:
            '*/*':
              schema:
                $ref: '#/components/schemas/ContainerClass'
components:
  schemas:
    ChildA:
      type: object
      description: ' Child A'
      allOf:
      - $ref: '#/components/schemas/ParentClass'
    ChildB:
      type: object
      description: ' Child B'
      allOf:
      - $ref: '#/components/schemas/ParentClass'
    ContainerClass:
      type: object
      properties:
        elements:
          type: array
          description: array schema description
          items:
            oneOf:
            - $ref: '#/components/schemas/ChildA'
            - $ref: '#/components/schemas/ChildB'
    ParentClass:
      type: object
      properties:
        classType:
          type: string
      description: Parent description
      discriminator:
        propertyName: classType
        mapping:
          CHILD_A: '#/components/schemas/ChildA'
          CHILD_B: '#/components/schemas/ChildB'

实际上，在我的上下文中，为了不对现有消费者进行任何重大更改，我需要 ContainerClass 模式中的 items 属性来包含 ParentClass 模式中包含的 discriminator 部分，如下所示：

ContainerClass:
  type: object
  properties:
    elements:
      type: array
      description: array schema description
      items:
        discriminator:
          propertyName: classType
          mapping:
            CHILD_A: '#/components/schemas/ChildA'
            CHILD_B: '#/components/schemas/ChildB'
        oneOf:
        - $ref: '#/components/schemas/ChildA'
        - $ref: '#/components/schemas/ChildB'

当我尝试在注释中设置属性时，我无法做到这一点。当我调试 io.swagger.v3.core.jackson.ModelResolver 的代码时，我无法找到一种方法来做到这一点。 到目前为止，我还没有找到对我有帮助的代码示例。

有没有一种方法可以让 CompedSchema（在我的例子中包含在 ContainerClass 中的数组）具有由 springdoc-openapi-maven-plugin 执行生成的 disciminator 部分？

Answer 1

这是默认的生成结构。由 swagger-api 直接处理（而不是 springdoc-openapi。

生成的 OpenAPI 描述看起来很正确。

使用springdoc-openapi，您可以定义一个OpenApiCustomiser Bean，您可以在其中更改OpenAPI级别上定义的组件元素的元素：

• https://springdoc.org/faq.html#_how_can_i_customise_the_openapi_object

Answer 2

这是我通过定义 OpenApiCustomiser Bean 的解决方案：

@Bean
    public OpenApiCustomiser myCustomiser() {

        Map<String, String> classTypeMapping = Map.ofEntries(
                new AbstractMap.SimpleEntry<String, String>("CHILD_A", "#/components/schemas/ChildA"),
                new AbstractMap.SimpleEntry<String, String>("CHILD_B", "#/components/schemas/ChildB")
        );

        Discriminator classTypeDiscriminator = new Discriminator().propertyName("classType")
                                                        .mapping(classTypeMapping);

        return openApi -> openApi.getComponents().getSchemas().values()
                                .stream()
                                .filter(schema -> "ContainerClass".equals(schema.getName()))
                                .map(schema -> schema.getProperties().get("elements"))
                                .forEach(arraySchema -> ((ArraySchema)arraySchema).getItems().discriminator(classTypeDiscriminator));
    }

我在合同文件中得到了预期的结果。