zhuang 1 рік тому
батько
коміт
277dfeef0d

+ 11 - 1
business-common/src/main/java/com/rongwei/bscommon/sys/config/SwaggerConfig.java

@@ -5,13 +5,19 @@ import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import springfox.documentation.builders.ApiInfoBuilder;
+import springfox.documentation.builders.ParameterBuilder;
 import springfox.documentation.builders.PathSelectors;
 import springfox.documentation.builders.RequestHandlerSelectors;
+import springfox.documentation.schema.ModelRef;
 import springfox.documentation.service.ApiInfo;
+import springfox.documentation.service.Parameter;
 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;
+
 @Configuration // 标明是配置类
 @EnableSwagger2 //开启swagger功能
 public class SwaggerConfig {
@@ -20,6 +26,10 @@ public class SwaggerConfig {
 
     @Bean
     public Docket createRestApi() {
+        ParameterBuilder tokenPar = new ParameterBuilder();
+        List<Parameter> pars = new ArrayList<>();
+        tokenPar.name("Incontrol-Token").description("令牌").defaultValue("").modelRef(new ModelRef("string")).parameterType("header").required(false).build();
+        pars.add(tokenPar.build());
         return new Docket(DocumentationType.SWAGGER_2).enable(enable)  // DocumentationType.SWAGGER_2 固定的,代表swagger2
 //                .groupName("分布式任务系统") // 如果配置多个文档的时候,那么需要配置groupName来分组标识
                 .apiInfo(apiInfo()) // 用于生成API信息
@@ -27,7 +37,7 @@ public class SwaggerConfig {
                 //.apis(RequestHandlerSelectors.basePackage("com.rongwei.bsserver.controller")) // 用于指定扫描哪个包下的接口
                 .apis(RequestHandlerSelectors.withClassAnnotation(Api.class))
                 .paths(PathSelectors.any())// 选择所有的API,如果你想只为部分API生成文档,可以配置这里
-                .build();
+                .build().globalOperationParameters(pars);
     }
 
     /**