Apply fetch graph by name in AOT repositories.

This commit fixes an issue where the entity graph was read but never applied to the query.

Author: christophstrobl

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/aot/JpaCodeBlocks.java

@@ -559,23 +559,22 @@ private CodeBlock applyEntityGraph(AotEntityGraph entityGraph, String queryVaria
 
 			CodeBlock.Builder builder = CodeBlock.builder();
 
+			String entityGraphVariable = context.localVariable("entityGraph");
 			if (StringUtils.hasText(entityGraph.name())) {
 
 				builder.addStatement("$T<?> $L = $L.getEntityGraph($S)", jakarta.persistence.EntityGraph.class,
-						context.localVariable("entityGraph"),
-						context.fieldNameOf(EntityManager.class), entityGraph.name());
+						entityGraphVariable, context.fieldNameOf(EntityManager.class), entityGraph.name());
 			} else {
 
-				builder.addStatement("$T<$T> $L = $L.createEntityGraph($T.class)",
-						jakarta.persistence.EntityGraph.class, context.getDomainType(),
-						context.localVariable("entityGraph"),
-						context.fieldNameOf(EntityManager.class), context.getDomainType());
+				builder.addStatement("$T<$T> $L = $L.createEntityGraph($T.class)", jakarta.persistence.EntityGraph.class,
+						context.getDomainType(), entityGraphVariable, context.fieldNameOf(EntityManager.class),
+						context.getDomainType());
 
 				for (String attributePath : entityGraph.attributePaths()) {
 
 					String[] pathComponents = StringUtils.delimitedListToStringArray(attributePath, ".");
 
-					StringBuilder chain = new StringBuilder(context.localVariable("entityGraph"));
+					StringBuilder chain = new StringBuilder(entityGraphVariable);
 					for (int i = 0; i < pathComponents.length; i++) {
 
 						if (i < pathComponents.length - 1) {
@@ -587,11 +586,10 @@ private CodeBlock applyEntityGraph(AotEntityGraph entityGraph, String queryVaria
 
 					builder.addStatement(chain.toString(), (Object[]) pathComponents);
 				}
-
-				builder.addStatement("$L.setHint($S, $L)", queryVariableName, entityGraph.type().getKey(),
-						context.localVariable("entityGraph"));
 			}
 
+			builder.addStatement("$L.setHint($S, $L)", queryVariableName, entityGraph.type().getKey(), entityGraphVariable);
+
 			return builder.build();
 		}
 

spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/aot/JpaRepositoryContributorIntegrationTests.java

@@ -637,13 +637,12 @@ void shouldApplyQueryHints() {
 				.withMessageContaining("No enum constant jakarta.persistence.CacheStoreMode.foo");
 	}
 
-	@Test // GH-3830
+	@Test // GH-3830, GH-4097
 	void shouldApplyNamedEntityGraph() {
 
 		User chewie = fragment.findWithNamedEntityGraphByFirstname("Chewbacca");
 
-		assertThat(chewie.getManager()).isInstanceOf(HibernateProxy.class);
-		assertThat(chewie.getRoles()).isNotInstanceOf(HibernateProxy.class);
+		assertThat(chewie.getManager()).isNotInstanceOf(HibernateProxy.class);
 	}
 
 	@Test // GH-3830

spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/aot/UserRepository.java

@@ -246,7 +246,7 @@ List<User> findWithParameterNameByLastnameStartingWithOrLastnameEndingWith(@Para
 	@QueryHints(value = { @QueryHint(name = "jakarta.persistence.cache.storeMode", value = "foo") }, forCounting = false)
 	List<User> findHintedByLastname(String lastname);
 
-	@EntityGraph(type = EntityGraph.EntityGraphType.FETCH, value = "User.overview")
+	@EntityGraph(type = EntityGraph.EntityGraphType.FETCH, value = "User.detail")
 	User findWithNamedEntityGraphByFirstname(String firstname);
 
 	@EntityGraph(type = EntityGraph.EntityGraphType.FETCH, attributePaths = { "roles", "manager.roles" })