GH-2305 - Add association support for DTO projections.

Author: meistermeier

src/main/java/org/springframework/data/neo4j/core/mapping/DtoInstantiatingConverter.java

@@ -33,6 +33,7 @@
 import org.springframework.data.mapping.PersistentPropertyAccessor;
 import org.springframework.data.mapping.PreferredConstructor;
 import org.springframework.data.mapping.PreferredConstructor.Parameter;
+import org.springframework.data.mapping.SimpleAssociationHandler;
 import org.springframework.data.mapping.SimplePropertyHandler;
 import org.springframework.data.mapping.model.ParameterValueProvider;
 import org.springframework.data.util.ClassTypeInformation;
@@ -105,19 +106,29 @@ public Object getParameterValue(Parameter parameter) {
 				});
 
 		PersistentPropertyAccessor<Object> dtoAccessor = targetEntity.getPropertyAccessor(dto);
-		targetEntity.doWithProperties((SimplePropertyHandler) property -> {
+		targetEntity.doWithProperties((SimplePropertyHandler) property ->
+				setPropertyOnDtoObject(entityInstanceAndSource, sourceEntity, sourceAccessor, constructor, dtoAccessor, property));
 
-			if (constructor.isConstructorParameter(property)) {
-				return;
-			}
-
-			Object propertyValue = getPropertyValueFor(property, sourceEntity, sourceAccessor, entityInstanceAndSource);
-			dtoAccessor.setProperty(property, propertyValue);
+		targetEntity.doWithAssociations((SimpleAssociationHandler) association -> {
+			PersistentProperty<?> property = association.getInverse();
+			setPropertyOnDtoObject(entityInstanceAndSource, sourceEntity, sourceAccessor, constructor, dtoAccessor, property);
 		});
 
 		return dto;
 	}
 
+	private void setPropertyOnDtoObject(EntityInstanceWithSource entityInstanceAndSource, PersistentEntity<?, ?> sourceEntity,
+			PersistentPropertyAccessor<Object> sourceAccessor, PreferredConstructor<?, ?> constructor,
+			PersistentPropertyAccessor<Object> dtoAccessor, PersistentProperty<?> property) {
+
+		if (constructor.isConstructorParameter(property)) {
+			return;
+		}
+
+		Object propertyValue = getPropertyValueFor(property, sourceEntity, sourceAccessor, entityInstanceAndSource);
+		dtoAccessor.setProperty(property, propertyValue);
+	}
+
 	@Nullable
 	Object getPropertyValueFor(PersistentProperty<?> targetProperty, PersistentEntity<?, ?> sourceEntity,
 			PersistentPropertyAccessor sourceAccessor, EntityInstanceWithSource entityInstanceAndSource) {

src/test/java/org/springframework/data/neo4j/integration/imperative/RepositoryIT.java

@@ -1301,9 +1301,9 @@ private void createOneToOneScenarioForNullValues() {
 			);
 		}
 
-		private void assertOneToOneScenarioWithNulls(List<OneToOneSource> oneToOnes) {
+		private void assertOneToOneScenarioWithNulls(List<OneToOneSource.OneToOneSourceProjection> oneToOnes) {
 			assertThat(oneToOnes).hasSize(2);
-			assertThat(oneToOnes).extracting(OneToOneSource::getName).containsExactlyInAnyOrder("s1", "s2");
+			assertThat(oneToOnes).extracting(OneToOneSource.OneToOneSourceProjection::getName).containsExactlyInAnyOrder("s1", "s2");
 			assertThat(oneToOnes).filteredOn(s -> s.getTarget() != null)
 					.extracting(s -> s.getTarget().getName()).containsExactly("t1");
 		}
@@ -1312,7 +1312,7 @@ private void assertOneToOneScenarioWithNulls(List<OneToOneSource> oneToOnes) {
 		void shouldFindOneToOneWithNullValues(@Autowired OneToOneRepository repository)  {
 			createOneToOneScenarioForNullValues();
 
-			List<OneToOneSource> oneToOnes = repository.findAllWithNullValues();
+			List<OneToOneSource.OneToOneSourceProjection> oneToOnes = repository.findAllWithNullValues();
 			assertOneToOneScenarioWithNulls(oneToOnes);
 		}
 	}
@@ -4146,8 +4146,8 @@ interface OneToOneRepository extends Neo4jRepository<OneToOneSource, String> {
 		@Query("MATCH (p1:#{#staticLabels})-[r:OWNS]-(p2) return *")
 		List<OneToOneSource> findAllWithCustomQueryReturnStar();
 
-		@Query("MATCH (p1:#{#staticLabels}) OPTIONAL MATCH (p1)-[r:OWNS]->(p2) return p1, r, p2")
-		List<OneToOneSource> findAllWithNullValues();
+		@Query("MATCH (p1:#{#staticLabels}) OPTIONAL MATCH (p1)-[r:OWNS]->(p2:OneToOneTarget) return p1, r, p2")
+		List<OneToOneSource.OneToOneSourceProjection> findAllWithNullValues();
 	}
 
 	interface RelationshipRepository extends Neo4jRepository<PersonWithRelationship, Long> {

src/test/java/org/springframework/data/neo4j/integration/shared/common/OneToOneSource.java

@@ -35,4 +35,13 @@ public class OneToOneSource {
 
 	@Relationship("OWNS")
 	private OneToOneTarget target;
+
+	/**
+	 * Simple DTO projection for OneToOneSource
+	 */
+	@Data
+	public static class OneToOneSourceProjection {
+		String name;
+		OneToOneTarget target;
+	}
 }