222 lines
12 KiB
Java
222 lines
12 KiB
Java
package com.moparisthebest.jdbc.codegen;
|
|
|
|
import com.moparisthebest.jdbc.CompilingRowToObjectMapper;
|
|
import com.moparisthebest.jdbc.ResultSetMapper;
|
|
|
|
import javax.annotation.processing.ProcessingEnvironment;
|
|
import javax.lang.model.element.ExecutableElement;
|
|
import javax.lang.model.type.ArrayType;
|
|
import javax.lang.model.type.DeclaredType;
|
|
import javax.lang.model.type.TypeKind;
|
|
import javax.lang.model.type.TypeMirror;
|
|
import javax.lang.model.util.Elements;
|
|
import javax.lang.model.util.Types;
|
|
import java.io.IOException;
|
|
import java.io.Writer;
|
|
import java.lang.reflect.Type;
|
|
import java.util.*;
|
|
|
|
import static com.moparisthebest.jdbc.codegen.JdbcMapperProcessor.typeMirrorToClass;
|
|
|
|
/**
|
|
* Created by mopar on 5/26/17.
|
|
*/
|
|
public class CompileTimeResultSetMapper {
|
|
|
|
private final Types types;
|
|
private final TypeMirror collectionType, mapType, mapCollectionType, iteratorType, listIteratorType;
|
|
|
|
public CompileTimeResultSetMapper(final ProcessingEnvironment processingEnv) {
|
|
types = processingEnv.getTypeUtils();
|
|
final Elements elements = processingEnv.getElementUtils();
|
|
|
|
collectionType = types.getDeclaredType(elements.getTypeElement(Collection.class.getCanonicalName()), types.getWildcardType(null, null));
|
|
mapType = types.getDeclaredType(elements.getTypeElement(Map.class.getCanonicalName()), types.getWildcardType(null, null), types.getWildcardType(null, null));
|
|
mapCollectionType = types.getDeclaredType(elements.getTypeElement(Map.class.getCanonicalName()), types.getWildcardType(null, null), types.getWildcardType(collectionType, null));
|
|
|
|
iteratorType = types.getDeclaredType(elements.getTypeElement(Iterator.class.getCanonicalName()), types.getWildcardType(null, null));
|
|
listIteratorType = types.getDeclaredType(elements.getTypeElement(ListIterator.class.getCanonicalName()), types.getWildcardType(null, null));
|
|
}
|
|
|
|
@SuppressWarnings({"unchecked"})
|
|
public void mapToResultType(final Writer w, final String[] keys, final ExecutableElement eeMethod, final int arrayMaxLength, final Calendar cal) throws IOException, NoSuchMethodException, ClassNotFoundException {
|
|
//final Method m = fromExecutableElement(eeMethod);
|
|
//final Class returnType = m.getReturnType();
|
|
final TypeMirror returnTypeMirror = eeMethod.getReturnType();
|
|
//final Class returnType = typeMirrorToClass(returnTypeMirror);
|
|
if (returnTypeMirror.getKind() == TypeKind.ARRAY) {
|
|
final TypeMirror componentType = ((ArrayType) returnTypeMirror).getComponentType();
|
|
toArray(w, keys, componentType, typeMirrorToClass(componentType), arrayMaxLength, cal);
|
|
} else if (types.isAssignable(returnTypeMirror, collectionType)) {
|
|
final List<? extends TypeMirror> typeArguments = ((DeclaredType) returnTypeMirror).getTypeArguments();
|
|
toCollection(w, keys, returnTypeMirror, (Class<? extends Collection>)typeMirrorToClass(returnTypeMirror), typeArguments.get(0), (Class) getActualTypeArguments(typeArguments)[0], arrayMaxLength, cal);
|
|
} else if (types.isAssignable(returnTypeMirror, mapType)) {
|
|
final List<? extends TypeMirror> typeArguments = ((DeclaredType) returnTypeMirror).getTypeArguments();
|
|
final Type[] typeArgs = getActualTypeArguments(typeArguments);
|
|
//if (types[1] instanceof ParameterizedType) { // for collectionMaps
|
|
if (types.isAssignable(returnTypeMirror, mapCollectionType)) { // for collectionMaps
|
|
//final ParameterizedType pt = (ParameterizedType) types[1];
|
|
//final Class collectionType = (Class) pt.getRawType();
|
|
final TypeMirror collectionTypeMirror = typeArguments.get(1);
|
|
final Class collectionType = typeMirrorToClass(collectionTypeMirror);
|
|
if (Collection.class.isAssignableFrom(collectionType)) {
|
|
final TypeMirror componentTypeMirror = ((DeclaredType) collectionTypeMirror).getTypeArguments().get(0);
|
|
//final Class componentType = (Class) pt.getActualTypeArguments()[0];
|
|
final Class componentType = typeMirrorToClass(componentTypeMirror);
|
|
toMapCollection(w, keys,
|
|
returnTypeMirror, (Class<? extends Map>)typeMirrorToClass(returnTypeMirror),
|
|
typeArguments.get(0), (Class) typeArgs[0],
|
|
collectionTypeMirror, collectionType,
|
|
componentTypeMirror, componentType,
|
|
arrayMaxLength, cal);
|
|
return;
|
|
}
|
|
}
|
|
toMap(w, keys, returnTypeMirror, (Class<? extends Map>)typeMirrorToClass(returnTypeMirror), typeArguments.get(0), (Class) typeArgs[0], typeArguments.get(1), (Class) typeArgs[1], arrayMaxLength, cal);
|
|
} else if (types.isAssignable(returnTypeMirror, iteratorType)) {
|
|
final List<? extends TypeMirror> typeArguments = ((DeclaredType) returnTypeMirror).getTypeArguments();
|
|
if (types.isAssignable(returnTypeMirror, listIteratorType))
|
|
toListIterator(w, keys, typeArguments.get(0), (Class) getActualTypeArguments(typeArguments)[0], arrayMaxLength, cal);
|
|
else
|
|
toIterator(w, keys, typeArguments.get(0), (Class) getActualTypeArguments(typeArguments)[0], arrayMaxLength, cal);
|
|
} else {
|
|
toObject(w, keys, returnTypeMirror, typeMirrorToClass(returnTypeMirror), cal);
|
|
}
|
|
}
|
|
|
|
public <K, T> CompilingRowToObjectMapper<K, T> getRowMapper(final String[] keys, Class<T> returnTypeClass, Calendar cal, Class<?> mapValType, Class<K> mapKeyType) {
|
|
return new CompilingRowToObjectMapper<K, T>(keys, returnTypeClass, cal, mapValType, mapKeyType);
|
|
}
|
|
|
|
public void writeObject(final Writer w, final String[] keys, final TypeMirror returnTypeMirror, final Class<?> returnType, final Calendar cal) throws IOException {
|
|
getRowMapper(keys, returnType, cal, null, null).gen(w, returnTypeMirror.toString());
|
|
}
|
|
|
|
public void toObject(final Writer w, final String[] keys, final TypeMirror returnTypeMirror, final Class<?> returnType, final Calendar cal) throws IOException {
|
|
w.write("\t\t\tif(rs.next()) {\n");
|
|
writeObject(w, keys, returnTypeMirror, returnType, cal);
|
|
w.write("\t\t\t\treturn ret;\n\t\t\t} else {\n\t\t\t\treturn null;\n\t\t\t}\n");
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
public <T extends Collection<E>, E> void writeCollection(final Writer w, final String[] keys, final String returnTypeString, Class<T> collectionType, final TypeMirror componentTypeMirror, Class<E> componentType, int arrayMaxLength, Calendar cal) throws IOException {
|
|
collectionType = (Class<T>) ResultSetMapper.getConcreteClass(collectionType, ArrayList.class);
|
|
w.write("\t\t\tfinal ");
|
|
w.write(returnTypeString);
|
|
w.write(" _colret = new ");
|
|
w.write(collectionType.getCanonicalName());
|
|
w.write(returnTypeString.substring(returnTypeString.indexOf('<')));
|
|
w.write("();\n\t\t\twhile(rs.next()) {\n");
|
|
writeObject(w, keys, componentTypeMirror, componentType, cal);
|
|
w.write("\t\t\t\t_colret.add(ret);\n\t\t\t}\n");
|
|
}
|
|
|
|
public <T extends Collection<E>, E> void toCollection(final Writer w, final String[] keys, final TypeMirror collectionTypeMirror, Class<T> collectionType, final TypeMirror componentTypeMirror, Class<E> componentType, int arrayMaxLength, Calendar cal) throws IOException {
|
|
writeCollection(w, keys, collectionTypeMirror.toString(), collectionType, componentTypeMirror, componentType, arrayMaxLength, cal);
|
|
w.write("\t\t\treturn _colret;\n");
|
|
}
|
|
|
|
public <E> void toArray(final Writer w, final String[] keys, final TypeMirror componentTypeMirror, Class<E> componentType, int arrayMaxLength, Calendar cal) throws IOException {
|
|
final String returnTypeString = componentTypeMirror.toString();
|
|
writeCollection(w, keys, "java.util.List<" + returnTypeString + ">", ArrayList.class, componentTypeMirror, componentType, arrayMaxLength, cal);
|
|
w.write("\t\t\treturn _colret.toArray(new ");
|
|
w.write(returnTypeString);
|
|
w.write("[_colret.size()]);\n");
|
|
}
|
|
|
|
public <E> void toListIterator(final Writer w, final String[] keys, final TypeMirror componentTypeMirror, Class<E> componentType, int arrayMaxLength, Calendar cal) throws IOException {
|
|
final String returnTypeString = componentTypeMirror.toString();
|
|
writeCollection(w, keys, "java.util.List<" + returnTypeString + ">", ArrayList.class, componentTypeMirror, componentType, arrayMaxLength, cal);
|
|
w.write("\t\t\treturn _colret.listIterator();\n");
|
|
}
|
|
|
|
public <E> void toIterator(final Writer w, final String[] keys, final TypeMirror componentTypeMirror, Class<E> componentType, int arrayMaxLength, Calendar cal) throws IOException {
|
|
final String returnTypeString = componentTypeMirror.toString();
|
|
writeCollection(w, keys, "java.util.List<" + returnTypeString + ">", ArrayList.class, componentTypeMirror, componentType, arrayMaxLength, cal);
|
|
w.write("\t\t\treturn _colret.iterator();\n");
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
public <T extends Map<K, E>, K, E> void toMap(final Writer w, final String[] keys, final TypeMirror mapTypeMirror, Class<T> mapType, final TypeMirror mapKeyTypeMirror, Class<E> mapKeyType, final TypeMirror componentTypeMirror, Class<E> componentType, int arrayMaxLength, Calendar cal) throws IOException {
|
|
mapType = (Class<T>) ResultSetMapper.getConcreteClass(mapType, HashMap.class);
|
|
final String returnTypeString = mapTypeMirror.toString();
|
|
w.write("\t\t\tfinal ");
|
|
w.write(returnTypeString);
|
|
w.write(" _colret = new ");
|
|
w.write(mapType.getCanonicalName());
|
|
w.write(returnTypeString.substring(returnTypeString.indexOf('<')));
|
|
w.write("();\n\t\t\twhile(rs.next()) {\n");
|
|
|
|
//writeObject(w, keys, componentTypeMirror, componentType, cal);
|
|
final CompilingRowToObjectMapper rm = getRowMapper(keys, componentType, cal, null, mapKeyType);
|
|
rm.gen(w, componentTypeMirror.toString());
|
|
w.write("\t\t\t\t_colret.put(");
|
|
rm.extractColumnValueString(w, 1, mapKeyType);
|
|
w.write(", ret);\n\t\t\t}\n");
|
|
w.write("\t\t\treturn _colret;\n");
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
public <T extends Map<K, E>, K, E extends Collection<C>, C> void toMapCollection(final Writer w, final String[] keys,
|
|
final TypeMirror mapTypeMirror, Class<T> mapType,
|
|
final TypeMirror mapKeyTypeMirror, Class<E> mapKeyType,
|
|
final TypeMirror collectionTypeMirror, Class<E> collectionType,
|
|
final TypeMirror componentTypeMirror, Class<E> componentType,
|
|
int arrayMaxLength, Calendar cal) throws IOException {
|
|
mapType = (Class<T>) ResultSetMapper.getConcreteClass(mapType, HashMap.class);
|
|
collectionType = (Class<E>) ResultSetMapper.getConcreteClass(collectionType, ArrayList.class);
|
|
final String returnTypeString = mapTypeMirror.toString();
|
|
final String collectionTypeString = collectionTypeMirror.toString();
|
|
w.write("\t\t\tfinal ");
|
|
w.write(returnTypeString);
|
|
w.write(" _colret = new ");
|
|
w.write(mapType.getCanonicalName());
|
|
w.write(returnTypeString.substring(returnTypeString.indexOf('<')));
|
|
w.write("();\n\t\t\twhile(rs.next()) {\n");
|
|
|
|
//writeObject(w, keys, componentTypeMirror, componentType, cal);
|
|
final CompilingRowToObjectMapper rm = getRowMapper(keys, componentType, cal, null, mapKeyType);
|
|
rm.gen(w, componentTypeMirror.toString());
|
|
|
|
w.write("\t\t\t\tfinal ");
|
|
w.write(mapKeyTypeMirror.toString());
|
|
w.write(" _colkey = ");
|
|
rm.extractColumnValueString(w, 1, mapKeyType);
|
|
w.write(";\n\t\t\t\t");
|
|
|
|
w.write(collectionTypeString);
|
|
w.write(" _collist = _colret.get(_colkey);\n\t\t\t\tif(_collist == null) {\n\t\t\t\t\t_collist = new ");
|
|
w.write(collectionType.getCanonicalName());
|
|
w.write(collectionTypeString.substring(collectionTypeString.indexOf('<')));
|
|
w.write("();\n\t\t\t\t\t_colret.put(_colkey, _collist);\n\t\t\t\t}\n\t\t\t\t_collist.add(ret);\n\t\t\t}\n\t\t\treturn _colret;\n");
|
|
}
|
|
|
|
private static Type[] getActualTypeArguments(final List<? extends TypeMirror> typeArguments) throws ClassNotFoundException {
|
|
final Type[] ret = new Type[typeArguments.size()];
|
|
int x = -1;
|
|
for(final TypeMirror tm : typeArguments)
|
|
ret[++x] = typeMirrorToClass(tm);
|
|
return ret;
|
|
}
|
|
|
|
/*
|
|
|
|
private static Type[] getActualTypeArguments(Method m) {
|
|
return ((ParameterizedType) m.getGenericReturnType()).getActualTypeArguments();
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
private Method fromExecutableElement(final ExecutableElement eeMethod) throws ClassNotFoundException, NoSuchMethodException {
|
|
final Class c = Class.forName(eeMethod.getEnclosingElement().asType().toString());
|
|
final List<? extends VariableElement> params = eeMethod.getParameters();
|
|
final Class<?>[] parameterTypes = new Class[params.size()];
|
|
int count = -1;
|
|
for (final VariableElement param : params) {
|
|
parameterTypes[++count] = typeMirrorToClass(param.asType());
|
|
}
|
|
return c.getDeclaredMethod(eeMethod.getSimpleName().toString(), parameterTypes);
|
|
}
|
|
|
|
*/
|
|
}
|