|
36 | 36 | import org.springframework.context.annotation.Bean;
|
37 | 37 | import org.springframework.context.annotation.Configuration;
|
38 | 38 | import org.springframework.context.annotation.Import;
|
| 39 | +import org.springframework.data.annotation.Persistent; |
39 | 40 | import org.springframework.data.neo4j.config.Neo4jDefaultCallbacksRegistrar;
|
40 | 41 | import org.springframework.data.neo4j.core.DatabaseSelectionProvider;
|
41 | 42 | import org.springframework.data.neo4j.core.Neo4jClient;
|
|
44 | 45 | import org.springframework.data.neo4j.core.convert.Neo4jConversions;
|
45 | 46 | import org.springframework.data.neo4j.core.mapping.Neo4jMappingContext;
|
46 | 47 | import org.springframework.data.neo4j.core.schema.Node;
|
| 48 | +import org.springframework.data.neo4j.core.schema.RelationshipProperties; |
47 | 49 | import org.springframework.data.neo4j.core.transaction.Neo4jTransactionManager;
|
48 | 50 | import org.springframework.data.neo4j.repository.config.Neo4jRepositoryConfigurationExtension;
|
49 | 51 | import org.springframework.transaction.PlatformTransactionManager;
|
@@ -79,7 +81,8 @@ public Neo4jConversions neo4jConversions() {
|
79 | 81 | @ConditionalOnMissingBean
|
80 | 82 | public Neo4jMappingContext neo4jMappingContext(ApplicationContext applicationContext,
|
81 | 83 | 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); |
83 | 86 | Neo4jMappingContext context = new Neo4jMappingContext(neo4jConversions);
|
84 | 87 | context.setInitialEntitySet(initialEntityClasses);
|
85 | 88 | return context;
|
|
0 commit comments