SpringBoot整合OpenApi的實踐

更新時間：2021年08月26日 16:20:33 作者：范閑

本文主要介紹了SpringBoot整合OpenApi，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下

SpringBoot整合OpenApi

確保SpringBoot版本在2.x級以上。

OpenAPI依賴

引入OpenApi依賴

<dependency>
    <groupId>io.springfox</groupId>
    <artifactId>springfox-boot-starter</artifactId>
    <version>3.0.0</version>
</dependency>

編寫配置類

編寫配置類，并手動裝配@EnableOpenApi

@EnableOpenApi
@Configuration
public class OpenApiConfiguration {

    @Bean
    public Docket createRestApi() {
        return new Docket(DocumentationType.OAS_30)
                .apiInfo(apiInfo())
                .select()
                .apis(apis())
                .paths(PathSelectors.any())
                .build()
                .enable(true);
    }

    private ApiInfo apiInfo() {
        return new ApiInfoBuilder()
                .title("項目名稱")
                .description("項目描述")
                .termsOfServiceUrl("項目地址")
                .version("API版本")
                .license("公司的license")
                .licenseUrl("公司的license地址")
                .build();
    }

    private Predicate<RequestHandler> apis() {
        return RequestHandlerSelectors.basePackage("controller包的路徑");
    }
}

到這里SpringBoot就整合OpenAPI成功了。需要一提的是，OpenAPI不僅可以通過掃描controller包的路徑生成OpenAPI，也可以通過掃描@ApiOperation來生成OpenAPI。

改造優(yōu)化

上面的示例通過硬編碼的形式，所以無法進(jìn)行代碼的復(fù)用，而且每次更新配置的時候，都需要更改代碼，比如我們的項目在開發(fā)或者測試環(huán)境時，我們希望能正常使用OpenAPI但是在生產(chǎn)環(huán)境需要禁用OpenAPI。
在工程學(xué)的角度，常用的做法是盡量通過更改配置的形式，問不是更改代碼?；诖?，我們可以借助SpringBoot的配置功能，對上述代碼進(jìn)行改造。

新增OpenApi配置類

@Data
@Component
@ConfigurationProperties(prefix = "example.web.swagger")
public class SwaggerProperties {
    /**
     * 是否swagger3啟用，默認(rèn)不啟用
     */
    private Boolean enable = false;
    /**
     * 掃描包路徑，可以不指定，系統(tǒng)會通過自動掃描{@link io.swagger.annotations.ApiOperation}
     */
    private String basePackage;
    /**
     * 標(biāo)題
     */
    private String title;
    /**
     * 應(yīng)用描述
     */
    private String description;
    /**
     * 服務(wù)地址
     */
    private String serviceUrl;
    /**
     * 版本，默認(rèn)V1.0.0
     */
    private String version = "V1.0.0";
    /**
     * license
     */
    private String license;
    /**
     * licenseUrl
     */
    private String licenseUrl;
}

配置替換硬編碼

@Slf4j
@Configuration
@EnableOpenApi
public class OpenApiConfiguration {

    @Resource
    private SwaggerProperties swaggerProperties;

    @Bean
    public Docket createRestApi() {
        return new Docket(DocumentationType.OAS_30)
                .apiInfo(apiInfo())
                .select()
                .apis(apis())
                .paths(PathSelectors.any())
                .build()
                .enable(isEnable());
    }

    private Boolean isEnable() {
        Boolean enable = swaggerProperties.getEnable();
        if (enable) {
            log.info("\nEnable Swagger3...");
        }
        return enable;
    }

    private ApiInfo apiInfo() {
        return new ApiInfoBuilder()
                .title(swaggerProperties.getTitle())
                .description(swaggerProperties.getDescription())
                .termsOfServiceUrl(swaggerProperties.getServiceUrl())
                .version(swaggerProperties.getVersion())
                .license(swaggerProperties.getLicense())
                .licenseUrl(swaggerProperties.getLicenseUrl())
                .build();
    }

    private Predicate<RequestHandler> apis() {
        String basePackage = swaggerProperties.getBasePackage();
        // 默認(rèn)通過掃描`ApiOperation`如果配置了包掃描路徑，使用配置的包掃描路徑
        return Strings.isNullOrEmpty(basePackage) ? withMethodAnnotation(ApiOperation.class) : basePackage(basePackage);
    }
}

通過這樣的簡單改造后，我們就可以完全通過配置文件的形式配置OpenApi

示例配置

example:
  web:
    swagger:
      enable: true
      title: 演示OpenAPI
      description: 演示OpenAPI
      base-package: com.example.controller

OpenAPI常用注解介紹

這里通過使用代碼演示注解的使用

實體類

需要在用戶的VO實體類標(biāo)注@ApiModel并說明該類的作用，屬性上通過@ApiModelProperty解析屬性的作用，并通過required值約定該屬性是否可以為空可以通過example說明該屬性的用例值

@Data
@ApiModel("用戶信息")
public class UserVO {

    @ApiModelProperty(value = "用戶id", required = true, example = "asdf124")
    private String userId;

    @ApiModelProperty(value = "用戶名稱", required = true)
    private String username;

    @ApiModelProperty(value = "用戶年齡")
    private Integer age;
}

controller類

這里主要通過用戶注冊和用戶信息接口演示。

在配上通過@Api標(biāo)注該類的功能；通過@ApiOperation說明接口的功能，如果傳入的數(shù)據(jù)不是實體類，而是一個基本數(shù)據(jù)類型，可以通過@ApiParam解釋參數(shù)的作用

@RequestMapping("user")
@RestController
@Api(tags = "用戶中心")
public class UserController {

    @ApiOperation("用戶注冊")
    @PostMapping
    public R<Void> register(@RequestBody UserVO userVO) {
        // todo 
        return R.ok();
    }

    @ApiOperation("通過Id獲取用戶信息")
    @GetMapping
    public R<UserVO> userInfo(@ApiParam(value = "用戶id",required = true) @RequestParam String userId) {
        // todo
        return R.ok(new UserVO());
    }
}