Как выполнить метод с тем же именем из разных аннотаций - PullRequest
Новая
       51

Как выполнить метод с тем же именем из разных аннотаций

0 голосов
/ 12 октября 2018

Несмотря на то, что мой вопрос возник из-за обработки аннотаций, мой вопрос больше связан с аннотациями Java.

Я писал некоторый код, пока не понял, что не знаю хорошего способа реализации чего-либо.

Программа использует обработку аннотаций, я пытаюсь получить значение нескольких аннотаций JAX-RS, давайте возьмем @PathParam и @QueryParam в качестве примера.Обе аннотации имеют абстрактный метод, называемый value()

Следующий фрагмент кода является примером того, как я не хочу его писать.Я должен был бы делать это для каждой аннотации JAX-RS. 

@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {

    for(Element element : roundEnv.getElementsAnnotatedWith(PathParam.class)) {
        PathParam parameter = element.getAnnotation(PathParam.class);
        String value = parameter.value();

        // Process data & more program code.
    }

    for(Element element : roundEnv.getElementsAnnotatedWith(QueryParam.class)) {
        QueryParam parameter = element.getAnnotation(QueryParam.class);
        String value = parameter.value();

        // Process data & more program code.
    }

    // Etc... do the same for other JAX-RS annotations.

    return true;
}

Я знаю, что с абстрактными классами вы можете делать следующее: 

abstract class Animal { 
    abstract String name();
}

class Dog extends Animal {
    public String name() {
        return "Dog";
    }
}

class Cat extends Animal {
    public String name() {
        return "Cat";
    }
}

Animal animal = new Cat();
System.out.println(animal.name()); // Prints 'Cat'

animal = new Dog();
System.out.println(animal.name()); // Prints 'Dog'

Но яне уверен, как сделать подобное с помощью аннотации, поскольку нет суперкласса, к которому он может быть приведен.Я представляю, что это должно быть что-то вроде этого: 

ArrayList<Class<? extends Annotation>> annotationsToCheck = 
        new ArrayList<>(Arrays.asList(PathParam.class, QueryParam.class));

for(Class<? extends Annotation> annotationToCheck : annotationsToCheck) {
    for(Element element : roundEnv.getElementsAnnotatedWith(annotationToCheck)) {

        // Somehow cast it to something so that the value() method can be accessed

        // Process data & more program code.

    }

}

Я чувствую, что я рядом, но я просто не могу понять, как это сделать.Есть ли хороший способ решить мою проблему?

1 Ответ

0 голосов
/ 13 октября 2018

В Java 9+ приведение не требуется: 

for (Element element : roundEnv.getElementsAnnotatedWithAny​(Set.of(PathParam.class,
                                                                   QueryParam.class))) {
    PathParam pathParam = element.getAnnotation(PathParam.class);
    if (pathParam != null) {
        String value = pathParam.value();
        // Process @PathParam value
    }
    QueryParam queryParam = element.getAnnotation(QueryParam.class);
    if (queryParam != null) {
        String value = queryParam.value();
        // Process @QueryParam value
    }
}

Или, если вы ожидаете только одного из них: 

for (Element element : roundEnv.getElementsAnnotatedWithAny​(Set.of(PathParam.class,
                                                                   QueryParam.class))) {
    String value = null;
    PathParam pathParam = null;
    QueryParam queryParam = null;
    if ((pathParam = element.getAnnotation(PathParam.class)) != null) {
        value = pathParam.value();
    } else if ((queryParam = element.getAnnotation(QueryParam.class)) != null) {
        value = queryParam.value();
    }
    // Process value. Exactly one of pathParam and queryParam will be non-null
}
...