SwaggerConfig.java 1.59 KB
package com.subsidy.common.configure;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

@Configuration
@EnableSwagger2
public class SwaggerConfig {

    @Value("${spring.profiles.active:}")
    private String activeProfile;


    @Bean
    public Docket createRestApi() {

        // 判断是否为生产环境
        boolean isProd = "prod".equals(activeProfile);

        return new Docket(DocumentationType.SWAGGER_2)
                .enable(!isProd)
                .pathMapping("/")
                .select()
                .apis(RequestHandlerSelectors.basePackage("com.subsidy"))
                .paths(PathSelectors.any())
                .build().apiInfo(new ApiInfoBuilder()
                        .title("ecnu")
                        .description("接口文档规范")
                        .version("1.0")
//                        .contact(new Contact("啊啊啊啊","www.youkehulian.com","tcp@meishu.com"))
//                        .license("The Apache License")
//                        .licenseUrl("http://www.baidu.com")
                        .build());
    }
}