$563

jjnnzb · Feb 8, 2020 · 5f9171b · 5f9171b
1 parent 018d75b
commit 5f9171b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/...ain/java/springfox/documentation/spring/web/readers/operation/ResponseMessagesReader.java b/...ain/java/springfox/documentation/spring/web/readers/operation/ResponseMessagesReader.java
@@ -28,9 +28,11 @@
 import springfox.documentation.builders.ResponseMessageBuilder;
 import springfox.documentation.schema.ModelReference;
 import springfox.documentation.schema.TypeNameExtractor;
+import springfox.documentation.schema.plugins.SchemaPluginsManager;
 import springfox.documentation.service.ResponseMessage;
 import springfox.documentation.spi.DocumentationType;
 import springfox.documentation.spi.schema.EnumTypeDeterminer;
+import springfox.documentation.spi.schema.ViewProviderPlugin;
 import springfox.documentation.spi.schema.contexts.ModelContext;
 import springfox.documentation.spi.service.OperationBuilderPlugin;
 import springfox.documentation.spi.service.contexts.OperationContext;
@@ -51,13 +53,16 @@ public class ResponseMessagesReader implements OperationBuilderPlugin {
 
   private final EnumTypeDeterminer enumTypeDeterminer;
   private final TypeNameExtractor typeNameExtractor;
+  private final SchemaPluginsManager pluginsManager;
 
   @Autowired
   public ResponseMessagesReader(
       EnumTypeDeterminer enumTypeDeterminer,
-      TypeNameExtractor typeNameExtractor) {
+      TypeNameExtractor typeNameExtractor,
+      SchemaPluginsManager pluginsManager) {
     this.enumTypeDeterminer = enumTypeDeterminer;
     this.typeNameExtractor = typeNameExtractor;
+    this.pluginsManager = pluginsManager;
   }
 
   @Override
@@ -78,10 +83,14 @@ private void applyReturnTypeOverride(OperationContext context) {
     int httpStatusCode = httpStatusCode(context);
     String message = message(context);
     ModelReference modelRef = null;
+
+    ViewProviderPlugin viewProvider = 
+        pluginsManager.viewProvider(context.getDocumentationContext().getDocumentationType());
+
     if (!isVoid(returnType)) {
       ModelContext modelContext = context.operationModelsBuilder().addReturn(
           returnType,
-          Optional.empty());
+          viewProvider.viewFor(returnType, context));
 
       Map<String, String> knownNames = new HashMap<>();
       Optional.ofNullable(context.getKnownModels().get(modelContext.getParameterId()))

diff --git a/...ingfox/documentation/spring/web/readers/operation/DefaultResponseMessageReaderSpec.groovy b/...ingfox/documentation/spring/web/readers/operation/DefaultResponseMessageReaderSpec.groovy
@@ -48,7 +48,10 @@ class DefaultResponseMessageReaderSpec extends DocumentationContextSpec {
         new TypeResolver(),
         modelNameRegistry,
         new JacksonEnumTypeDeterminer())
-    sut = new ResponseMessagesReader(new JacksonEnumTypeDeterminer(), typeNameExtractor)
+    sut = new ResponseMessagesReader(
+        new JacksonEnumTypeDeterminer(),
+        typeNameExtractor,
+        defaultSchemaPlugins())
   }
 
   def "Should add default response messages"() {