com.github.xiaoymin.swaggerbootstrapui.plugin.DynamicParameterBuilderPlugin Maven / Gradle / Ivy
/*
* Copyright (C) 2018 Zhejiang xiaominfo Technology CO.,LTD.
* All rights reserved.
* Official Web Site: http://www.xiaominfo.com.
* Developer Web Site: http://open.xiaominfo.com.
*/
package com.github.xiaoymin.swaggerbootstrapui.plugin;
import com.google.common.base.Optional;
import io.swagger.annotations.ApiOperationSupport;
import io.swagger.annotations.DynamicParameter;
import io.swagger.annotations.DynamicParameters;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import springfox.documentation.schema.ModelRef;
import springfox.documentation.service.ResolvedMethodParameter;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spi.service.ParameterBuilderPlugin;
import springfox.documentation.spi.service.contexts.ParameterContext;
import java.util.HashMap;
import java.util.Map;
/***
*
* @since:swagger-bootstrap-ui 1.0
* @author [email protected]
* 2019/06/09 15:30
*/
@Component
@Order(Ordered.HIGHEST_PRECEDENCE+10)
public class DynamicParameterBuilderPlugin implements ParameterBuilderPlugin {
private final Map cacheGenModelMaps=new HashMap<>();
@Override
public void apply(ParameterContext parameterContext) {
ResolvedMethodParameter resolvedMethodParameter=parameterContext.resolvedMethodParameter();
if (resolvedMethodParameter!=null&&resolvedMethodParameter.getParameterType()!=null&&resolvedMethodParameter.getParameterType().getErasedType()!=null){
if (Map.class.isAssignableFrom(resolvedMethodParameter.getParameterType().getErasedType()) || resolvedMethodParameter.getParameterType().getErasedType().getName() == "com.google.gson.JsonObject" ){
//查询注解
Optional supportOptional=parameterContext.getOperationContext().findAnnotation(ApiOperationSupport.class);
if (supportOptional.isPresent()){
ApiOperationSupport support=supportOptional.get();
//判断是否包含自定义注解
changeDynamicParameterType(support.params(),parameterContext);
}else{
Optional dynamicParametersOptional=parameterContext.getOperationContext().findAnnotation(DynamicParameters.class);
if (dynamicParametersOptional.isPresent()){
changeDynamicParameterType(dynamicParametersOptional.get(),parameterContext);
}
}
}
}
}
private void changeDynamicParameterType(DynamicParameters dynamicParameters,ParameterContext parameterContext){
if (dynamicParameters!=null){
//name是否包含
String name=dynamicParameters.name();
if (name==null||"".equals(name)){
//gen
name=genClassName(parameterContext);
}
//判断是否存在
if (cacheGenModelMaps.containsKey(name)){
//存在,以方法名称作为ClassName
name=genClassName(parameterContext);
}
name=name.replaceAll("[_-]","");
DynamicParameter[] dynamics=dynamicParameters.properties();
if (dynamics!=null&&dynamics.length>0){
cacheGenModelMaps.put(name,name);
parameterContext.parameterBuilder() //修改Map参数的ModelRef为我们动态生成的class
.parameterType("body")
.modelRef(new ModelRef(name))
.name(name);
}
}
}
public String genClassName(ParameterContext parameterContext){
//gen
String name=parameterContext.getOperationContext().getName();
if (name!=null&&!"".equals(name)){
if (name.length()==1){
name=name.toUpperCase();
}else{
name=name.substring(0,1).toUpperCase()+name.substring(1);
}
}
return name;
}
@Override
public boolean supports(DocumentationType delimiter) {
return true;
}
}