Spring boot集成Swagger,并配置多个扫描路径

1:认识Swagger

Swagger 是一个规范和完整的框架,用于生成、描述、调用和可视化 RESTful 风格的 Web 服务。总体目标是使客户端和文件系统作为服务器以同样的速度来更新。文件的方法,参数和模型紧密集成到服务器端的代码,允许API来始终保持同步。

作用:

1. 接口的文档在线自动生成。

2. 功能测试。

Swagger是一组开源项目,其中主要要项目如下:

  1. Swagger-tools:提供各种与Swagger进行集成和交互的工具。例如模式检验、Swagger 1.2文档转换成Swagger 2.0文档等功能。
  2. Swagger-core: 用于Java/Scala的的Swagger实现。与JAX-RS(Jersey、Resteasy、CXF...)、Servlets和Play框架进行集成。
  3. Swagger-js: 用于JavaScript的Swagger实现。
  4. Swagger-node-express: Swagger模块,用于node.js的Express web应用框架。
  5. Swagger-ui:一个无依赖的HTML、JS和CSS集合,可以为Swagger兼容API动态生成优雅文档。
  6. Swagger-codegen:一个模板驱动引擎,通过分析用户Swagger资源声明以各种语言生成客户端代码。

2:spring boot 集成 swagger

2.1 引入POM

<!-- swagger2 -->
<dependency>
 <groupId>io.springfox</groupId>
 <artifactId>springfox-swagger2</artifactId>
 <version>2.8.0</version>
</dependency>
<dependency>
 <groupId>io.springfox</groupId>
 <artifactId>springfox-swagger-ui</artifactId>
 <version>2.8.0</version>
</dependency>

2.2 配置swagger

在Application.java同级或子包中创建SwaggerConfig.java

package com.example.demo.swagger;
import org.springframework.context.annotation.Bean;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
@Configuration
@EnableSwagger2
public class SwaggerConfig {
 /**
 * 创建API应用
 * apiInfo() 增加API相关信息
 * 通过select()函数返回一个ApiSelectorBuilder实例,用来控制哪些接口暴露给Swagger来展现,
 * 本例采用指定扫描的包路径来定义指定要建立API的目录。
 *
 * @return
 */
 @Bean
 public Docket createRestApi() {
 return new Docket(DocumentationType.SWAGGER_2)
 .apiInfo(apiInfo())
 .select()
 .apis(RequestHandlerSelectors.basePackage("com.example.demo.controller"))
 .paths(PathSelectors.any())
 .build();
 }
 /**
 * 创建该API的基本信息(这些基本信息会展现在文档页面中)
 * 访问地址:http://项目实际地址/swagger-ui.html
 *
 * @return
 */
 private ApiInfo apiInfo() {
 return new ApiInfoBuilder()
 .title("测试 APIs")
 .description("测试api接口文档")
 .termsOfServiceUrl("http://www.baidu.com")
 .version("1.0")
 .build();
 }
}
2.3 添加文档内容

在完成了上述配置后,其实已经可以生产文档内容

如图

Spring boot集成Swagger,并配置多个扫描路径

但是这样的文档主要针对请求本身,描述的主要来源是函数的命名,对用户并不友好,我们通常需要自己增加一些说明来丰富文档内容。

Swagger使用的注解及其说明:

@Api:用在类上,说明该类的作用。

@ApiOperation:注解来给API增加方法说明。

@ApiImplicitParams : 用在方法上包含一组参数说明。

@ApiImplicitParam:用来注解来给方法入参增加说明。

@ApiResponses:用于表示一组响应

@ApiResponse:用在@ApiResponses中,一般用于表达一个错误的响应信息

​ l code:数字,例如400

​ l message:信息,例如"请求参数没填好"

​ l response:抛出异常的类

@ApiModel:描述一个Model的信息(一般用在请求参数无法使用@ApiImplicitParam注解进行描述的时候)

​ l @ApiModelProperty:描述一个model的属性

例如:

package com.example.demo.controller;
import com.example.demo.entity.User;
import com.example.demo.service.UserService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/user")
@Api(tags="用户系统-用户管理")
public class UserController {
 @Autowired
 private UserService userService;
 @GetMapping("updateMoblie")
 @ApiOperation(value="更新手机号", notes = "更新手机号接口")
 public String updateMoblie(Long userId){
 userService.updateMobile(userId);
 return "success";
 }
}

结果如图

Spring boot集成Swagger,并配置多个扫描路径

以上我们完成了spring boot与swagger的集成,但是使用springfox中的 RequestHandlerSelectors.basePackage("com.xxx") 只能支持单个包路径的扫描匹配,如果我们业务中分了多个包,swagger怎么能扫描到呢?

3: swagger 多包扫描配置

要解决这个问题,我们可以参考一下RequestHandlerSelectors.basePackage的源码

