29 lines
962 B
Java
29 lines
962 B
Java
|
package com.greenorange.promotion.config;
|
||
|
|
|
||
|
|
import com.google.common.net.HttpHeaders;
|
||
|
|
import io.swagger.v3.oas.models.Operation;
|
||
|
|
import io.swagger.v3.oas.models.security.SecurityRequirement;
|
||
|
|
import lombok.extern.slf4j.Slf4j;
|
||
|
|
import org.springdoc.core.customizers.GlobalOperationCustomizer;
|
||
|
|
import org.springframework.stereotype.Component;
|
||
|
|
import org.springframework.web.method.HandlerMethod;
|
||
|
|
|
||
|
|
import java.util.List;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @author: cxz
|
||
|
|
* @description 为每个接口添加鉴权
|
||
|
|
*/
|
||
|
|
@Slf4j
|
||
|
|
@Component
|
||
|
|
public class Knife4jOperationCustomizer implements GlobalOperationCustomizer {
|
||
|
|
@Override
|
||
|
|
public Operation customize(Operation operation, HandlerMethod handlerMethod) {
|
||
|
|
List<SecurityRequirement> security = operation.getSecurity();
|
||
|
|
if (security == null) {
|
||
|
|
security = List.of(new SecurityRequirement().addList(HttpHeaders.AUTHORIZATION));
|
||
|
|
operation.setSecurity(security);
|
||
|
|
}
|
||
|
|
return operation;
|
||
|
|
}
|
||
|
|
}
|