全局响应消息实现

Author: JeffLi1993

src/main/java/com/spring4all/swagger/SwaggerAutoConfiguration.java

@@ -90,19 +90,7 @@ public List<Docket> createRestApi(SwaggerProperties swaggerProperties) {
 
             // 全局响应消息
             if (!swaggerProperties.getApplyDefaultResponseMessages()) {
-
-                ResponseMessageBuilder responseMessageBuilder401 = new ResponseMessageBuilder()
-                        .code(401)
-                        .message(swaggerProperties.getGlobalResponseMessage().getMessage401());
-                if (!StringUtils.isEmpty(swaggerProperties.getGlobalResponseMessage().getModelRef401()))
-                    responseMessageBuilder401.responseModel(
-                            new ModelRef(swaggerProperties.getGlobalResponseMessage().getModelRef401()));
-
-
-                List<ResponseMessage> responseMessages = new ArrayList();
-                responseMessages.add(responseMessageBuilder401.build());
-                docketForBuilder.useDefaultResponseMessages(swaggerProperties.getApplyDefaultResponseMessages())
-                        .globalResponseMessage(RequestMethod.GET,responseMessages);
+                buildGlobalResponseMessage(swaggerProperties, docketForBuilder);
             }
 
             Docket docket = docketForBuilder.select()
@@ -176,11 +164,15 @@ public List<Docket> createRestApi(SwaggerProperties swaggerProperties) {
         return docketList;
     }
 
+
+
     @Override
     public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
         this.beanFactory = beanFactory;
     }
 
+
+
     private List<Parameter> buildGlobalOperationParametersFromSwaggerProperties(
             List<SwaggerProperties.GlobalOperationParameter> globalOperationParameters) {
         List<Parameter> parameters = Lists.newArrayList();
@@ -232,4 +224,31 @@ private List<Parameter> assemblyGlobalOperationParameters(
         resultOperationParameters.addAll(docketOperationParameters);
         return buildGlobalOperationParametersFromSwaggerProperties(resultOperationParameters);
     }
+
+    /**
+     * 设置全局响应消息
+     *
+     * @param swaggerProperties
+     * @param docketForBuilder
+     */
+    private void buildGlobalResponseMessage(SwaggerProperties swaggerProperties, Docket docketForBuilder) {
+        List<ResponseMessage> responseMessages = new ArrayList();
+        List<SwaggerProperties.GlobalResponseMessage> globalResponseMessages =
+                swaggerProperties.getGlobalResponseMessages();
+        for (SwaggerProperties.GlobalResponseMessage globalResponseMessage : globalResponseMessages) {
+            ResponseMessageBuilder responseMessageBuilder = new ResponseMessageBuilder();
+            responseMessageBuilder
+                    .code(globalResponseMessage.getCode())
+                    .message(globalResponseMessage.getMessage());
+
+            if (!StringUtils.isEmpty(globalResponseMessage.getModelRef())) {
+                responseMessageBuilder
+                        .responseModel(new ModelRef(globalResponseMessage.getModelRef()));
+            }
+            responseMessages.add(responseMessageBuilder.build());
+        }
+
+        docketForBuilder.useDefaultResponseMessages(swaggerProperties.getApplyDefaultResponseMessages())
+                .globalResponseMessage(RequestMethod.GET,responseMessages);
+    }
 }

src/main/java/com/spring4all/swagger/SwaggerProperties.java

@@ -58,7 +58,7 @@ public class SwaggerProperties {
     private Boolean applyDefaultResponseMessages = true;
 
     /** 全局响应消息 ，目前支持 GET **/
-    private GlobalResponseMessage globalResponseMessage;
+    private List<GlobalResponseMessage> globalResponseMessages;
 
     @Data
     @NoArgsConstructor
@@ -128,11 +128,14 @@ public static class Contact {
     @NoArgsConstructor
     public static class GlobalResponseMessage {
 
-        /** 401 响应消息 **/
-        private String message401;
+        /** 响应码 **/
+        private int code;
 
-        /** 401 响应体 **/
-        private String modelRef401;
+        /** 响应消息 **/
+        private String message;
+
+        /** 响应体 **/
+        private String modelRef;
     }
 
 }