Move reflection hint handling to repository context.

Author: christophstrobl

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

@@ -33,6 +33,7 @@
 import org.springframework.aot.generate.GenerationContext;
 import org.springframework.aot.hint.MemberCategory;
 import org.springframework.aot.hint.TypeReference;
+import org.springframework.aot.hint.annotation.ReflectiveRuntimeHintsRegistrar;
 import org.springframework.beans.factory.BeanFactory;
 import org.springframework.beans.factory.NoSuchBeanDefinitionException;
 import org.springframework.beans.factory.aot.AotProcessingException;
@@ -63,6 +64,7 @@ class DefaultAotContext implements AotContext {
 
 	private final Map<Class<?>, ContextualTypeConfiguration> typeConfigurations = new HashMap<>();
 	private final Environment environment;
+	private final ReflectiveRuntimeHintsRegistrar runtimeHintsRegistrar = new ReflectiveRuntimeHintsRegistrar();
 
 	public DefaultAotContext(BeanFactory beanFactory, Environment environment) {
 		this(beanFactory, environment, new AotMappingContext());
@@ -257,6 +259,7 @@ private void doContribute(Environment environment, GenerationContext generationC
 			}
 
 			if (forDataBinding) {
+				runtimeHintsRegistrar.registerRuntimeHints(generationContext.getRuntimeHints(), type);
 				TypeContributor.contribute(type, Set.of(TypeContributor.DATA_NAMESPACE), generationContext);
 			}
 

src/main/java/org/springframework/data/repository/config/RepositoryRegistrationAotProcessor.java

@@ -26,12 +26,9 @@
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.jspecify.annotations.Nullable;
-
 import org.springframework.aot.generate.GenerationContext;
 import org.springframework.aot.hint.MemberCategory;
-import org.springframework.aot.hint.RuntimeHints;
 import org.springframework.aot.hint.TypeReference;
-import org.springframework.aot.hint.annotation.ReflectiveRuntimeHintsRegistrar;
 import org.springframework.beans.BeansException;
 import org.springframework.beans.factory.BeanFactory;
 import org.springframework.beans.factory.BeanFactoryAware;
@@ -216,7 +213,7 @@ protected void registerRepositoryCompositionHints(AotRepositoryContext repositor
 	 * @param generationContext the generation context.
 	 * @since 4.0
 	 */
-	private void configureTypeContributions(AotRepositoryContext repositoryContext, GenerationContext generationContext) {
+	protected void configureTypeContributions(AotRepositoryContext repositoryContext, GenerationContext generationContext) {
 
 		RepositoryInformation information = repositoryContext.getRepositoryInformation();
 
@@ -243,25 +240,22 @@ private void configureTypeContributions(AotRepositoryContext repositoryContext,
 	 * ({@link AotRepositoryContext#getResolvedTypes()})
 	 *
 	 * @param repositoryContext the repository context.
-	 * @param generationContext the generation context.
 	 * @since 4.0
 	 */
-	protected void configureDomainTypeContributions(AotRepositoryContext repositoryContext,
-			GenerationContext generationContext) {
+	private void configureDomainTypeContributions(AotRepositoryContext repositoryContext, GenerationContext generationContext) {
 
 		RepositoryInformation information = repositoryContext.getRepositoryInformation();
-		RuntimeHints hints = generationContext.getRuntimeHints();
 
 		// Domain types, related types, projections
-		ReflectiveRuntimeHintsRegistrar registrar = new ReflectiveRuntimeHintsRegistrar();
+		repositoryContext.typeConfiguration(information.getDomainType(), config -> config.forDataBinding().forQuerydsl());
+
 		Stream.concat(Stream.of(information.getDomainType()), information.getAlternativeDomainTypes().stream())
 				.forEach(it -> {
-
-					registrar.registerRuntimeHints(hints, it);
-					configureTypeContribution(it, repositoryContext);
+					repositoryContext.typeConfiguration(information.getDomainType(),
+							config -> config.forDataBinding().forQuerydsl().contributeAccessors());
 				});
 
-		// TODO: Looks like a duplicate
+		// Domain types my be part of this, but it also contains reachable ones.
 		repositoryContext.getResolvedTypes().stream()
 				.filter(it -> TypeContributor.isPartOf(it, Set.of(information.getDomainType().getPackageName())))
 				.forEach(it -> configureTypeContribution(it, repositoryContext));