|
|
|
@ -1,5 +1,5 @@ |
|
|
|
/* |
|
|
|
/* |
|
|
|
* Copyright 2012-2018 the original author or authors. |
|
|
|
* Copyright 2012-2019 the original author or authors. |
|
|
|
* |
|
|
|
* |
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
@ -21,7 +21,9 @@ import java.util.Collection; |
|
|
|
import java.util.Collections; |
|
|
|
import java.util.Collections; |
|
|
|
import java.util.EnumMap; |
|
|
|
import java.util.EnumMap; |
|
|
|
import java.util.HashMap; |
|
|
|
import java.util.HashMap; |
|
|
|
|
|
|
|
import java.util.List; |
|
|
|
import java.util.Map; |
|
|
|
import java.util.Map; |
|
|
|
|
|
|
|
import java.util.Map.Entry; |
|
|
|
import java.util.regex.Pattern; |
|
|
|
import java.util.regex.Pattern; |
|
|
|
import java.util.stream.Collectors; |
|
|
|
import java.util.stream.Collectors; |
|
|
|
|
|
|
|
|
|
|
|
@ -33,8 +35,10 @@ import javax.lang.model.type.DeclaredType; |
|
|
|
import javax.lang.model.type.PrimitiveType; |
|
|
|
import javax.lang.model.type.PrimitiveType; |
|
|
|
import javax.lang.model.type.TypeKind; |
|
|
|
import javax.lang.model.type.TypeKind; |
|
|
|
import javax.lang.model.type.TypeMirror; |
|
|
|
import javax.lang.model.type.TypeMirror; |
|
|
|
|
|
|
|
import javax.lang.model.type.TypeVariable; |
|
|
|
import javax.lang.model.util.SimpleTypeVisitor8; |
|
|
|
import javax.lang.model.util.SimpleTypeVisitor8; |
|
|
|
import javax.lang.model.util.Types; |
|
|
|
import javax.lang.model.util.Types; |
|
|
|
|
|
|
|
import javax.tools.Diagnostic.Kind; |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* Type Utilities. |
|
|
|
* Type Utilities. |
|
|
|
@ -73,18 +77,22 @@ class TypeUtils { |
|
|
|
|
|
|
|
|
|
|
|
private final ProcessingEnvironment env; |
|
|
|
private final ProcessingEnvironment env; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private final Types types; |
|
|
|
|
|
|
|
|
|
|
|
private final TypeExtractor typeExtractor; |
|
|
|
private final TypeExtractor typeExtractor; |
|
|
|
|
|
|
|
|
|
|
|
private final TypeMirror collectionType; |
|
|
|
private final TypeMirror collectionType; |
|
|
|
|
|
|
|
|
|
|
|
private final TypeMirror mapType; |
|
|
|
private final TypeMirror mapType; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private final Map<TypeElement, TypeDescriptor> typeDescriptors = new HashMap<>(); |
|
|
|
|
|
|
|
|
|
|
|
TypeUtils(ProcessingEnvironment env) { |
|
|
|
TypeUtils(ProcessingEnvironment env) { |
|
|
|
this.env = env; |
|
|
|
this.env = env; |
|
|
|
Types types = env.getTypeUtils(); |
|
|
|
this.types = env.getTypeUtils(); |
|
|
|
this.typeExtractor = new TypeExtractor(types); |
|
|
|
this.typeExtractor = new TypeExtractor(this.types); |
|
|
|
this.collectionType = getDeclaredType(types, Collection.class, 1); |
|
|
|
this.collectionType = getDeclaredType(this.types, Collection.class, 1); |
|
|
|
this.mapType = getDeclaredType(types, Map.class, 2); |
|
|
|
this.mapType = getDeclaredType(this.types, Map.class, 2); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
private TypeMirror getDeclaredType(Types types, Class<?> typeClass, |
|
|
|
private TypeMirror getDeclaredType(Types types, Class<?> typeClass, |
|
|
|
@ -115,14 +123,15 @@ class TypeUtils { |
|
|
|
/** |
|
|
|
/** |
|
|
|
* Return the type of the specified {@link TypeMirror} including all its generic |
|
|
|
* Return the type of the specified {@link TypeMirror} including all its generic |
|
|
|
* information. |
|
|
|
* information. |
|
|
|
|
|
|
|
* @param element the {@link TypeElement} in which this {@code type} is declared |
|
|
|
* @param type the type to handle |
|
|
|
* @param type the type to handle |
|
|
|
* @return a representation of the type including all its generic information |
|
|
|
* @return a representation of the type including all its generic information |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
public String getType(TypeMirror type) { |
|
|
|
public String getType(TypeElement element, TypeMirror type) { |
|
|
|
if (type == null) { |
|
|
|
if (type == null) { |
|
|
|
return null; |
|
|
|
return null; |
|
|
|
} |
|
|
|
} |
|
|
|
return type.accept(this.typeExtractor, null); |
|
|
|
return type.accept(this.typeExtractor, createTypeDescriptor(element)); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
public boolean isCollectionOrMap(TypeMirror type) { |
|
|
|
public boolean isCollectionOrMap(TypeMirror type) { |
|
|
|
@ -160,11 +169,49 @@ class TypeUtils { |
|
|
|
return WRAPPER_TO_PRIMITIVE.get(type.toString()); |
|
|
|
return WRAPPER_TO_PRIMITIVE.get(type.toString()); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
TypeDescriptor resolveTypeDescriptor(TypeElement element) { |
|
|
|
|
|
|
|
if (this.typeDescriptors.containsKey(element)) { |
|
|
|
|
|
|
|
return this.typeDescriptors.get(element); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
return createTypeDescriptor(element); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private TypeDescriptor createTypeDescriptor(TypeElement element) { |
|
|
|
|
|
|
|
TypeDescriptor descriptor = new TypeDescriptor(); |
|
|
|
|
|
|
|
process(descriptor, element.asType()); |
|
|
|
|
|
|
|
this.typeDescriptors.put(element, descriptor); |
|
|
|
|
|
|
|
return descriptor; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void process(TypeDescriptor descriptor, TypeMirror type) { |
|
|
|
|
|
|
|
try { |
|
|
|
|
|
|
|
if (type.getKind() == TypeKind.DECLARED) { |
|
|
|
|
|
|
|
DeclaredType declaredType = (DeclaredType) type; |
|
|
|
|
|
|
|
DeclaredType freshType = (DeclaredType) this.env.getElementUtils() |
|
|
|
|
|
|
|
.getTypeElement(this.types.asElement(type).toString()).asType(); |
|
|
|
|
|
|
|
List<? extends TypeMirror> arguments = declaredType.getTypeArguments(); |
|
|
|
|
|
|
|
for (int i = 0; i < arguments.size(); i++) { |
|
|
|
|
|
|
|
TypeMirror specificType = arguments.get(i); |
|
|
|
|
|
|
|
TypeMirror signatureType = freshType.getTypeArguments().get(i); |
|
|
|
|
|
|
|
descriptor.registerIfNecessary(signatureType, specificType); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
TypeElement element = (TypeElement) this.types.asElement(type); |
|
|
|
|
|
|
|
process(descriptor, element.getSuperclass()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
catch (Exception ex) { |
|
|
|
|
|
|
|
this.env.getMessager().printMessage(Kind.WARNING, |
|
|
|
|
|
|
|
"Failed to generated type descriptor for " + type, |
|
|
|
|
|
|
|
this.types.asElement(type)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* A visitor that extracts the fully qualified name of a type, including generic |
|
|
|
* A visitor that extracts the fully qualified name of a type, including generic |
|
|
|
* information. |
|
|
|
* information. |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
private static class TypeExtractor extends SimpleTypeVisitor8<String, Void> { |
|
|
|
private static class TypeExtractor |
|
|
|
|
|
|
|
extends SimpleTypeVisitor8<String, TypeDescriptor> { |
|
|
|
|
|
|
|
|
|
|
|
private final Types types; |
|
|
|
private final Types types; |
|
|
|
|
|
|
|
|
|
|
|
@ -173,34 +220,60 @@ class TypeUtils { |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@Override |
|
|
|
public String visitDeclared(DeclaredType type, Void none) { |
|
|
|
public String visitDeclared(DeclaredType type, TypeDescriptor descriptor) { |
|
|
|
TypeElement enclosingElement = getEnclosingTypeElement(type); |
|
|
|
TypeElement enclosingElement = getEnclosingTypeElement(type); |
|
|
|
if (enclosingElement != null) { |
|
|
|
String qualifiedName = determineQualifiedName(type, enclosingElement); |
|
|
|
return getQualifiedName(enclosingElement) + "$" |
|
|
|
|
|
|
|
+ type.asElement().getSimpleName(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
String qualifiedName = getQualifiedName(type.asElement()); |
|
|
|
|
|
|
|
if (type.getTypeArguments().isEmpty()) { |
|
|
|
if (type.getTypeArguments().isEmpty()) { |
|
|
|
return qualifiedName; |
|
|
|
return qualifiedName; |
|
|
|
} |
|
|
|
} |
|
|
|
StringBuilder name = new StringBuilder(); |
|
|
|
StringBuilder name = new StringBuilder(); |
|
|
|
name.append(qualifiedName); |
|
|
|
name.append(qualifiedName); |
|
|
|
name.append("<").append(type.getTypeArguments().stream() |
|
|
|
name.append("<").append(type.getTypeArguments().stream() |
|
|
|
.map(TypeMirror::toString).collect(Collectors.joining(","))) |
|
|
|
.map((t) -> visit(t, descriptor)).collect(Collectors.joining(","))) |
|
|
|
.append(">"); |
|
|
|
.append(">"); |
|
|
|
return name.toString(); |
|
|
|
return name.toString(); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private String determineQualifiedName(DeclaredType type, |
|
|
|
|
|
|
|
TypeElement enclosingElement) { |
|
|
|
|
|
|
|
if (enclosingElement != null) { |
|
|
|
|
|
|
|
return getQualifiedName(enclosingElement) + "$" |
|
|
|
|
|
|
|
+ type.asElement().getSimpleName(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
return getQualifiedName(type.asElement()); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
|
|
|
|
public String visitTypeVariable(TypeVariable t, TypeDescriptor descriptor) { |
|
|
|
|
|
|
|
TypeMirror typeMirror = descriptor.resolveGeneric(t); |
|
|
|
|
|
|
|
if (typeMirror != null) { |
|
|
|
|
|
|
|
if (typeMirror instanceof TypeVariable) { |
|
|
|
|
|
|
|
// Still unresolved, let's use upper bound
|
|
|
|
|
|
|
|
return visit(((TypeVariable) typeMirror).getUpperBound(), descriptor); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
else { |
|
|
|
|
|
|
|
return visit(typeMirror, descriptor); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
// Unresolved generics, use upper bound
|
|
|
|
|
|
|
|
return visit(t.getUpperBound(), descriptor); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@Override |
|
|
|
public String visitArray(ArrayType t, Void none) { |
|
|
|
public String visitArray(ArrayType t, TypeDescriptor descriptor) { |
|
|
|
return t.getComponentType().accept(this, none) + "[]"; |
|
|
|
return t.getComponentType().accept(this, descriptor) + "[]"; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@Override |
|
|
|
public String visitPrimitive(PrimitiveType t, Void none) { |
|
|
|
public String visitPrimitive(PrimitiveType t, TypeDescriptor descriptor) { |
|
|
|
return this.types.boxedClass(t).getQualifiedName().toString(); |
|
|
|
return this.types.boxedClass(t).getQualifiedName().toString(); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
|
|
|
|
protected String defaultAction(TypeMirror t, TypeDescriptor descriptor) { |
|
|
|
|
|
|
|
return t.toString(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
public String getQualifiedName(Element element) { |
|
|
|
public String getQualifiedName(Element element) { |
|
|
|
if (element == null) { |
|
|
|
if (element == null) { |
|
|
|
return null; |
|
|
|
return null; |
|
|
|
@ -230,4 +303,42 @@ class TypeUtils { |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
|
|
|
* Descriptor for a given type. |
|
|
|
|
|
|
|
*/ |
|
|
|
|
|
|
|
static class TypeDescriptor { |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private final Map<TypeVariable, TypeMirror> generics = new HashMap<>(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public Map<TypeVariable, TypeMirror> getGenerics() { |
|
|
|
|
|
|
|
return Collections.unmodifiableMap(this.generics); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public TypeMirror resolveGeneric(TypeVariable typeVariable) { |
|
|
|
|
|
|
|
return resolveGeneric(getParameterName(typeVariable)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public TypeMirror resolveGeneric(String parameterName) { |
|
|
|
|
|
|
|
return this.generics.entrySet().stream() |
|
|
|
|
|
|
|
.filter((e) -> getParameterName(e.getKey()).equals(parameterName)) |
|
|
|
|
|
|
|
.findFirst().map(Entry::getValue).orElse(null); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void registerIfNecessary(TypeMirror variable, TypeMirror resolution) { |
|
|
|
|
|
|
|
if (variable instanceof TypeVariable) { |
|
|
|
|
|
|
|
TypeVariable typeVariable = (TypeVariable) variable; |
|
|
|
|
|
|
|
if (this.generics.keySet().stream() |
|
|
|
|
|
|
|
.noneMatch((candidate) -> getParameterName(candidate) |
|
|
|
|
|
|
|
.equals(getParameterName(typeVariable)))) { |
|
|
|
|
|
|
|
this.generics.put(typeVariable, resolution); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private String getParameterName(TypeVariable typeVariable) { |
|
|
|
|
|
|
|
return typeVariable.asElement().getSimpleName().toString(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|