优化代码

RuoYi

2023-03-29 0b21d15f26010d4736b431bd359900d3ca23da21

 ruoyi-common/ruoyi-common-swagger/src/main/java/com/ruoyi/common/swagger/config/SwaggerAutoConfiguration.java

@@ -3,13 +3,12 @@
import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import org.springframework.boot.autoconfigure.EnableAutoConfiguration;

import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;

import java.util.function.Predicate;

import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;

import org.springframework.boot.context.properties.EnableConfigurationProperties;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import com.google.common.base.Predicate;

import com.google.common.base.Predicates;

import org.springframework.context.annotation.Import;

import springfox.documentation.builders.ApiInfoBuilder;

import springfox.documentation.builders.PathSelectors;

import springfox.documentation.builders.RequestHandlerSelectors;

@@ -18,15 +17,18 @@
import springfox.documentation.service.AuthorizationScope;

import springfox.documentation.service.Contact;

import springfox.documentation.service.SecurityReference;

import springfox.documentation.service.SecurityScheme;

import springfox.documentation.spi.DocumentationType;

import springfox.documentation.spi.service.contexts.SecurityContext;

import springfox.documentation.spring.web.plugins.ApiSelectorBuilder;

import springfox.documentation.spring.web.plugins.Docket;

import springfox.documentation.swagger2.annotations.EnableSwagger2;



@Configuration

@EnableSwagger2

@EnableAutoConfiguration

@EnableConfigurationProperties(SwaggerProperties.class)

@ConditionalOnProperty(name = "swagger.enabled", matchIfMissing = true)

@Import({SwaggerBeanPostProcessor.class, SwaggerWebConfiguration.class})

public class SwaggerAutoConfiguration

{

    /**

@@ -35,13 +37,6 @@
    private static final List<String> DEFAULT_EXCLUDE_PATH = Arrays.asList("/error", "/actuator/**");



    private static final String BASE_PATH = "/**";



    @Bean

    @ConditionalOnMissingBean

    public SwaggerProperties swaggerProperties()

    {

        return new SwaggerProperties();

    }



    @Bean

    public Docket api(SwaggerProperties swaggerProperties)

@@ -60,27 +55,26 @@
        {

            swaggerProperties.getExcludePath().addAll(DEFAULT_EXCLUDE_PATH);

        }



        List<Predicate<String>> excludePath = new ArrayList<>();

        swaggerProperties.getExcludePath().forEach(path -> excludePath.add(PathSelectors.ant(path)));



        //noinspection Guava

        return new Docket(DocumentationType.SWAGGER_2)

                .host(swaggerProperties.getHost())

        ApiSelectorBuilder builder = new Docket(DocumentationType.SWAGGER_2).host(swaggerProperties.getHost())

                .apiInfo(apiInfo(swaggerProperties)).select()

                .apis(RequestHandlerSelectors.basePackage(swaggerProperties.getBasePackage()))

                .paths(Predicates.and(Predicates.not(Predicates.or(excludePath)), Predicates.or(basePath)))

                .build()

                .securitySchemes(securitySchemes())

                .securityContexts(securityContexts())

                .pathMapping("/");

                .apis(RequestHandlerSelectors.basePackage(swaggerProperties.getBasePackage()));



        swaggerProperties.getBasePath().forEach(p -> builder.paths(PathSelectors.ant(p)));

        swaggerProperties.getExcludePath().forEach(p -> builder.paths(PathSelectors.ant(p).negate()));



        return builder.build().securitySchemes(securitySchemes()).securityContexts(securityContexts()).pathMapping("/");

    }



    /**

     * 安全模式，这里指定token通过Authorization头请求头传递

     */

    private List<ApiKey> securitySchemes()

    private List<SecurityScheme> securitySchemes()

    {

        List<ApiKey> apiKeyList = new ArrayList<ApiKey>();

        List<SecurityScheme> apiKeyList = new ArrayList<SecurityScheme>();

        apiKeyList.add(new ApiKey("Authorization", "Authorization", "header"));

        return apiKeyList;

    }

@@ -94,7 +88,7 @@
        securityContexts.add(

                SecurityContext.builder()

                        .securityReferences(defaultAuth())

                        .forPaths(PathSelectors.regex("^(?!auth).*$"))

                        .operationSelector(o -> o.requestMappingPattern().matches("/.*"))

                        .build());

        return securityContexts;

    }