From cc696b9fa0d7e1a16c3b5723cc2b0d2571b98fd3 Mon Sep 17 00:00:00 2001 From: Gerrit Meier Date: Wed, 30 Aug 2023 18:09:18 +0200 Subject: [PATCH] GH-2781 - Align mappingContext assignement in ReactiveNeo4jAuditingRegistrar. original change was introduced in 2b06d452 (Add AOT support) Closes #2781 --- .../neo4j/config/ReactiveNeo4jAuditingRegistrar.java | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/main/java/org/springframework/data/neo4j/config/ReactiveNeo4jAuditingRegistrar.java b/src/main/java/org/springframework/data/neo4j/config/ReactiveNeo4jAuditingRegistrar.java index 5f9936eb4..d0be9852f 100644 --- a/src/main/java/org/springframework/data/neo4j/config/ReactiveNeo4jAuditingRegistrar.java +++ b/src/main/java/org/springframework/data/neo4j/config/ReactiveNeo4jAuditingRegistrar.java @@ -24,7 +24,6 @@ import org.springframework.data.auditing.ReactiveIsNewAwareAuditingHandler; import org.springframework.data.auditing.config.AuditingBeanDefinitionRegistrarSupport; import org.springframework.data.auditing.config.AuditingConfiguration; import org.springframework.data.config.ParsingUtils; -import org.springframework.data.mapping.context.PersistentEntities; import org.springframework.data.neo4j.core.mapping.callback.ReactiveAuditingBeforeBindCallback; import org.springframework.util.Assert; @@ -86,11 +85,11 @@ final class ReactiveNeo4jAuditingRegistrar extends AuditingBeanDefinitionRegistr BeanDefinitionBuilder builder = BeanDefinitionBuilder.rootBeanDefinition(ReactiveIsNewAwareAuditingHandler.class); - BeanDefinitionBuilder persistentEntities = BeanDefinitionBuilder.genericBeanDefinition(PersistentEntities.class) - .setFactoryMethod("of"); - persistentEntities.addConstructorArgReference(MAPPING_CONTEXT_BEAN_NAME); - - builder.addConstructorArgValue(persistentEntities.getBeanDefinition()); return configureDefaultAuditHandlerAttributes(configuration, builder); } + + @Override + public void postProcess(BeanDefinitionBuilder builder, AuditingConfiguration configuration, BeanDefinitionRegistry registry) { + builder.setFactoryMethod("from").addConstructorArgReference(MAPPING_CONTEXT_BEAN_NAME); + } }