Protobuf Schema Support Multiple Response

Author: lucy66hw

bin/configs/protobuf-schema-config.yaml

@@ -8,6 +8,7 @@ additionalProperties:
   numberedFieldNumberList: true
   startEnumsWithUnspecified: true
   wrapComplexType: false
+  supportMultipleResponses: true
   aggregateModelsName: data
 typeMappings:
   object: "google.protobuf.Struct"

docs/generators/protobuf-schema.md

@@ -22,6 +22,7 @@ These options may be applied as additional-properties (cli) or configOptions (pl
 |aggregateModelsName|Aggregated model filename. If set, all generated models will be combined into this single file.| |null|
 |numberedFieldNumberList|Field numbers in order.| |false|
 |startEnumsWithUnspecified|Introduces "UNSPECIFIED" as the first element of enumerations.| |false|
+|supportMultipleResponses|Support multiple responses| |false|
 |wrapComplexType|Generate Additional message for complex type| |true|
 
 ## IMPORT MAPPING

modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/ProtobufSchemaCodegen.java

@@ -17,6 +17,7 @@
 package org.openapitools.codegen.languages;
 
 import io.swagger.v3.oas.models.OpenAPI;
+import io.swagger.v3.oas.models.PathItem;
 import io.swagger.v3.oas.models.media.ArraySchema;
 import io.swagger.v3.oas.models.media.MapSchema;
 import io.swagger.v3.oas.models.media.ObjectSchema;
@@ -68,6 +69,8 @@ public class ProtobufSchemaCodegen extends DefaultCodegen implements CodegenConf
 
     public static final String AGGREGATE_MODELS_NAME = "aggregateModelsName";
 
+    public static final String SUPPORT_MULTIPLE_RESPONSES = "supportMultipleResponses";
+
     private final Logger LOGGER = LoggerFactory.getLogger(ProtobufSchemaCodegen.class);
 
     @Setter protected String packageName = "openapitools";
@@ -82,6 +85,8 @@ public class ProtobufSchemaCodegen extends DefaultCodegen implements CodegenConf
 
     private boolean wrapComplexType = true;
 
+    private boolean supportMultipleResponses = false;
+
     @Override
     public CodegenType getTag() {
         return CodegenType.SCHEMA;
@@ -192,6 +197,7 @@ public ProtobufSchemaCodegen() {
         addSwitch(START_ENUMS_WITH_UNSPECIFIED, "Introduces \"UNSPECIFIED\" as the first element of enumerations.", startEnumsWithUnspecified);
         addSwitch(ADD_JSON_NAME_ANNOTATION, "Append \"json_name\" annotation to message field when the specification name differs from the protobuf field name", addJsonNameAnnotation);
         addSwitch(WRAP_COMPLEX_TYPE, "Generate Additional message for complex type", wrapComplexType);
+        addSwitch(SUPPORT_MULTIPLE_RESPONSES, "Support multiple responses", supportMultipleResponses);
         addOption(AGGREGATE_MODELS_NAME, "Aggregated model filename. If set, all generated models will be combined into this single file.", null);
     }
 
@@ -239,6 +245,10 @@ public void processOpts() {
             this.setAggregateModelsName((String) additionalProperties.get(AGGREGATE_MODELS_NAME));
         }
 
+        if(additionalProperties.containsKey(this.SUPPORT_MULTIPLE_RESPONSES)) {
+            this.supportMultipleResponses = convertPropertyToBooleanAndWriteBack(SUPPORT_MULTIPLE_RESPONSES);
+        }
+
         supportingFiles.add(new SupportingFile("README.mustache", "", "README.md"));
     }
 
@@ -442,7 +452,6 @@ private void wrapModels() {
             } else if (ModelUtils.isAnyOf(schema)) {
                 wrapComposedChildren(schema.getAnyOf(), visitedSchema);
             }
-
         }
     }
 
@@ -475,7 +484,6 @@ public void preprocessOpenAPI(OpenAPI openAPI) {
         }
     }
 
