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 and T;我怎样才能获得KpA and KpB。

Answer 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);

以下是这里发生的事情：

迭代寻找DeclaredType的父类型（例如类和接口）
检查声明的类型是否是我们的基本接口
如果我们找到它，请将其返回给调用者，以便它可以检查类型args

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

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

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

如何从typeElement获取实际类型？

1 个答案: