com.webapp.utils.thrift.Generic Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of utils-code Show documentation
The newest version!
/**
 * 
 */
package com.webapp.utils.thrift;

import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;

/**
 * 泛型描述类,是一个树形的结构,每个Generic都有可能有多个types
 *
 */
public class Generic extends ThriftType {
	
	/**
	 * 可能是ThriftType,也可能还是泛型
	 */
	private List types = new ArrayList();

	/**
	 * @return the types
	 */
	@SuppressWarnings("unchecked")
	public List getTypes() {
		return (List) types;
	}

	/**
	 * @param types the types to set
	 */
	public void setTypes(List types) {
		this.types = types;
	}
	
	public void addGeneric(ThriftType generic) {
		types.add(generic);
	}
	
	@Override
	public String toString() {
		StringBuilder sb = new StringBuilder("<");
		for (Object type : types) {
			if(type instanceof Generic) {
				sb.append(type.toString());
			}else {
				ThriftType thriftType = (ThriftType) type;
				if(thriftType.isStruct()) {
					sb.append(thriftType.getValue());
				}else {
					sb.append(thriftType.getWarpperClassName());
				}
			}
		}
		sb.append(">");
		return sb.toString();
	}
	
	public String toThriftString() {
		if(types == null || types.isEmpty()) {
			return this.getValue();
		}
		StringBuilder sb = new StringBuilder();
		sb.append(this.getValue());
		sb.append("<");
		for (int i = 0; i < types.size(); i++) {
			Object type = types.get(i);
			ThriftType thriftType = (ThriftType) type;
			
			if(type instanceof Generic) {
				sb.append(((Generic) type).toThriftString());
			}else {
				sb.append(thriftType.getValue());
			}
			
			if(i != types.size() - 1) {
				sb.append(", ");
			}
		}
		sb.append(">");
		return sb.toString();
	}
	
	public static Generic fromType(Type type) {
		Generic generic = new Generic();
		if(!(type instanceof ParameterizedType)) {
			ThriftType thriftType = ThriftType.fromJavaType(type);
			generic.setJavaClass(thriftType.getJavaClass());
			generic.setJavaTypeName(thriftType.getJavaTypeName());
			generic.settType(thriftType.gettType());
			generic.setValue(thriftType.getValue());
			generic.setWarpperClassName(thriftType.getWarpperClassName());
			generic.setType(thriftType.getType());
			return generic;
		}
		ThriftType thriftType = ThriftType.fromJavaType(type);
		generic.setValue(thriftType.getValue());
		ParameterizedType parameterizedType = (ParameterizedType) type;
		Type[] types = parameterizedType.getActualTypeArguments();
		for (Type typeArgument : types) {
			if(typeArgument instanceof ParameterizedType) {
				generic.addGeneric(fromType(typeArgument));
				continue;
			}
			ThriftType typeArgumentThriftType = ThriftType.fromJavaType((Class)typeArgument);
			if(typeArgumentThriftType == ThriftType.STRUCT) {
				typeArgumentThriftType = typeArgumentThriftType.clone();
				typeArgumentThriftType.setJavaClass((Class)typeArgument);
				typeArgumentThriftType.setValue(((Class)typeArgument).getSimpleName());
			}
			generic.addGeneric(typeArgumentThriftType);
		}
		return generic;
	}

	
}