com.github.xiaoymin.swaggerbootstrapui.plugin.OperationDynamicResponseModelProvider 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.fasterxml.classmate.ResolvedType;
import com.fasterxml.classmate.TypeResolver;
import com.github.xiaoymin.swaggerbootstrapui.model.DynamicClass;
import com.github.xiaoymin.swaggerbootstrapui.util.CommonUtils;
import com.google.common.base.Optional;
import io.swagger.annotations.ApiOperationSupport;
import io.swagger.annotations.DynamicParameter;
import io.swagger.annotations.DynamicParameters;
import io.swagger.annotations.DynamicResponseParameters;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RequestBody;
import springfox.documentation.schema.ModelRef;
import springfox.documentation.service.ResolvedMethodParameter;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spi.service.OperationModelsProviderPlugin;
import springfox.documentation.spi.service.contexts.RequestMappingContext;
import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/***
* 动态添加响应类
* @since:swagger-bootstrap-ui 1.9.5
* @author [email protected]
* 2019/06/10 13:01
*/
@Component
@Order(Ordered.HIGHEST_PRECEDENCE+12)
public class OperationDynamicResponseModelProvider implements OperationModelsProviderPlugin {
@Autowired
private TypeResolver typeResolver;
private final Map cacheGenModelMaps=new HashMap<>();
@Override
public void apply(RequestMappingContext context) {
Optional supportOptional=context.findAnnotation(ApiOperationSupport.class);
if(supportOptional.isPresent()){
collectDynamicParameter(supportOptional.get().responses(),context);
}else{
Optional dynamicParametersOptional=context.findAnnotation(DynamicResponseParameters.class);
if (dynamicParametersOptional.isPresent()){
collectDynamicParameter(dynamicParametersOptional.get(),context);
}
}
}
@Override
public boolean supports(DocumentationType delimiter) {
return true;
}
private void collectDynamicParameter(DynamicResponseParameters dynamicParameters,RequestMappingContext context){
if (dynamicParameters!=null){
//name是否包含
String name=dynamicParameters.name();
if (name==null||"".equals(name)){
//gen
name=genClassName(context);
}
//判断是否存在
if (cacheGenModelMaps.containsKey(name)){
//存在,以方法名称作为ClassName
name=genClassName(context);
}
DynamicParameter[] dynamics=dynamicParameters.properties();
if (dynamics!=null&&dynamics.length>0){
cacheGenModelMaps.put(name,name);
//追加groupController
name=context.getGroupName().replaceAll("[_-]","")+"."+name+"Response";
/*Class fieldClazz=null;
String genericFieldName=dynamicParameters.genericFieldName();
if (genericFieldName!=null&&!"".equals(genericFieldName)&&!"null".equals(genericFieldName)){
ResolvedType defaultType=context.alternateFor(context.getReturnType());
Class defaultReturnClazz=defaultType.getErasedType();
DynamicClass dynamicClass=CommonUtils.createDynamicModelByOriginalGenericClass(defaultReturnClazz,name,dynamicParameters);
clazz=dynamicClass.getTargetClazz();
fieldClazz=dynamicClass.getFieldClazz();
if (clazz==null){
clazz= CommonUtils.createDynamicModelClass(name,dynamics);
}
}else{
clazz= CommonUtils.createDynamicModelClass(name,dynamics);
}*/
Class clazz= CommonUtils.createDynamicModelClass(name,dynamics);
if (clazz!=null){
ResolvedType modelType=context.alternateFor(typeResolver.resolve(clazz));
context.operationModelsBuilder().addReturn(modelType);
}
//context.operationModelsBuilder().addInputParam(typeResolver.resolve(clazz));
}
}
}
public String genClassName(RequestMappingContext context){
//gen
String name=context.getName();
if (name!=null&&!"".equals(name)){
name=name.replaceAll("[_-]","");
if (name.length()==1){
name=name.toUpperCase();
}else{
name=name.substring(0,1).toUpperCase()+name.substring(1);
}
}
return name;
}
}