hacking - mongodb sample

Author: christophstrobl

src/main/java/org/springframework/data/aot/RepositoryRegistrationAotContribution.java

@@ -396,39 +396,25 @@ private void contributeProjection(Class<?> type, GenerationContext generationCon
 			.registerJdkProxy(type, TargetAware.class, SpringProxy.class, DecoratingProxy.class);
 	}
 
-	protected void contribute(AotRepositoryContext repositoryContext, GenerationContext generationContext) {
-
-		repositoryContext.getResolvedTypes().stream()
-			.filter(it -> !isJavaOrPrimitiveType(it))
-			.forEach(it -> contributeType(it, generationContext));
-
-		repositoryContext.getResolvedAnnotations().stream()
-			.filter(this::isSpringDataManagedAnnotation)
-			.map(MergedAnnotation::getType)
-			.forEach(it -> contributeType(it, generationContext));
-	}
-
-	private boolean isJavaOrPrimitiveType(Class<?> type) {
+	static boolean isJavaOrPrimitiveType(Class<?> type) {
 
 		return TypeUtils.type(type).isPartOf("java")
 			|| type.isPrimitive()
 			|| ClassUtils.isPrimitiveArray(type);
 	}
 
-	private boolean isSpringDataManagedAnnotation(MergedAnnotation<?> annotation) {
+	static boolean isSpringDataManagedAnnotation(MergedAnnotation<?> annotation) {
 
 		return annotation != null
 				&& (isInSpringDataNamespace(annotation.getType()) || annotation.getMetaTypes().stream()
-						.anyMatch(this::isInSpringDataNamespace));
+						.anyMatch(RepositoryRegistrationAotContribution::isInSpringDataNamespace));
 	}
 
-	private boolean isInSpringDataNamespace(Class<?> type) {
+	static boolean isInSpringDataNamespace(Class<?> type) {
 		return type != null && type.getPackage().getName().startsWith(TypeContributor.DATA_NAMESPACE);
 	}
 
-	protected void contributeType(Class<?> type, GenerationContext generationContext) {
-
-		logTrace("Contributing type information for [%s]", type);
+	static void contributeType(Class<?> type, GenerationContext generationContext) {
 		TypeContributor.contribute(type, it -> true, generationContext);
 	}
 

src/main/java/org/springframework/data/aot/RepositoryRegistrationAotProcessor.java

@@ -34,6 +34,7 @@
 import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
 import org.springframework.beans.factory.support.RegisteredBean;
 import org.springframework.beans.factory.support.RootBeanDefinition;
+import org.springframework.core.annotation.MergedAnnotation;
 import org.springframework.data.repository.config.RepositoryConfigurationExtension;
 import org.springframework.data.repository.config.RepositoryMetadata;
 import org.springframework.data.repository.core.support.RepositoryFactoryBeanSupport;
@@ -51,7 +52,7 @@
  * via the {@link FactoryBean#OBJECT_TYPE_ATTRIBUTE}.
  * </p>
  * <p>
- * With {@link RepositoryRegistrationAotContribution#contribute(AotRepositoryContext, GenerationContext)}, stores
+ * With {@link RepositoryRegistrationAotProcessor#contribute(AotRepositoryContext, GenerationContext)}, stores
  * can provide custom logic for contributing additional (eg. reflection) configuration. By default, reflection
  * configuration will be added for types reachable from the repository declaration and query methods as well as
  * all used {@link Annotation annotations} from the {@literal org.springframework.data} namespace.
@@ -83,6 +84,19 @@ public BeanRegistrationAotContribution processAheadOfTime(@NonNull RegisteredBea
 				: null;
 	}
 
+	protected void contribute(AotRepositoryContext repositoryContext, GenerationContext generationContext) {
+
+		repositoryContext.getResolvedTypes().stream()
+				.filter(it -> !RepositoryRegistrationAotContribution
+						.isJavaOrPrimitiveType(it))
+				.forEach(it -> RepositoryRegistrationAotContribution.contributeType(it, generationContext));
+
+		repositoryContext.getResolvedAnnotations().stream()
+				.filter(RepositoryRegistrationAotContribution::isSpringDataManagedAnnotation)
+				.map(MergedAnnotation::getType)
+				.forEach(it -> RepositoryRegistrationAotContribution.contributeType(it, generationContext));
+	}
+
 	private boolean isRepositoryBean(RegisteredBean bean) {
 		return bean != null && getConfigMap().containsKey(bean.getBeanName());
 	}
@@ -93,7 +107,7 @@ protected RepositoryRegistrationAotContribution newRepositoryRegistrationAotCont
 		RepositoryRegistrationAotContribution contribution =
 				RepositoryRegistrationAotContribution.fromProcessor(this).forBean(repositoryBean);
 
-		return contribution.withModuleContribution(contribution::contribute);
+		return contribution.withModuleContribution(this::contribute);
 	}
 
 	@Override

src/main/java/org/springframework/data/aot/TypeContributor.java

@@ -29,7 +29,7 @@
  * @author Christoph Strobl
  * @since 3.0
  */
-class TypeContributor {
+public class TypeContributor {
 
 	public static final String DATA_NAMESPACE = "org.springframework.data";
 
@@ -39,7 +39,7 @@ class TypeContributor {
 	 * @param type
 	 * @param contribution
 	 */
-	static void contribute(Class<?> type, GenerationContext contribution) {
+	public static void contribute(Class<?> type, GenerationContext contribution) {
 		contribute(type, Collections.emptySet(), contribution);
 	}
 
@@ -51,7 +51,7 @@ static void contribute(Class<?> type, GenerationContext contribution) {
 	 * @param contribution
 	 */
 	@SuppressWarnings("unchecked")
-	static void contribute(Class<?> type, Predicate<Class<? extends Annotation>> filter, GenerationContext contribution) {
+	public static void contribute(Class<?> type, Predicate<Class<? extends Annotation>> filter, GenerationContext contribution) {
 
 		if (type.isPrimitive()) {
 			return;
@@ -87,15 +87,15 @@ static void contribute(Class<?> type, Predicate<Class<? extends Annotation>> fil
 	 * @param annotationNamespaces
 	 * @param contribution
 	 */
-	static void contribute(Class<?> type, Set<String> annotationNamespaces, GenerationContext contribution) {
+	public static void contribute(Class<?> type, Set<String> annotationNamespaces, GenerationContext contribution) {
 		contribute(type, it -> isPartOfOrMetaAnnotatedWith(it, annotationNamespaces), contribution);
 	}
 
-	private static boolean isPartOf(Class<?> type, Set<String> namespaces) {
+	public static boolean isPartOf(Class<?> type, Set<String> namespaces) {
 		return namespaces.stream().anyMatch(namespace -> type.getPackageName().startsWith(namespace));
 	}
 
-	protected static boolean isPartOfOrMetaAnnotatedWith(Class<? extends Annotation> annotation, Set<String> namespaces) {
+	public static boolean isPartOfOrMetaAnnotatedWith(Class<? extends Annotation> annotation, Set<String> namespaces) {
 
 		if (isPartOf(annotation, namespaces)) {
 			return true;