Question

我不明白如何在注释处理器中检索枚举值。

我的注释是一个自定义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的所有名字，请有人帮助我。

编辑：我无法访问此枚举的Class对象，因为我们在一个注释处理器中，而不是标准的Java反射代码。因此，除非您告诉我如何从上述类型中获取Class对象，否则我无法调用Class#getEnumConstants()或EnumSet.allOf(MyEnum.class)。

Answer 1

我找到了一个解决方案（这使用了番石榴）：

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

塞巴斯蒂安给出的答案是正确的，但是如果您使用的是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

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

3 个答案: