如何从typeElement获取实际类型?

时间:2017-12-29 19:44:03

标签: java annotation-processing

我正在编写自定义注释处理器。虽然我能够处理没有泛型类型的类类型,但无法获得泛型类型的实际类型。

MyBase界面

interface MyBase<T, S> {
  void method1(S, T);
}

示例Class1

public class KpA{
 ...
}

示例Class2

public class KpB{
 ...
}

目标注释课程

@MyAnnotation
interface MyHandler extends MyBase<KpA, KpB>{

}

处理代码段

    TypeElement[] getActualTypes(ProcessingEnvironment pEnv, TypeElement element){
        List<TypeElement> myBaseElement = element.getInterfaces().stream().map(v-> (TypeElement) pEnv.getTypeUtils().asElement(v)).collect(Collectors.toList());
        myBaseElement.get(0).getTypeParameters().stream().forEach(System.out::println);
         return null;
    }

当我Sysout getTypeParameters时,我输出为S and T;我怎样才能获得KpA and KpB

1 个答案:

答案 0 :(得分:1)

当您处理类型系统时,了解您正好想要的内容非常重要。您是否需要这些类型参数,因为它们的类型变量用于某些基本类型的字段/方法?例如。你的参数化基类型是Collection<SomeItem>,你想知道add的返回类型吗?或者是MyBase某种标记界面,例如Clonable,你想要精确地它的参数吗?

无论;让我们假设,您的情况与您的问题中描述的情况完全相同:MyBase是某个界面,您只是因为未知原因而想要它的参数。这是一种不灵活,愚蠢的方法来实现这一目标:

// instance of javax.lang.model.util.Types
protected final Types types;

// instance of javax.lang.model.util.Elements
protected final Elements el;

// TypeMirror for java.lang.Object
protected final TypeMirror theObject = ...

private static final TYPE_REFINER = new DeclaredTypeRefiner();

private class DeclaredTypeRefiner extends TypeKindVisitor6<DeclaredType, TypeMirror> {
  @Override
  public DeclaredType visitDeclared(DeclaredType type, TypeMirror desirableParent) {
    if (types.isSameType(types.erasure(type), desirableParent)) {
      return type;
    }

    return null;
  }

  @Override
  public DeclaredType visitUnknown(TypeMirror typeMirror, TypeMirror typeMirror2) {
    return defaultAction(typeMirror, typeMirror2);
  }

  @Override
  protected DeclaredType defaultAction(TypeMirror type, TypeMirror desirableParent) {
    if (types.isSameType(type, theObject)) {
      return null;
    }

    final List<? extends TypeMirror> superTypes = types.directSupertypes(type);

    for (TypeMirror parent : superTypes) {
        final DeclaredType discovered = visit(parent, desirableParent);

      if (discovered != null) {
         return discovered;
      }
    }

    return null;
  }
}

public TypeMirror getTypeArg(TypeMirror actual, TypeMirror base, int idx) {
  DeclaredType found = TYPE_REFINER.visit(actual, types.erasure(base));

  return found.getTypeArguments().get(idx);
}

获取第二个参数的类型(例如KpB):

// type mirror of MyHandler goes here
TypeMirror concrete = ...

// your base type goes here. Real code should cache it
TypeMirror base = types.getDeclaredType(el.getTypeElement("com.example.MyBase"));

// boom
TypeMirror arg = typeHelper.getTypeArg(concrete, base, 1);

以下是这里发生的事情:

  1. 迭代寻找DeclaredType的父类型(例如类和接口)
  2. 检查声明的类型是否是我们的基本接口
  3. 如果我们找到它,请将其返回给调用者,以便它可以检查类型args
  4. 诀窍在于调用directSupertypes:正如其文档中所解释的那样,该方法为您自动解析了类型参数。

    我们比较类型擦除(又称原始类型),因为它们有意义地以有意义的方式进行比较(它与比较TypeElements或多或少相同)。请不要尝试将泛型类型与isSameType进行比较 - 由于类型系统的工作方式,这可以保证产生令人惊讶的结果。

    请注意,由于多重继承,上述代码可能无法始终提供正确的结果:MyHandler可能会从多个父级继承MyBase,从而导致对T和{的真实身份的混淆{1}}。如果这些变量没有在任何地方使用(例如我们谈论的是纯粹的标记接口),那将会是一个谜:编译器并不关心,所以没有人知道(我认为,你不想手动计算上限/下限,不是吗?)如果在某些方法/字段中使用这些变量,您应该能够通过检查S上的那些方法/字段来获取它们的类型(如何做到这一点是一个有点棘手的问题值得另一个问题)。