Extract Wrapped support to PersistentEntity.

Author: mp911de

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/convert/DefaultDbRefProxyHandler.java

@@ -19,7 +19,8 @@
 
 import org.bson.Document;
 import org.jspecify.annotations.Nullable;
-import org.springframework.data.mapping.PersistentPropertyAccessor;
+
+import org.springframework.beans.DirectFieldAccessor;
 import org.springframework.data.mapping.context.MappingContext;
 import org.springframework.data.mapping.model.ValueExpressionEvaluator;
 import org.springframework.data.mongodb.core.mapping.MongoPersistentEntity;
@@ -65,11 +66,15 @@ public Object populateId(MongoPersistentProperty property, @Nullable DBRef sourc
 		}
 
 		ValueExpressionEvaluator evaluator = evaluatorFactory.apply(proxy);
-		PersistentPropertyAccessor accessor = entity.getPropertyAccessor(proxy);
+
+		DirectFieldAccessor accessor = new DirectFieldAccessor(proxy);
+		// TODO
+		// PersistentPropertyAccessor accessor = entity.getPropertyAccessor(proxy);
 
 		Document object = new Document(idProperty.getFieldName(), source.getId());
 		ObjectPath objectPath = ObjectPath.ROOT.push(proxy, entity, null);
-		accessor.setProperty(idProperty, resolver.getValueInternal(idProperty, object, evaluator, objectPath));
+		accessor.setPropertyValue(idProperty.getName(),
+				resolver.getValueInternal(idProperty, object, evaluator, objectPath));
 
 		return proxy;
 	}

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/convert/LazyLoadingProxy.java

@@ -17,6 +17,8 @@
 
 import org.jspecify.annotations.Nullable;
 
+import org.springframework.data.mongodb.core.mapping.Wrapped;
+
 import com.mongodb.DBRef;
 
 /**
@@ -28,14 +30,15 @@
  * @since 1.5
  * @see LazyLoadingProxyFactory
  */
