注释处理器：从TypeMirror或TypeElement中获取所有枚举值。

Question

我不明白如何在注解处理器中检索Enum值。

我的注解是一个自定义的Java Bean验证注解。

  @StringEnumeration(enumClass = UserCivility.class)
  private String civility;

在我的注解处理器上，我可以访问这些实例。

javax.lang.model.element.AnnotationValue
javax.lang.model.type.TypeMirror
javax.lang.model.element.TypeElement

我知道它包含了关于我的枚举的数据，因为我可以在调试模式下看到它。我还可以看到 ElementKind == Enum

但是我想把那个Enum的所有名字都弄出来，谁能帮帮我。

---

编辑。 我无法访问这个Enum的Class对象 因为我们是在一个注释处理器中 而不是在标准的Java反射代码里 所以我不能调用 Class#getEnumConstants() 或 EnumSet.allOf(MyEnum.class) 除非你告诉我如何从上面提到的类型中获取Class对象。

Answer 1

我找到了一个解决方案（这个使用Guava）。

class ElementKindPredicate<T extends Element> implements Predicate<T> {
    private final ElementKind kind;
    public ElementKindPredicate(ElementKind kind) {
      Preconditions.checkArgument(kind != null);
      this.kind = kind;
    }
    @Override
    public boolean apply(T input) {
      return input.getKind().equals(kind);
    }
}

private static final ElementKindPredicate ENUM_VALUE_PREDICATE = new ElementKindPredicate(ElementKind.ENUM_CONSTANT);

public static List<String> getEnumValues(TypeElement enumTypeElement) {
    Preconditions.checkArgument(enumTypeElement.getKind() == ElementKind.ENUM);
    return FluentIterable.from(enumTypeElement.getEnclosedElements())
        .filter(ENUM_VALUE_PREDICATE)
        .transform(Functions.toStringFunction())
        .toList();
}

Answer 2

Sebastian给出的答案是正确的，但如果你使用的是Java 8或以上版本，你可以使用以下（更干净）的方法，而不是使用Google Guava。

List<String> getEnumValues(TypeElement enumTypeElement) {
    return enumTypeElement.getEnclosedElements().stream()
            .filter(element -> element.getKind().equals(ElementKind.ENUM_CONSTANT))
            .map(Object::toString)
            .collect(Collectors.toList());
}

Answer 3

这里有一个完整的例子。注意在枚举值上使用getEnumConstants。

public class Annotate {

    public enum MyValues {
        One, Two, Three     
    };

    @Retention(RetentionPolicy.RUNTIME)
    public @interface StringEnumeration {   
        MyValues enumClass();           
    }

    @StringEnumeration(enumClass = MyValues.Three)
    public static String testString = "foo";

    public static void main(String[] args) throws Exception {

        Class<Annotate> a = Annotate.class; 
        Field f = a.getField("testString");
        StringEnumeration se = f.getAnnotation(StringEnumeration.class);
        if (se != null) {
            System.out.println(se.enumClass());

            for( Object o : se.enumClass().getClass().getEnumConstants() ) {
                System.out.println(o);
            }

        }                                   
    }
}

这将打印出来。

Three
One
Two
Three