Add TYPE_HIERARCHY_AND_ENCLOSING_CLASSES strategy
Add a `TYPE_HIERARCHY_AND_ENCLOSING_CLASSES` annotation search strategy that can be used to search the full type hierarchy as well as any enclosing classes. Closes gh-23378
This commit is contained in:
@@ -123,10 +123,12 @@ abstract class AnnotationsScanner {
|
||||
case INHERITED_ANNOTATIONS:
|
||||
return processClassInheritedAnnotations(context, source, processor, classFilter);
|
||||
case SUPERCLASS:
|
||||
return processClassHierarchy(context, new int[] {0}, source, processor, classFilter, false);
|
||||
return processClassHierarchy(context, source, processor, classFilter, false, false);
|
||||
case EXHAUSTIVE:
|
||||
case TYPE_HIERARCHY:
|
||||
return processClassHierarchy(context, new int[] {0}, source, processor, classFilter, true);
|
||||
return processClassHierarchy(context, source, processor, classFilter, true, false);
|
||||
case TYPE_HIERARCHY_AND_ENCLOSING_CLASSES:
|
||||
return processClassHierarchy(context, source, processor, classFilter, true, true);
|
||||
}
|
||||
throw new IllegalStateException("Unsupported search strategy " + searchStrategy);
|
||||
}
|
||||
@@ -184,10 +186,22 @@ abstract class AnnotationsScanner {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Nullable
|
||||
private static <C, R> R processClassHierarchy(C context, Class<?> source,
|
||||
AnnotationsProcessor<C, R> processor,
|
||||
@Nullable BiPredicate<C, Class<?>> classFilter, boolean includeInterfaces,
|
||||
boolean includeEnclosing) {
|
||||
|
||||
int[] aggregateIndex = new int[] { 0 };
|
||||
return processClassHierarchy(context, aggregateIndex, source, processor,
|
||||
classFilter, includeInterfaces, includeEnclosing);
|
||||
}
|
||||
|
||||
@Nullable
|
||||
private static <C, R> R processClassHierarchy(C context, int[] aggregateIndex,
|
||||
Class<?> source, AnnotationsProcessor<C, R> processor,
|
||||
@Nullable BiPredicate<C, Class<?>> classFilter, boolean includeInterfaces) {
|
||||
@Nullable BiPredicate<C, Class<?>> classFilter, boolean includeInterfaces,
|
||||
boolean includeEnclosing) {
|
||||
|
||||
R result = processor.doWithAggregate(context, aggregateIndex[0]);
|
||||
if (result != null) {
|
||||
@@ -205,7 +219,7 @@ abstract class AnnotationsScanner {
|
||||
if (includeInterfaces) {
|
||||
for (Class<?> interfaceType : source.getInterfaces()) {
|
||||
R interfacesResult = processClassHierarchy(context, aggregateIndex,
|
||||
interfaceType, processor, classFilter, true);
|
||||
interfaceType, processor, classFilter, true, includeEnclosing);
|
||||
if (interfacesResult != null) {
|
||||
return interfacesResult;
|
||||
}
|
||||
@@ -214,11 +228,21 @@ abstract class AnnotationsScanner {
|
||||
Class<?> superclass = source.getSuperclass();
|
||||
if (superclass != Object.class && superclass != null) {
|
||||
R superclassResult = processClassHierarchy(context, aggregateIndex,
|
||||
superclass, processor, classFilter, includeInterfaces);
|
||||
superclass, processor, classFilter, includeInterfaces,
|
||||
includeEnclosing);
|
||||
if (superclassResult != null) {
|
||||
return superclassResult;
|
||||
}
|
||||
}
|
||||
Class<?> enclosingClass = source.getEnclosingClass();
|
||||
if (includeEnclosing && enclosingClass != null) {
|
||||
R enclosingResult = processClassHierarchy(context, aggregateIndex,
|
||||
enclosingClass, processor, classFilter, includeInterfaces,
|
||||
includeEnclosing);
|
||||
if (enclosingResult != null) {
|
||||
return enclosingResult;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
@@ -237,6 +261,7 @@ abstract class AnnotationsScanner {
|
||||
processor, classFilter, source, false);
|
||||
case EXHAUSTIVE:
|
||||
case TYPE_HIERARCHY:
|
||||
case TYPE_HIERARCHY_AND_ENCLOSING_CLASSES:
|
||||
return processMethodHierarchy(context, new int[] {0}, source.getDeclaringClass(),
|
||||
processor, classFilter, source, true);
|
||||
}
|
||||
|
||||
@@ -434,12 +434,22 @@ public interface MergedAnnotations extends Iterable<MergedAnnotation<Annotation>
|
||||
EXHAUSTIVE,
|
||||
|
||||
/**
|
||||
* Perform a full search of the entire type hierarchy , including
|
||||
* Perform a full search of the entire type hierarchy, including
|
||||
* superclasses and implemented interfaces. Superclass annotations do
|
||||
* not need to be meta-annotated with {@link Inherited @Inherited}.
|
||||
*/
|
||||
TYPE_HIERARCHY
|
||||
TYPE_HIERARCHY,
|
||||
|
||||
/**
|
||||
* Perform a full search of the entire type hierarchy on the source
|
||||
* <em>and</em> any enclosing classes. This strategy is similar to
|
||||
* {@link #TYPE_HIERARCHY} except that {@link Class#getEnclosingClass()
|
||||
* enclosing classes} are also searched. Superclass annotations do not
|
||||
* need to be meta-annotated with {@link Inherited @Inherited}. When
|
||||
* searching a {@link Method} source, this strategy is identical to
|
||||
* {@link #TYPE_HIERARCHY}.
|
||||
*/
|
||||
TYPE_HIERARCHY_AND_ENCLOSING_CLASSES
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user