-public interface LazyLoadingProxy {
+public interface LazyLoadingProxy extends Wrapped {
 
 	/**
 	 * Initializes the proxy and returns the wrapped value.
 	 *
 	 * @return
 	 * @since 1.5
 	 */
+	@Override
 	Object getTarget();
 
 	/**

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/mapping/BasicMongoPersistentEntity.java

@@ -26,13 +26,15 @@
 import java.util.Map;
 
 import org.jspecify.annotations.Nullable;
+
 import org.springframework.data.annotation.Id;
 import org.springframework.data.expression.ValueEvaluationContext;
 import org.springframework.data.expression.ValueExpression;
 import org.springframework.data.expression.ValueExpressionParser;
 import org.springframework.data.mapping.Association;
 import org.springframework.data.mapping.AssociationHandler;
 import org.springframework.data.mapping.MappingException;
+import org.springframework.data.mapping.PersistentPropertyAccessor;
 import org.springframework.data.mapping.PropertyHandler;
 import org.springframework.data.mapping.model.BasicPersistentEntity;
 import org.springframework.data.mongodb.MongoCollectionUtils;
@@ -188,6 +190,16 @@ public void verify() {
 		verifyFieldTypes();
 	}
 
+	@Override
+	public boolean isNew(Object bean) {
+		return super.isNew(Wrapped.getTargetObject(bean));
+	}
+
+	@Override
+	public <B> PersistentPropertyAccessor<B> getPropertyAccessor(B bean) {
+		return super.getPropertyAccessor(Wrapped.getTargetObject(bean));
+	}
+
 	@Override
 	public EvaluationContext getEvaluationContext(@Nullable Object rootObject) {
 		return super.getEvaluationContext(rootObject);

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/mapping/Wrapped.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright 2025 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.mongodb.core.mapping;
+
+/**
+ * @author Mark Paluch
+ */
+public interface Wrapped {
+
+	/**
+	 * Returns the wrapped value.
+	 *
+	 * @return
+	 */
+	Object getTarget();
+
+	/**
+	 * Unwraps the given object if it is an instance of {@link Wrapped}. If not, returns the object as-is.
+	 *
+	 * @param object the object to unwrap, must not be {@literal null}.
+	 * @return the unwrapped object or the original object if it is not wrapped.
+	 */
+	static <T> T getTargetObject(T object) {
+		return object instanceof Wrapped w ? (T) w.getTarget() : object;
+	}
+
+}

spring-data-mongodb/src/main/java/org/springframework/data/mongodb/repository/support/MappingMongoEntityInformation.java

@@ -17,8 +17,8 @@
 
 import org.bson.types.ObjectId;
 import org.jspecify.annotations.Nullable;
+
 import org.springframework.data.mapping.PersistentPropertyAccessor;
-import org.springframework.data.mongodb.core.convert.LazyLoadingProxy;
 import org.springframework.data.mongodb.core.mapping.MongoPersistentEntity;
 import org.springframework.data.mongodb.core.query.Collation;
 import org.springframework.data.mongodb.repository.query.MongoEntityInformation;
@@ -108,13 +108,6 @@ public Class<ID> getIdType() {
 		return fallbackIdType;
 	}
 
-	@Override
-	public boolean isNew(T entity) {
-
-		T unwrapped = entity instanceof LazyLoadingProxy proxy ? (T) proxy.getTarget() : entity;
-		return super.isNew(unwrapped);
-	}
-
 	@Override
 	public boolean isVersioned() {
 		return this.entityMetadata.hasVersionProperty();

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/config/AuditingIntegrationTests.java

@@ -19,31 +19,40 @@
 
 import java.util.Date;
 
+import org.bson.types.ObjectId;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
 
-import org.springframework.context.support.AbstractApplicationContext;
-import org.springframework.context.support.ClassPathXmlApplicationContext;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
 import org.springframework.data.annotation.CreatedDate;
 import org.springframework.data.annotation.Id;
 import org.springframework.data.annotation.LastModifiedDate;
 import org.springframework.data.mapping.callback.EntityCallbacks;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.mapping.DocumentReference;
 import org.springframework.data.mongodb.core.mapping.MongoMappingContext;
 import org.springframework.data.mongodb.core.mapping.event.BeforeConvertCallback;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
 
 /**
  * Integration test for the auditing support.
  *
  * @author Oliver Gierke
  * @author Mark Paluch
  */
-public class AuditingIntegrationTests {
+@ContextConfiguration(locations = "auditing.xml")
+@ExtendWith(SpringExtension.class)
+class AuditingIntegrationTests {
 
-	@Test // DATAMONGO-577, DATAMONGO-800, DATAMONGO-883, DATAMONGO-2261
-	public void enablesAuditingAndSetsPropertiesAccordingly() throws Exception {
+	@Autowired MongoMappingContext mappingContext;
+	@Autowired ApplicationContext context;
+	@Autowired MongoTemplate template;
 
-		AbstractApplicationContext context = new ClassPathXmlApplicationContext("auditing.xml", getClass());
+	@Test // DATAMONGO-577, DATAMONGO-800, DATAMONGO-883, DATAMONGO-2261
+	void enablesAuditingAndSetsPropertiesAccordingly() throws Exception {
 
-		MongoMappingContext mappingContext = context.getBean(MongoMappingContext.class);
 		mappingContext.getPersistentEntity(Entity.class);
 
 		EntityCallbacks callbacks = EntityCallbacks.create(context);
@@ -55,24 +64,68 @@ public void enablesAuditingAndSetsPropertiesAccordingly() throws Exception {
 		assertThat(entity.modified).isEqualTo(entity.created);
 
 		Thread.sleep(10);
-		entity.id = 1L;
+		entity.id = "foo";
 
 		entity = callbacks.callback(BeforeConvertCallback.class, entity, "collection-1");
 
 		assertThat(entity.created).isNotNull();
 		assertThat(entity.modified).isNotEqualTo(entity.created);
-		context.close();
 	}
 
-	class Entity {
+	@Test // GH-5031
+	void handlesAuditingCorrectly() throws InterruptedException {
+
+		template.remove(TopLevelEntity.class).all();
+
+		Entity entity = new Entity();
+		template.insert(entity);
+
+		TopLevelEntity tle = new TopLevelEntity();
+		tle.entity = entity;
+		template.insert(tle);
+
+		Thread.sleep(200);
+
+		TopLevelEntity loadAndModify = template.findById(tle.id, TopLevelEntity.class);
+		Date created = loadAndModify.entity.getCreated();
+		Date modified = loadAndModify.entity.getModified();
+		template.save(loadAndModify.entity);
+
+		TopLevelEntity loaded = template.findById(tle.id, TopLevelEntity.class);
+
+		assertThat(loaded.entity.getCreated()).isEqualTo(created);
+		assertThat(loaded.entity.getModified()).isNotEqualTo(modified);
+	}
+
+	static class Entity {
 
-		@Id Long id;
+		@Id String id;
 		@CreatedDate Date created;
 		Date modified;
 
 		@LastModifiedDate
 		public Date getModified() {
 			return modified;
 		}
+
+		public Date getCreated() {
+			return created;
+		}
+
+		public void setCreated(Date created) {
+			this.created = created;
+		}
+
+		public void setModified(Date modified) {
+			this.modified = modified;
+		}
 	}
+
+	static class TopLevelEntity {
+
+		@Id ObjectId id;
+		@DocumentReference(lazy = true) Entity entity;
+
+	}
+
 }

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/convert/DbRefMappingMongoConverterUnitTests.java

@@ -44,11 +44,11 @@
 import org.mockito.Mockito;
 import org.mockito.junit.jupiter.MockitoExtension;
 
+import org.springframework.beans.DirectFieldAccessor;
 import org.springframework.data.annotation.AccessType;
 import org.springframework.data.annotation.AccessType.Type;
 import org.springframework.data.annotation.Id;
 import org.springframework.data.annotation.PersistenceCreator;
-import org.springframework.data.mapping.PersistentPropertyAccessor;
 import org.springframework.data.mapping.PropertyPath;
 import org.springframework.data.mongodb.MongoDatabaseFactory;
 import org.springframework.data.mongodb.core.MongoExceptionTranslator;
@@ -487,11 +487,11 @@ void shouldEagerlyResolveIdPropertyWithFieldAccess() {
 
 		ClassWithLazyDbRefs result = converter.read(ClassWithLazyDbRefs.class, object);
 
-		PersistentPropertyAccessor accessor = propertyEntity.getPropertyAccessor(result.dbRefToConcreteType);
+		DirectFieldAccessor accessor = new DirectFieldAccessor(result.dbRefToConcreteType);
 		MongoPersistentProperty idProperty = mappingContext.getRequiredPersistentEntity(LazyDbRefTarget.class)
 				.getIdProperty();
 
-		assertThat(accessor.getProperty(idProperty)).isNotNull();
+		assertThat(accessor.getPropertyValue(idProperty.getName())).isNotNull();
 		assertProxyIsResolved(result.dbRefToConcreteType, false);
 	}
 

spring-data-mongodb/src/test/resources/org/springframework/data/mongodb/config/auditing.xml

@@ -5,8 +5,15 @@
 	xsi:schemaLocation="http://www.springframework.org/schema/data/mongo https://www.springframework.org/schema/data/mongo/spring-mongo.xsd
 		http://www.springframework.org/schema/beans https://www.springframework.org/schema/beans/spring-beans.xsd">
 
-	<mongo:auditing mapping-context-ref="customMappingContext" />
-	
-	<bean id="customMappingContext" class="org.springframework.data.mongodb.core.mapping.MongoMappingContext" />
+	<mongo:auditing mapping-context-ref="mongoMappingContext"/>
+
+	<mongo:db-factory dbname="repositories"/>
+
+	<mongo:mapping-converter auto-index-creation="true"/>
+
+	<bean id="mongoTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
+		<constructor-arg ref="mongoDbFactory"/>
+		<constructor-arg ref="mappingConverter"/>
+	</bean>
 
 </beans>