添加鉴权

src/main/java/com/muyu/common/api/doc/config/SpringDocConfig.java

@@ -1,12 +1,20 @@
 package com.muyu.common.api.doc.config;
 
+import io.swagger.v3.oas.models.Components;
 import io.swagger.v3.oas.models.ExternalDocumentation;
 import io.swagger.v3.oas.models.OpenAPI;
 import io.swagger.v3.oas.models.info.Info;
+import io.swagger.v3.oas.models.security.SecurityRequirement;
+import io.swagger.v3.oas.models.security.SecurityScheme;
+import org.springdoc.core.customizers.GlobalOpenApiCustomizer;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.http.HttpHeaders;
 
+/**
+ * @author Lenovo
+ */
 @Configuration
 public class SpringDocConfig {
 
@@ -24,11 +32,41 @@ public class SpringDocConfig {
                 .description("服务总站")
                 .url("https://gitea.qinmian.online");
     }
-
     @Bean
     public OpenAPI springShopOpenAPI() {
         return new OpenAPI()
                 .info(info())
-                .externalDocs(externalDocumentation());
+                .externalDocs(externalDocumentation())
+                //配置全局鉴权参数-Authorize
+                .components(new Components()
+                        .addSecuritySchemes(HttpHeaders.AUTHORIZATION,
+                                new SecurityScheme()
+                                        .name(HttpHeaders.AUTHORIZATION)
+                                        .type(SecurityScheme.Type.APIKEY)
+                                        .in(SecurityScheme.In.HEADER)
+                                        .scheme("Bearer")
+                                        .bearerFormat("JWT")
+                        )
+                );
+    }
+
+    @Bean
+    public GlobalOpenApiCustomizer globalOpenApiCustomizer(){
+        return openApi -> {
+            //全局添加鉴权参数
+            if(openApi.getPaths() != null){
+                openApi.getPaths().forEach((s,pathItem) -> {
+                    //登录接口/验证码不需要添加鉴权参数
+                    if(s.equals("/login") || s.equals("/code") || s.equals("/register")){
+                        return;
+                    }
+                    //接口添加鉴权参数
+                    pathItem.readOperations()
+                            .forEach(operation ->
+                                    operation.addSecurityItem(new SecurityRequirement().addList(HttpHeaders.AUTHORIZATION))
+                            );
+                });
+            }
+        };
     }
 }