public class RequestHandlerSelectors {
 private RequestHandlerSelectors() {
 throw new UnsupportedOperationException();
 }
 /**
 * Any RequestHandler satisfies this condition
 *
 * @return predicate that is always true
 */
 public static Predicate<RequestHandler> any() {
 return Predicates.alwaysTrue();
 }
 /**
 * No RequestHandler satisfies this condition
 *
 * @return predicate that is always false
 */
 public static Predicate<RequestHandler> none() {
 return Predicates.alwaysFalse();
 }
 /**
 * Predicate that matches RequestHandler with handlers methods annotated with given annotation
 *
 * @param annotation - annotation to check
 * @return this
 */
 public static Predicate<RequestHandler> withMethodAnnotation(final Class<? extends Annotation> annotation) {
 return new Predicate<RequestHandler>() {
 @Override
 public boolean apply(RequestHandler input) {
 return input.isAnnotatedWith(annotation);
 }
 };
 }
 /**
 * Predicate that matches RequestHandler with given annotation on the declaring class of the handler method
 *
 * @param annotation - annotation to check
 * @return this
 */
 public static Predicate<RequestHandler> withClassAnnotation(final Class<? extends Annotation> annotation) {
 return new Predicate<RequestHandler>() {
 @Override
 public boolean apply(RequestHandler input) {
 return declaringClass(input).transform(annotationPresent(annotation)).or(false);
 }
 };
 }
 private static Function<Class<?>, Boolean> annotationPresent(final Class<? extends Annotation> annotation) {
 return new Function<Class<?>, Boolean>() {
 @Override
 public Boolean apply(Class<?> input) {
 return input.isAnnotationPresent(annotation);
 }
 };
 }
 private static Function<Class<?>, Boolean> handlerPackage(final String basePackage) {
 return new Function<Class<?>, Boolean>() {
 @Override
 public Boolean apply(Class<?> input) {
 return ClassUtils.getPackageName(input).startsWith(basePackage);
 }
 };
 }
 /**
 * Predicate 匹配RequestHandler,并为处理程序方法的类提供基本包名.
 * predicate 包括与所提供的basePackage匹配的所有请求处理程序
 *
 * @param basePackage - base package of the classes
 * @return this
 */
 public static Predicate<RequestHandler> basePackage(final String basePackage) {
 return new Predicate<RequestHandler>() {
 @Override
 public boolean apply(RequestHandler input) {
 return declaringClass(input).transform(handlerPackage(basePackage)).or(true);
 }
 };
 }
 private static Optional<? extends Class<?>> declaringClass(RequestHandler input) {
 return Optional.fromNullable(input.declaringClass());
 }
}

我们看到 swagger 是通过Predicate 的apply 方法的返回值来判断是非匹配的 我们可以通过改造basePackage方法来实现多包扫描,改造 SwaggerConfig 如下

package com.example.demo.swagger;
import com.google.common.base.Function;
import com.google.common.base.Optional;
import com.google.common.base.Predicate;
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.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;
@Configuration
@EnableSwagger2
public class SwaggerConfig {
 // 定义分隔符
 private static final String splitor = ";";
 /**
 * 创建API应用
 * api() 增加API相关信息
 * 通过select()函数返回一个ApiSelectorBuilder实例,用来控制哪些接口暴露给Swagger来展现,
 * 本例采用指定扫描的包路径来定义指定要建立API的目录。
 *
 * @return
 */
 @Bean
 public Docket api() {
 return new Docket(DocumentationType.SWAGGER_2)
 .apiInfo(apiInfo())
 .select()
 .apis(basePackage("com.example.demo.controller"+splitor+"com.example.demo.test"))
 .paths(PathSelectors.any())
 .build();
 }
 /**
 * 构建 api文档的详细信息函数
 *
 * @return
 */
 private ApiInfo apiInfo() {
 return new ApiInfoBuilder()
 .title("测试 APIs")
 .description("测试api接口文档")
 .version("1.0")
 .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());
 }
 /**
 * http://localhost:8080/swagger-ui.html
 */
}

通过以上配置 我们就可以扫描到 com.example.demo.controller 和 com.example.demo.test 两个包下的接口信息了。

欢迎工作一到五年的Java工程师朋友们加入Java程序员开发: 854393687

群内提供免费的Java架构学习资料(里面有高可用、高并发、高性能及分布式、Jvm性能调优、Spring源码,MyBatis,Netty,Redis,Kafka,Mysql,Zookeeper,Tomcat,Docker,Dubbo,Nginx等多个知识点的架构资料)合理利用自己每一分每一秒的时间来学习提升自己,不要再用"没有时间“来掩饰自己思想上的懒惰!趁年轻,使劲拼,给未来的自己一个交代!

相关推荐