package com.qianwen.smartman.common.launch;
|
|
import cn.hutool.core.bean.BeanUtil;
|
import cn.hutool.core.util.StrUtil;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
import org.springframework.core.annotation.Order;
|
import org.springframework.stereotype.Component;
|
import org.springframework.web.method.HandlerMethod;
|
import springfox.documentation.spi.DocumentationType;
|
import springfox.documentation.spi.service.OperationBuilderPlugin;
|
import springfox.documentation.spi.service.contexts.OperationContext;
|
import springfox.documentation.spi.service.contexts.RequestMappingContext;
|
import springfox.documentation.spring.web.WebMvcRequestHandler;
|
|
@Component
|
@Order(-2147481647)
|
@ConditionalOnProperty(name ="enable" ,prefix = "swagger",havingValue = "true",matchIfMissing = true)
|
public class SwaggerPlugin implements OperationBuilderPlugin {
|
private static final Logger log = LoggerFactory.getLogger(SwaggerPlugin.class);
|
|
@Override
|
public void apply(OperationContext operationContext) {
|
RequestMappingContext requestContext = (RequestMappingContext) BeanUtil.getProperty(operationContext, "requestContext");
|
WebMvcRequestHandler handler = (WebMvcRequestHandler) BeanUtil.getProperty(requestContext, "handler");
|
HandlerMethod handlerMethod = (HandlerMethod) BeanUtil.getProperty(handler, "handlerMethod");
|
String controllerName = handlerMethod.getBeanType().getSimpleName();
|
String operationNameStem = operationContext.getName();
|
String prefix = StrUtil.replace(controllerName, "Controller", "");
|
String newName = StrUtil.format("{}_{}", new Object[]{prefix, operationNameStem});
|
log.info("operationId:{}", newName);
|
operationContext.operationBuilder().codegenMethodNameStem(newName);
|
}
|
|
public boolean supports(DocumentationType documentationType) {
|
return true;
|
}
|
}
|