-
     /**
      * Adds prefix to the enum allowable values
      * NOTE: Enum values use C++ scoping rules, meaning that enum values are siblings of their type, not children of it. Therefore, enum value must be unique
@@ -542,24 +550,28 @@ public void addEnumIndexes(List<Map<String, Object>> enumVars) {
 
     public List<CodegenProperty> processOneOfAnyOfItems(List<CodegenProperty> composedSchemasProperty) {
         for(CodegenProperty cd: composedSchemasProperty) {
-            if (cd.getTitle() != null) {
-                cd.name = cd.getTitle();
-                cd.baseName = cd.getTitle();
-            } else{
-                cd.name = getNameFromDataType(cd);
-                cd.baseName = getNameFromDataType(cd);
-            }
+            cd.name = resolveVarName(cd);
+            cd.baseName = resolveVarName(cd);
         }
         return composedSchemasProperty;
     }
 
+
+    private String resolveVarName(CodegenProperty property) {
+        if(property.getTitle() != null) {
+            return toVarName(property.getTitle());
+        } else {
+            return getNameFromDataType(property);
+        }
+    }
+
     public String getNameFromDataType(CodegenProperty property) {
         if (Boolean.TRUE.equals(property.getIsArray())){
-            return underscore(property.mostInnerItems.dataType + ARRAY_SUFFIX);
+            return toVarName(property.mostInnerItems.dataType + ARRAY_SUFFIX);
         } else if (Boolean.TRUE.equals(property.getIsMap())) {
-            return underscore(property.mostInnerItems.dataType + MAP_SUFFIX);
+            return toVarName(property.mostInnerItems.dataType + MAP_SUFFIX);
         } else {
-            return underscore(property.dataType);
+            return toVarName(property.dataType);
         }
     }
 
@@ -944,6 +956,30 @@ public OperationsMap postProcessOperationsWithModels(OperationsMap objs, List<Mo
                     }
                 }
             }
+
+            if(this.supportMultipleResponses) {
+                int responseIdx = 1;
+                op.vendorExtensions.put("x-grpc-response", op.operationId+"Response");
+                for (CodegenResponse r : op.responses) {
+                    if (r.returnProperty == null) {
+                        r.vendorExtensions.put("x-oneOf-response-type", "google.protobuf.Empty");
+                        r.vendorExtensions.put("x-oneOf-response-name", "empty");
+                    } else if (r.isMap) {
+                        r.vendorExtensions.put("x-oneOf-response-type", r.returnProperty.additionalProperties.dataType);
+                        r.vendorExtensions.put("x-oneOf-response-name", resolveVarName(r.returnProperty.additionalProperties));
+                        LOGGER.warn("Mapping responses for operations with supportMultipleResponses flag (operation ID: {}) is not currently supported.", op.operationId);
+                    } else if (r.isArray) {
+                        r.vendorExtensions.put("x-oneOf-response-type", r.returnProperty.items.dataType);
+                        r.vendorExtensions.put("x-oneOf-response-name", resolveVarName(r.returnProperty.items));
+                        LOGGER.warn("Array responses for operations with supportMultipleResponses flag (operation ID: {}) is not currently supported.", op.operationId);
+                    }
+                    else {
+                        r.vendorExtensions.put("x-oneOf-response-type", r.returnProperty.dataType);
+                        r.vendorExtensions.put("x-oneOf-response-name", resolveVarName(r.returnProperty));
+                    }
+                    r.vendorExtensions.put("x-oneOf-response-index", responseIdx++);
+                }
+            }
         }
 
         if (this.aggregateModelsName != null) {

modules/openapi-generator/src/main/resources/protobuf-schema/api.mustache

@@ -42,5 +42,14 @@ message {{operationId}}Response {
 }
 
 {{/vendorExtensions.x-grpc-response}}
+{{#supportMultipleResponses}}
+message {{operationId}}Response {
+    oneof response {
+    {{#responses}}
+        {{{vendorExtensions.x-oneOf-response-type}}} {{vendorExtensions.x-oneOf-response-name}}_{{code}} = {{vendorExtensions.x-oneOf-response-index}};
+    {{/responses}}
+    }
+}
+{{/supportMultipleResponses}}
 {{/operation}}
 {{/operations}}

modules/openapi-generator/src/test/resources/3_0/protobuf/petstore.yaml

@@ -59,7 +59,7 @@ paths:
               schema:
                 $ref: '#/components/schemas/Pet'
         '405':
-          description: Invalid input
+          $ref: '#/components/responses/ErrorResponse'
       security:
         - petstore_auth:
             - 'write:pets'
@@ -83,11 +83,11 @@ paths:
               schema:
                 $ref: '#/components/schemas/Pet'
         '400':
-          description: Invalid ID supplied
+          $ref: '#/components/responses/ErrorResponse'
         '404':
-          description: Pet not found
+          $ref: '#/components/responses/ErrorResponse'
         '405':
-          description: Validation exception
+          $ref: '#/components/responses/ErrorResponse'
       security:
         - petstore_auth:
             - 'write:pets'
@@ -593,6 +593,13 @@ externalDocs:
   description: Find out more about Swagger
   url: 'http://swagger.io'
 components:
+  responses:
+    ErrorResponse:
+      description: An error response.
+      content:
+        application/json:
+          schema:
+            $ref: '#/components/schemas/Error'
   requestBodies:
     UserArray:
       content:
@@ -786,3 +793,16 @@ components:
           uniqueItems: true
           items:
             type: string
+    Error:
+      type: object
+      required:
+        - code
+        - message
+      properties:
+        code:
+          type: integer
+          format: int32
+          description: Error code.
+        message:
+          type: string
+          description: Detailed error message.

samples/config/petstore/protobuf-schema-config/models/data.proto

@@ -56,6 +56,16 @@ message Dog {
 
 }
 
+message Error {
+
+  // Error code.
+  int32 code = 1;
+
+  // Detailed error message.
+  string message = 2;
+
+}
+
 message Order {
 
   int64 id = 1;

samples/config/petstore/protobuf-schema-config/services/default_service.proto

@@ -16,9 +16,9 @@ import "google/protobuf/empty.proto";
 import public "models/data.proto";
 
 service DefaultService {
-  rpc PetsGet (PetsGetRequest) returns (google.protobuf.Empty);
+  rpc PetsGet (PetsGetRequest) returns (PetsGetResponse);
 
-  rpc PetsPost (PetsPostRequest) returns (google.protobuf.Empty);
+  rpc PetsPost (PetsPostRequest) returns (PetsPostResponse);
 
 }
 
@@ -27,8 +27,18 @@ message PetsGetRequest {
 
 }
 
+message PetsGetResponse {
+    oneof response {
+        google.protobuf.Empty empty_200 = 1;
+    }
+}
 message PetsPostRequest {
   PetsPostRequest pets_post_request = 1;
 
 }
 
+message PetsPostResponse {
+    oneof response {
+        google.protobuf.Empty empty_200 = 1;
+    }
+}

samples/config/petstore/protobuf-schema-config/services/pet_service.proto

@@ -16,21 +16,21 @@ import "google/protobuf/empty.proto";
 import public "models/data.proto";
 
 service PetService {
-  rpc AddPet (AddPetRequest) returns (Pet);
+  rpc AddPet (AddPetRequest) returns (AddPetResponse);
 
-  rpc DeletePet (DeletePetRequest) returns (google.protobuf.Empty);
+  rpc DeletePet (DeletePetRequest) returns (DeletePetResponse);
 
   rpc FindPetsByStatus (FindPetsByStatusRequest) returns (FindPetsByStatusResponse);
 
   rpc FindPetsByTags (FindPetsByTagsRequest) returns (FindPetsByTagsResponse);
 
-  rpc GetPetById (GetPetByIdRequest) returns (Pet);
+  rpc GetPetById (GetPetByIdRequest) returns (GetPetByIdResponse);
 
-  rpc UpdatePet (UpdatePetRequest) returns (Pet);
+  rpc UpdatePet (UpdatePetRequest) returns (UpdatePetResponse);
 
-  rpc UpdatePetWithForm (UpdatePetWithFormRequest) returns (google.protobuf.Empty);
+  rpc UpdatePetWithForm (UpdatePetWithFormRequest) returns (UpdatePetWithFormResponse);
 
-  rpc UploadFile (UploadFileRequest) returns (ApiResponse);
+  rpc UploadFile (UploadFileRequest) returns (UploadFileResponse);
 
 }
 
@@ -40,45 +40,75 @@ message AddPetRequest {
 
 }
 
+message AddPetResponse {
+    oneof response {
+        Pet pet_200 = 1;
+        Error error_405 = 2;
+    }
+}
 message DeletePetRequest {
   // Pet id to delete
   int64 pet_id = 1 [json_name="petId"];
   string api_key = 2;
 
 }
 
+message DeletePetResponse {
+    oneof response {
+        google.protobuf.Empty empty_400 = 1;
+    }
+}
 message FindPetsByStatusRequest {
   // Status values that need to be considered for filter
   repeated string status = 1;
 
 }
 
 message FindPetsByStatusResponse {
-  repeated Pet data = 1;
+    oneof response {
+        Pet pet_200 = 1;
+        google.protobuf.Empty empty_400 = 2;
+    }
 }
-
 message FindPetsByTagsRequest {
   // Tags to filter by
   repeated string tags = 1;
 
 }
 
 message FindPetsByTagsResponse {
-  repeated Pet data = 1;
+    oneof response {
+        Pet pet_200 = 1;
+        google.protobuf.Empty empty_400 = 2;
+    }
 }
-
 message GetPetByIdRequest {
   // ID of pet to return
   int64 pet_id = 1 [json_name="petId"];
 
 }
 
+message GetPetByIdResponse {
+    oneof response {
+        Pet pet_200 = 1;
+        google.protobuf.Empty empty_400 = 2;
+        google.protobuf.Empty empty_404 = 3;
+    }
+}
 message UpdatePetRequest {
   // Pet object that needs to be added to the store
   Pet pet = 1;
 
 }
 
+message UpdatePetResponse {
+    oneof response {
+        Pet pet_200 = 1;
+        Error error_400 = 2;
+        Error error_404 = 3;
+        Error error_405 = 4;
+    }
+}
 message UpdatePetWithFormRequest {
   // ID of pet that needs to be updated
   int64 pet_id = 1 [json_name="petId"];
@@ -89,6 +119,11 @@ message UpdatePetWithFormRequest {
 
 }
 
+message UpdatePetWithFormResponse {
+    oneof response {
+        google.protobuf.Empty empty_405 = 1;
+    }
+}
 message UploadFileRequest {
   // ID of pet to update
   int64 pet_id = 1 [json_name="petId"];
@@ -99,3 +134,8 @@ message UploadFileRequest {
 
 }
 
+message UploadFileResponse {
+    oneof response {
+        ApiResponse api_response_200 = 1;
+    }
+}