com.github.xiaoymin.swaggerbootstrapui.util.CommonUtils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of swagger-bootstrap-ui Show documentation
Show all versions of swagger-bootstrap-ui Show documentation
Swagger-Bootstrap-UI is the front of the UI Swagger implementation, using jQuery+bootstrap implementation, the purpose is to replace the default UI Swagger implementation of the Swagger-UI, so that the document is more friendly...
The newest version!
/*
* 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.util;
import com.github.xiaoymin.swaggerbootstrapui.model.DynamicClass;
import io.swagger.annotations.ApiModelProperty;
import io.swagger.annotations.DynamicParameter;
import io.swagger.annotations.DynamicResponseParameters;
import javassist.*;
import javassist.bytecode.AnnotationsAttribute;
import javassist.bytecode.ConstPool;
import javassist.bytecode.annotation.Annotation;
import javassist.bytecode.annotation.BooleanMemberValue;
import javassist.bytecode.annotation.IntegerMemberValue;
import javassist.bytecode.annotation.StringMemberValue;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.*;
import java.lang.reflect.Field;
/***
*
* @since:swagger-bootstrap-ui 1.8.5
* @author [email protected]
* 2018/10/11 13:47
*/
public class CommonUtils {
static Logger logger= LoggerFactory.getLogger(CommonUtils.class);
public static final String basePackage="com.github.xiaoymin.swaggerbootstrapui.model.";
static final ClassPool classPool=ClassPool.getDefault();
public static String UpperCase(String str){
StringBuffer aa=new StringBuffer();
int index = 0;
int index22 = 0;
int len = str.length();
begin:
for (int i = 1; i < len; i++) {
if (Character.isUpperCase(str.charAt(i))) {
//判断下一个是大写还是小写
if(Character.isUpperCase(str.charAt(i+1))){
aa.append(str.substring(index, i)).append("");
}else {
aa.append(str.substring(index, i)).append(" ");
}
index = i;
index22=index22+1;
continue begin;
}
index22=0;
}
aa.append(str.substring(index, len));
return aa.toString();
}
public static byte[] readBytes(File file) {
long len = file.length();
if (len >= Integer.MAX_VALUE) {
throw new RuntimeException("File is larger then max array size");
}
byte[] bytes = new byte[(int) len];
FileInputStream in = null;
int readLength;
try {
in = new FileInputStream(file);
readLength = in.read(bytes);
if(readLength < len){
throw new IOException("File length is ["+len+"] but read ["+readLength+"]!");
}
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
closeQuiltly(in);
}
return bytes;
}
public static byte[] readBytes(InputStream ins){
if (ins==null){
return null;
}
ByteArrayOutputStream byteOutArr=new ByteArrayOutputStream();
int r=-1;
byte[] bytes = new byte[1024*1024];
try {
while ((r=ins.read(bytes))!=-1){
byteOutArr.write(bytes,0,r);
}
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
closeQuiltly(ins);
}
return byteOutArr.toByteArray();
}
public static void closeQuiltly(InputStream ins){
if (ins!=null){
try {
ins.close();
} catch (IOException e) {
logger.error(e.getMessage(),e);
}
}
}
public static void closeQuiltly(Reader reader){
if (reader!=null){
try {
reader.close();
} catch (IOException e) {
logger.error(e.getMessage(),e);
}
}
}
/**
* createModel
* @param name 类名
* @param parameters 属性集合
* @return 动态生成类
*/
public static Class createDynamicModelClass(String name,DynamicParameter[] parameters){
String clazzName=basePackage+name;
try {
CtClass tmp=classPool.getCtClass(clazzName);
if (tmp!=null){
tmp.detach();
}
} catch (NotFoundException e) {
}
CtClass ctClass=classPool.makeClass(clazzName);
try{
int fieldCount=0;
for (DynamicParameter dynamicParameter:parameters){
//field名称不能为空,导致非空异常
// https://gitee.com/xiaoym/swagger-bootstrap-ui/issues/IYLVC
if (dynamicParameter.name()!=null&&!"".equals(dynamicParameter.name())&&!"null".equals(dynamicParameter.name())){
ctClass.addField(createField(dynamicParameter,ctClass));
fieldCount++;
}
}
if (fieldCount>0){
return ctClass.toClass();
}
}catch (Throwable e){
logger.error(e.getMessage());
}
return null;
}
/**
* 根据原始泛型类创建一个新的Class
* @param originalGenericType 原始泛型类
* @param name 新类的名称
* @param dynamicResponseParameters 动态字段
* @return 动态生成类
*/
public static DynamicClass createDynamicModelByOriginalGenericClass(Class originalGenericType, String name, DynamicResponseParameters dynamicResponseParameters){
DynamicClass dynamicClass=new DynamicClass();
try{
if (originalGenericType!=null){
String clazzName=basePackage+name;
try {
CtClass tmp=classPool.getCtClass(clazzName);
if (tmp!=null){
tmp.detach();
}
} catch (NotFoundException e) {
}
CtClass ctClass=classPool.makeClass(clazzName);
Field[] originalFields=originalGenericType.getDeclaredFields();
//String genericFieldName=dynamicResponseParameters.genericFieldName();
String genericFieldName="";
String fieldClassName=name+genericFieldName;
for (Field field:originalFields){
if (field.getName().equals(genericFieldName)){
//ctClass.addField(createByDynamicField(field,fieldClassName,dynamicResponseParameters,ctClass,dynamicClass));
}else{
ctClass.addField(createByOriginalField(field,ctClass));
}
}
dynamicClass.setTargetClazz(ctClass.toClass());
//return ctClass.toClass();
}
}catch (Throwable e){
logger.error(e.getMessage(),e);
}
return dynamicClass;
}
private static CtField createByDynamicField(Field field,String fieldClassName,DynamicResponseParameters dynamicResponseParameters,CtClass ctClass,DynamicClass dynamicClass) throws CannotCompileException {
Class fieldClazz=createDynamicModelClass(fieldClassName,dynamicResponseParameters.properties());
dynamicClass.setFieldClazz(fieldClazz);
CtField ctField=new CtField(getFieldType(fieldClazz),field.getName(),ctClass);
ctField.setModifiers(Modifier.PUBLIC);
addAnnotation(ctField,field,ctClass);
return ctField;
}
private static CtField createByOriginalField(Field field,CtClass ctClass) throws CannotCompileException {
CtField ctField=new CtField(getFieldType(field.getType()),field.getName(),ctClass);
ctField.setModifiers(Modifier.PUBLIC);
//此处需要判断原field是否包含注解
addAnnotation(ctField,field,ctClass);
return ctField;
}
private static void addAnnotation(CtField target,Field source,CtClass ctClass){
//此处需要判断原field是否包含注解
if (source.isAnnotationPresent(ApiModelProperty.class)){
//如果包含
ApiModelProperty apiModelProperty=source.getAnnotation(ApiModelProperty.class);
ConstPool constPool=ctClass.getClassFile().getConstPool();
AnnotationsAttribute attr = new AnnotationsAttribute(constPool, AnnotationsAttribute.visibleTag);
Annotation ann = new Annotation("io.swagger.annotations.ApiModelProperty", constPool);
ann.addMemberValue("value", new StringMemberValue(apiModelProperty.value(), constPool));
ann.addMemberValue("example", new StringMemberValue(apiModelProperty.example(), constPool));
ann.addMemberValue("name", new StringMemberValue(apiModelProperty.name(), constPool));
ann.addMemberValue("dataType", new StringMemberValue(apiModelProperty.dataType(), constPool));
ann.addMemberValue("access", new StringMemberValue(apiModelProperty.access(), constPool));
ann.addMemberValue("allowableValues", new StringMemberValue(apiModelProperty.allowableValues(), constPool));
ann.addMemberValue("notes", new StringMemberValue(apiModelProperty.notes(), constPool));
//ann.addMemberValue("position", new IntegerMemberValue(apiModelProperty.position(), constPool));
ann.addMemberValue("reference", new StringMemberValue(apiModelProperty.reference(), constPool));
ann.addMemberValue("value", new BooleanMemberValue(apiModelProperty.readOnly(), constPool));
ann.addMemberValue("readOnly", new BooleanMemberValue(apiModelProperty.hidden(), constPool));
ann.addMemberValue("required", new BooleanMemberValue(apiModelProperty.required(),constPool));
//判断example是否不为空
attr.addAnnotation(ann);
target.getFieldInfo().addAttribute(attr);
}
}
public static Class load(String classPathName){
try{
return Class.forName(classPathName);
}catch (Exception e){
}
return null;
}
private static CtField createField(DynamicParameter parameter, CtClass ctClass) throws NotFoundException, CannotCompileException {
CtField field=new CtField(getFieldType(parameter.dataTypeClass()),parameter.name(),ctClass);
field.setModifiers(Modifier.PUBLIC);
ConstPool constPool=ctClass.getClassFile().getConstPool();
AnnotationsAttribute attr = new AnnotationsAttribute(constPool, AnnotationsAttribute.visibleTag);
Annotation ann = new Annotation("io.swagger.annotations.ApiModelProperty", constPool);
ann.addMemberValue("value", new StringMemberValue(parameter.value(), constPool));
ann.addMemberValue("required", new BooleanMemberValue(parameter.required(),constPool));
//判断example是否不为空
if (parameter.example()!=null&&!"".equals(parameter.example())){
ann.addMemberValue("example", new StringMemberValue(parameter.example(), constPool));
}
attr.addAnnotation(ann);
field.getFieldInfo().addAttribute(attr);
return field;
}
public static CtClass getFieldType(Class propetyType) {
CtClass fieldType= null;
try{
if (!propetyType.isAssignableFrom(Void.class)){
fieldType=classPool.get(propetyType.getName());
}else{
fieldType=classPool.get(String.class.getName());
}
}catch (NotFoundException e){
//抛异常
ClassClassPath path=new ClassClassPath(propetyType);
classPool.insertClassPath(path);
try {
fieldType=classPool.get(propetyType.getName());
} catch (NotFoundException e1) {
logger.error(e1.getMessage(),e1);
//can't find
}
}
return fieldType;
}
}