11

In writing an Annotation Processor using the Java 6 API, I came across a need to handle all Maps in a particular fashion, but I'm clearly misunderstanding what the API is intended to do or how to invoke it. Here's the code that's making me unhappy:

import javax.lang.model.element.Element;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.util.Elements;
import javax.lang.model.util.Types;
import javax.annotation.processing.ProcessingEnvironment;
...

public String doThing(Element el, ProcessingEnvironment processingEnv) {
    // Utilities from the ProcessingEnvironment
    Types typeUtils = processingEnv.getTypeUtils();
    Elements elementUtils = processingEnv.getElementUtils();

    // The type of the element I'm handling
    TypeMirror elType = el.asType();

    // Compare the element's type to Map
    TypeMirror mapType = elementUtils.getTypeElement("java.util.Map").asType();

    System.out.println(elType + " > " + mapType + " = " + typeUtils.isSubtype(elType, mapType));
    System.out.println(mapType + " > " + elType + " = " + typeUtils.isSubtype(mapType, elType));
    System.out.println(elType + " > " + mapType + " = " + typeUtils.isAssignable(elType, mapType));
    System.out.println(mapType + " > " + elType + " = " + typeUtils.isAssignable(mapType, elType));

    // Compare the element's type to HashMap
    TypeMirror hashmapType = elementUtils.getTypeElement("java.util.HashMap").asType();

    System.out.println(elType + " > " + hashmapType + " = " + typeUtils.isSubtype(elType, hashmapType));
    System.out.println(hashmapType + " > " + elType + " = " + typeUtils.isSubtype(hashmapType, elType));
    System.out.println(elType + " > " + hashmapType + " = " + typeUtils.isAssignable(elType, hashmapType));
    System.out.println(hashmapType + " > " + elType + " = " + typeUtils.isAssignable(hashmapType, elType));


    // Compare the element's type to Object
    TypeMirror objectType = elementUtils.getTypeElement("java.lang.Object").asType();

    System.out.println(elType + " > " + objectType + " = " + typeUtils.isSubtype(elType, objectType));
    System.out.println(objectType + " > " + elType + " = " + typeUtils.isSubtype(objectType, elType));
    System.out.println(elType + " > " + objectType + " = " + typeUtils.isAssignable(elType, objectType));
    System.out.println(objectType + " > " + elType + " = " + typeUtils.isAssignable(objectType, elType));
}

Given that, here's the output of it:

java.util.HashMap<K,V> > java.util.Map<K,V> = false
java.util.Map<K,V> > java.util.HashMap<K,V> = false
java.util.HashMap<K,V> > java.util.Map<K,V> = false
java.util.Map<K,V> > java.util.HashMap<K,V> = false

java.util.HashMap<K,V> > java.util.HashMap<K,V> = true
java.util.HashMap<K,V> > java.util.HashMap<K,V> = true
java.util.HashMap<K,V> > java.util.HashMap<K,V> = true
java.util.HashMap<K,V> > java.util.HashMap<K,V> = true

java.util.HashMap<K,V> > java.lang.Object = true
java.lang.Object > java.util.HashMap<K,V> = false
java.util.HashMap<K,V> > java.lang.Object = true
java.lang.Object > java.util.HashMap<K,V> = false

This makes perfect sense to me except for the first block where I'd expect a HashMap element to be assignable to Map, and I'd expect HashMap to be a subtype of Map.

What am I missing here?

4

2 回答 2

9

I suspect it's because of the type variables. HashMap<String, String> is assignable to Map<String, String> but without the concrete instantiation of the type variables you can't be sure that an arbitrary HashMap<A,B> is assignable to Map<X,Y>.

如果您使用通配符实例化变量，那么您应该得到您期望的结果

DeclaredType wildcardMap = typeUtils.getDeclaredType(
    elementUtils.getTypeElement("java.util.Map"),
    typeUtils.getWildcardType(null, null),
    typeUtils.getWildcardType(null, null));

这将为您提供类型 mirror Map<?,?>，所有HashMap实例化都可以分配给它。

于 2012-10-05T16:18:40.563 回答
2

更新（2016 年 3 月）：根据 @user1643723 的评论，似乎有一个types.erasure(TypeMirror)我在 2012 年不知道的库函数。

根据Ian's Answer，我现在使用以下方法来匹配我在问题中为 Map 描述的基本类型。

TypeElement COLLECTION = elementUtils.getTypeElement("java.util.Collection");
TypeElement MAP = elementUtils.getTypeElement("java.util.Map");
TypeElement VOID = elementUtils.getTypeElement("java.lang.Void");
WildcardType WILDCARD_TYPE_NULL = typeUtils.getWildcardType(null, null);
Map<String,DeclaredType> cachedParentTypes = new HashMap<String, DeclaredType>();

...

public static boolean isA(TypeMirror type, TypeElement typeElement) {

    // Have we used this type before?
    DeclaredType parentType = cachedParentTypes.get(typeElement.getQualifiedName().toString());
    if (parentType == null) {
        // How many generic type parameters does this typeElement require?
        int genericsCount = typeElement.getTypeParameters().size();

        // Fill the right number of types with nulls
        TypeMirror[] types = new TypeMirror[genericsCount];
        for (int i = 0; i < genericsCount; i++) {
            types[i] = WILDCARD_TYPE_NULL;
        }

        // Locate the correct DeclaredType to match with the type
        parentType = typeUtils.getDeclaredType(typeElement, types);

        // Remember this DeclaredType
        cachedParentTypes.put(typeElement.getQualifiedName().toString(), parentType);
    }

    // Is the given type able to be assigned as the typeElement?
    return typeUtils.isAssignable(type, parentType);
}

我像这样调用

if (isA(elType, VOID)) {
    isVoid = true;
} else if (isA(elType, COLLECTION) || elType.getKind() == TypeKind.ARRAY) {
    isCollectionOrArray = true;
} else if (isA(elType, MAP)){
    isMap = true;
}
于 2012-10-06T19:16:00.863 回答