115 lines
4.6 KiB
Java
115 lines
4.6 KiB
Java
package com.zhgd.config;
|
||
|
||
import com.google.common.base.Function;
|
||
import com.google.common.base.Optional;
|
||
import com.google.common.base.Predicate;
|
||
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
||
import org.springframework.context.annotation.Bean;
|
||
import org.springframework.context.annotation.Configuration;
|
||
import springfox.documentation.RequestHandler;
|
||
import springfox.documentation.builders.ApiInfoBuilder;
|
||
import springfox.documentation.builders.PathSelectors;
|
||
import springfox.documentation.service.ApiInfo;
|
||
import springfox.documentation.service.ApiKey;
|
||
import springfox.documentation.service.AuthorizationScope;
|
||
import springfox.documentation.service.SecurityReference;
|
||
import springfox.documentation.spi.DocumentationType;
|
||
import springfox.documentation.spi.service.contexts.SecurityContext;
|
||
import springfox.documentation.spring.web.plugins.Docket;
|
||
import springfox.documentation.swagger2.annotations.EnableSwagger2;
|
||
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
|
||
|
||
/**
|
||
* @program: Iot
|
||
* @description: Swagger配置类
|
||
* @author: Mr.Wang
|
||
* @create: 2019-04-16 14:46
|
||
**/
|
||
@Configuration
|
||
@EnableSwagger2
|
||
@ConditionalOnProperty(name = "swagger.enable", havingValue = "true")
|
||
public class SwaggerConfig {
|
||
// 定义分隔符
|
||
private static final String splitor = ";";
|
||
|
||
@Bean
|
||
public Docket createZwRestApi() {
|
||
return new Docket(DocumentationType.SWAGGER_2).groupName("2.政务接口").apiInfo(apiInfo("政务管理API", "政务前后端联调api文档", "1.0")).select()
|
||
//.apis(RequestHandlerSelectors.basePackage("com.zhgd.xmgl"))
|
||
.apis(basePackage("com.zhwl.zw")).paths(PathSelectors.any())
|
||
//.paths(PathSelectors.regex("/zw.*"))
|
||
.build();
|
||
}
|
||
|
||
@Bean
|
||
public Docket createRestApi() {
|
||
return new Docket(DocumentationType.SWAGGER_2).groupName("1.项目接口").apiInfo(apiInfo("项目管理API", "项目前后端联调api文档", "1.0")).select()
|
||
//.apis(RequestHandlerSelectors.basePackage("com.zhgd.xmgl"))
|
||
.apis(basePackage("com.zhgd.xmgl"))
|
||
//.apis(basePackage("com.zhgd.xmgl"+splitor+"com.zhgd.xmgl.device"))
|
||
.paths(PathSelectors.any())
|
||
|
||
//.paths(PathSelectors.regex("/xmgl.*"))
|
||
.build()
|
||
/* 设置安全模式，swagger可以设置访问token */.securitySchemes(securitySchemes()).securityContexts(securityContexts());
|
||
}
|
||
|
||
private ApiInfo apiInfo(String titleName, String description, String version) {
|
||
return new ApiInfoBuilder().title(titleName).description(description).version(version).build();
|
||
}
|
||
|
||
public static Predicate<RequestHandler> basePackage(final String basePackage) {
|
||
return input -> declaringClass(input).transform(handlerPackage(basePackage)).or(true);
|
||
}
|
||
|
||
private static Function<Class<?>, Boolean> handlerPackage(final String basePackage) {
|
||
return input -> {
|
||
// 循环判断匹配
|
||
for (String strPackage : basePackage.split(splitor)) {
|
||
boolean isMatch = input.getPackage().getName().startsWith(strPackage);
|
||
if (isMatch) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
};
|
||
}
|
||
|
||
private static Optional<? extends Class<?>> declaringClass(RequestHandler input) {
|
||
return Optional.fromNullable(input.declaringClass());
|
||
}
|
||
|
||
/**
|
||
* 安全模式，这里指定token通过Authorization头请求头传递
|
||
*/
|
||
private List<ApiKey> securitySchemes() {
|
||
List<ApiKey> apiKeyList = new ArrayList<ApiKey>();
|
||
apiKeyList.add(new ApiKey("Authorization", "Authorization", "header"));
|
||
return apiKeyList;
|
||
}
|
||
|
||
/**
|
||
* 安全上下文
|
||
*/
|
||
private List<SecurityContext> securityContexts() {
|
||
List<SecurityContext> securityContexts = new ArrayList<>();
|
||
securityContexts.add(SecurityContext.builder().securityReferences(defaultAuth()).forPaths(PathSelectors.regex("^(?!auth).*$")).build());
|
||
return securityContexts;
|
||
}
|
||
|
||
/**
|
||
* 默认的安全上引用
|
||
*/
|
||
private List<SecurityReference> defaultAuth() {
|
||
AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverything");
|
||
AuthorizationScope[] authorizationScopes = new AuthorizationScope[1];
|
||
authorizationScopes[0] = authorizationScope;
|
||
List<SecurityReference> securityReferences = new ArrayList<>();
|
||
securityReferences.add(new SecurityReference("Authorization", authorizationScopes));
|
||
return securityReferences;
|
||
}
|
||
}
|