com.github.fridujo.automocker.utils.Annotations Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-automocker Show documentation
The newest version!
package com.github.fridujo.automocker.utils;

import java.lang.annotation.Annotation;
import java.lang.reflect.AnnotatedElement;
import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.Set;

public final class Annotations {
    private Annotations() {
    }

    /**
     * @return all {@link Annotation Annotations} found on given {@link AnnotatedElement} that have a meta-annotation of given type {@link A}
     */
    // TODO make this recursive on superclass and interfaces
    public static  Set> getAnnotationsAnnotatedWith(AnnotatedElement annotatedElement, Class annotationType) {
        return getAnnotationsAnnotatedWith(annotatedElement, annotationType, new LinkedHashSet<>());
    }

    private static  Set> getAnnotationsAnnotatedWith(AnnotatedElement annotatedElement, Class annotationType, Set visited) {
        Set> result = new HashSet<>();

        if (visited.add(annotatedElement)) {
            Arrays.stream(annotatedElement.getDeclaredAnnotations()).forEach(annotation -> {
                Annotation superAnnotation = annotation.annotationType().getAnnotation(annotationType);
                if (superAnnotation != null) {
                    result.add(new AnnotatedAnnotation(superAnnotation, annotation));
                } else {
                    result.addAll(getAnnotationsAnnotatedWith(annotation.annotationType(), annotationType, visited));
                }
            });
        }

        return result;
    }

    /**
     * Describe an annotation annotated by another annotation.
     *
     * @param  type of the meta-annotation
     */
    public static class AnnotatedAnnotation {

        private final A parentAnnotation;
        private final Annotation annotation;

        private AnnotatedAnnotation(A parentAnnotation, Annotation annotation) {
            this.parentAnnotation = parentAnnotation;
            this.annotation = annotation;
        }

        public A parentAnnotation() {
            return parentAnnotation;
        }

        public Annotation annotation() {
            return annotation;
        }
    }
}