Skip to content

Commit 3545c45

Browse files
committed
Add additional types for mapping support.
1 parent 8bcc3d1 commit 3545c45

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/data/neo4j/Neo4jDataAutoConfiguration.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
import org.springframework.context.annotation.Bean;
3737
import org.springframework.context.annotation.Configuration;
3838
import org.springframework.context.annotation.Import;
39+
import org.springframework.data.annotation.Persistent;
3940
import org.springframework.data.neo4j.config.Neo4jDefaultCallbacksRegistrar;
4041
import org.springframework.data.neo4j.core.DatabaseSelectionProvider;
4142
import org.springframework.data.neo4j.core.Neo4jClient;
@@ -44,6 +45,7 @@
4445
import org.springframework.data.neo4j.core.convert.Neo4jConversions;
4546
import org.springframework.data.neo4j.core.mapping.Neo4jMappingContext;
4647
import org.springframework.data.neo4j.core.schema.Node;
48+
import org.springframework.data.neo4j.core.schema.RelationshipProperties;
4749
import org.springframework.data.neo4j.core.transaction.Neo4jTransactionManager;
4850
import org.springframework.data.neo4j.repository.config.Neo4jRepositoryConfigurationExtension;
4951
import org.springframework.transaction.PlatformTransactionManager;
@@ -79,7 +81,8 @@ public Neo4jConversions neo4jConversions() {
7981
@ConditionalOnMissingBean
8082
public Neo4jMappingContext neo4jMappingContext(ApplicationContext applicationContext,
8183
Neo4jConversions neo4jConversions) throws ClassNotFoundException {
82-
Set<Class<?>> initialEntityClasses = new EntityScanner(applicationContext).scan(Node.class);
84+
Set<Class<?>> initialEntityClasses = new EntityScanner(applicationContext).scan(Node.class, Persistent.class,
85+
RelationshipProperties.class);
8386
Neo4jMappingContext context = new Neo4jMappingContext(neo4jConversions);
8487
context.setInitialEntitySet(initialEntityClasses);
8588
return context;

0 commit comments

Comments
 (0)