better home

Author: hauner

src/main/groovy/com/github/hauner/openapi/spring/model/EndpointResponse.groovy

@@ -36,7 +36,11 @@ class EndpointResponse {
 
 
     String getResponseType () {
-        main.responseType.name
+        if (hasMultipleResponses ()) {
+            '?'
+        } else {
+            main.responseType.name
+        }
     }
 
     String getContentType () {

src/main/groovy/com/github/hauner/openapi/spring/writer/MethodWriter.groovy

@@ -37,18 +37,10 @@ class MethodWriter {
     void write (Writer target, Endpoint endpoint, EndpointResponse endpointResponse) {
         target.write ("""\
     ${createMappingAnnotation (endpoint, endpointResponse)}
-    ResponseEntity<${getResponseEntityType(endpointResponse)}> ${createMethodName (endpoint, endpointResponse)}(${createParameters(endpoint)});
+    ResponseEntity<${endpointResponse.responseType}> ${createMethodName (endpoint, endpointResponse)}(${createParameters(endpoint)});
 """)
     }
 
-    private String getResponseEntityType (EndpointResponse endpointResponse) {
-        if (endpointResponse.hasMultipleResponses ()) {
-            '?'
-        } else {
-            endpointResponse.responseType
-        }
-    }
-
     private String createMappingAnnotation (Endpoint endpoint, EndpointResponse endpointResponse) {
         String mapping = "${endpoint.method.mappingAnnotation}"
         mapping += "("