refactor: knife4j 全局设置参数排除无需鉴权的接口

This commit is contained in:
haoxr
2023-12-15 23:43:09 +08:00
parent 974f32b268
commit 31a2fbeb42

View File

@@ -5,9 +5,12 @@ import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.oas.models.info.Info; import io.swagger.v3.oas.models.info.Info;
import io.swagger.v3.oas.models.security.SecurityRequirement; import io.swagger.v3.oas.models.security.SecurityRequirement;
import io.swagger.v3.oas.models.security.SecurityScheme; import io.swagger.v3.oas.models.security.SecurityScheme;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springdoc.core.customizers.GlobalOpenApiCustomizer; import org.springdoc.core.customizers.GlobalOpenApiCustomizer;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.http.HttpHeaders; import org.springframework.http.HttpHeaders;
/** /**
@@ -19,19 +22,26 @@ import org.springframework.http.HttpHeaders;
* @since 2023/2/17 * @since 2023/2/17
*/ */
@Configuration @Configuration
@Slf4j
@RequiredArgsConstructor
public class SwaggerConfig { public class SwaggerConfig {
private final Environment environment;
/** /**
* 接口信息 * 接口信息
*/ */
@Bean @Bean
public OpenAPI openApi() { public OpenAPI openApi() {
String appVersion = environment.getProperty("project.version", "1.0.0");
return new OpenAPI() return new OpenAPI()
.info(new Info() .info(new Info()
.title("系统接口文档") .title("系统接口文档")
.version("2.4.0") .version(appVersion)
) )
// 全局安全校验项也可以在对应的controller上加注解SecurityRequirement // 配置全局鉴权参数-Authorize
.components(new Components() .components(new Components()
.addSecuritySchemes(HttpHeaders.AUTHORIZATION, .addSecuritySchemes(HttpHeaders.AUTHORIZATION,
new SecurityScheme() new SecurityScheme()
@@ -41,17 +51,34 @@ public class SwaggerConfig {
.scheme("Bearer") .scheme("Bearer")
.bearerFormat("JWT") .bearerFormat("JWT")
) )
) );
.addSecurityItem(new SecurityRequirement().addList(HttpHeaders.AUTHORIZATION)) ;
} }
/**
* 全局自定义扩展
* <p>
* 在OpenAPI规范中Operation 是一个表示 API 端点Endpoint或操作的对象。
* 每个路径Path对象可以包含一个或多个 Operation 对象,用于描述与该路径相关联的不同 HTTP 方法(例如 GET、POST、PUT 等)。
*/
@Bean @Bean
public GlobalOpenApiCustomizer globalOpenApiCustomizer() { public GlobalOpenApiCustomizer globalOpenApiCustomizer() {
return openApi -> openApi.getPaths().values() return openApi -> {
.stream() // 全局添加鉴权参数
.flatMap(pathItem -> pathItem.readOperations().stream()) if (openApi.getPaths() != null) {
.forEach(operation -> operation.security(openApi.getSecurity())); openApi.getPaths().forEach((s, pathItem) -> {
// 登录接口/验证码不需要添加鉴权参数
if (s.equals("/api/v1/auth/login") || s.equals("/api/v1/auth/captcha")) {
return;
}
// 接口添加鉴权参数
pathItem.readOperations()
.forEach(operation ->
operation.addSecurityItem(new SecurityRequirement().addList(HttpHeaders.AUTHORIZATION))
);
});
}
};
} }
} }