From 668a26ddbf358c812e88e9799489002ea0d59d0c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8D=AD=E7=90=AA=E6=9E=AB?= <374696376@qq.com>
Date: Fri, 5 May 2023 22:42:46 +0800
Subject: [PATCH] add swagger support

---
 .../iotkit/manager/config/SwaggerConfig.java  | 35 +++++++++++++++----
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/iot-standalone/src/main/java/cc/iotkit/manager/config/SwaggerConfig.java b/iot-standalone/src/main/java/cc/iotkit/manager/config/SwaggerConfig.java
index a2ff5d78..e9145ddc 100644
--- a/iot-standalone/src/main/java/cc/iotkit/manager/config/SwaggerConfig.java
+++ b/iot-standalone/src/main/java/cc/iotkit/manager/config/SwaggerConfig.java
@@ -4,15 +4,16 @@ import com.github.xiaoymin.knife4j.spring.annotations.EnableKnife4j;
 import io.swagger.annotations.Api;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import springfox.documentation.builders.ApiInfoBuilder;
-import springfox.documentation.builders.PathSelectors;
-import springfox.documentation.builders.RequestHandlerSelectors;
-import springfox.documentation.service.ApiInfo;
-import springfox.documentation.service.Contact;
+import org.springframework.http.HttpMethod;
+import springfox.documentation.builders.*;
+import springfox.documentation.service.*;
 import springfox.documentation.spi.DocumentationType;
 import springfox.documentation.spring.web.plugins.Docket;
 import springfox.documentation.swagger2.annotations.EnableSwagger2;
 
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * @Author: 石恒
  * @Date: 2023/5/4 20:12
@@ -30,7 +31,10 @@ public class SwaggerConfig {
                 .select()
                 .apis(RequestHandlerSelectors.withClassAnnotation(Api.class))
                 .paths(PathSelectors.any())
-                .build();
+                .build()
+                .globalRequestParameters(getGlobalRequestParameters())
+                .globalResponses(HttpMethod.GET, getGlobalResponseMessage())
+                .globalResponses(HttpMethod.POST, getGlobalResponseMessage());
     }
 
     @Bean
@@ -43,6 +47,25 @@ public class SwaggerConfig {
                 .build();
     }
 
+    /**
+     * 添加head参数配置
+     */
+    private List<RequestParameter> getGlobalRequestParameters() {
+        List<RequestParameter> parameters = new ArrayList<>();
+        parameters.add(new RequestParameterBuilder()
+                .name("token")
+                .description("令牌")
+                .required(false)
+                .in(ParameterType.HEADER)
+                .build());
+        return parameters;
+    }
+    private List<Response> getGlobalResponseMessage() {
+        List<Response> responseList = new ArrayList<>();
+        responseList.add(new ResponseBuilder().code("404").description("找不到资源").build());
+        return responseList;
+    }
+
 }