java - 如何从 typeElement 获取实际类型？

Question

我正在编写自定义注释处理器。虽然我可以处理没有泛型类型的类类型，但无法获取泛型类型的实际类型。

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 和 T；我怎样才能得到KpA和KpB。

Answer 1

当您处理类型系统时，了解您准确想要什么非常重要。您是否需要这些类型参数，因为它们的类型变量在基类型的某些字段/方法中使用？例如。你的参数化基本类型类似于 Collection<SomeItem>并且您想知道 add 的返回类型？或者是MyBase某种标记界面，例如 Clonable您想要准确它的参数吗？

无论如何；让我们假设，您的情况与您的问题中描述的完全一样:MyBase是某个接口(interface)，您只是出于未知原因想要它的参数。这是一种不灵活、愚蠢的方法来实现这一点:

// 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 s(例如类和接口(interface))
2. 检查声明的类型是否是我们的基接口(interface)
3. 如果我们找到它，请将其返回给调用者，以便它可以检查类型参数

诀窍在于调用directSupertypes :正如其文档中所解释的，该方法会自动为您解析类型参数。

我们比较类型删除(又名原始类型)，因为它们期望以有意义的方式进行比较(它或多或少与比较 TypeElements 相同)。请不要尝试将泛型类型与 isSameType 进行比较——由于类型系统的工作方式，这肯定会产生令人惊讶的结果。

<小时/>

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