Allow customizing order for AuditingEntityCallback and ValidatingEntityCallback.

Added corresponding test cases for this functionality.

Closes #4914
Signed-off-by: yangchef1 <[email protected]>

Author: yangchef1

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

@@ -32,6 +32,7 @@
 public class AuditingEntityCallback implements BeforeConvertCallback<Object>, Ordered {
 
 	private final ObjectFactory<IsNewAwareAuditingHandler> auditingHandlerFactory;
+	private int order = 100;
 
 	/**
 	 * Creates a new {@link AuditingEntityCallback} using the given {@link MappingContext} and {@link AuditingHandler}
@@ -52,6 +53,10 @@ public Object onBeforeConvert(Object entity, String collection) {
 
 	@Override
 	public int getOrder() {
-		return 100;
+		return this.order;
+	}
+
+	public void setOrder(int order) {
+		this.order = order;
 	}
 }

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

@@ -38,6 +38,7 @@
 public class ValidatingEntityCallback implements BeforeSaveCallback<Object>, Ordered {
 
 	private final BeanValidationDelegate delegate;
+	private int order = 100;
 
 	/**
 	 * Creates a new {@link ValidatingEntityCallback} using the given {@link Validator}.
@@ -62,7 +63,10 @@ public Object onBeforeSave(Object entity, Document document, String collection)
 
 	@Override
 	public int getOrder() {
-		return 100;
+		return this.order;
 	}
 
+	public void setOrder(int order) {
+		this.order = order;
+	}
 }

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/mapping/event/AuditingEntityCallbackUnitTests.java

@@ -92,6 +92,16 @@ void hasExplicitOrder() {
 		assertThat(callback.getOrder()).isEqualTo(100);
 	}
 
+	@Test // GH-4914
+	void allowsChangingOrderDynamically() {
+
+		assertThat(callback).isInstanceOf(Ordered.class);
+		assertThat(callback.getOrder()).isEqualTo(100);
+
+		callback.setOrder(50);
+		assertThat(callback.getOrder()).isEqualTo(50);
+	}
+
 	@Test // DATAMONGO-2261
 	void propagatesChangedInstanceToEvent() {
 

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/mapping/event/ValidatingEntityCallbackUnitTests.java

@@ -26,6 +26,7 @@
 import org.bson.Document;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.springframework.core.Ordered;
 
 /**
  * Unit tests for {@link ValidatingEntityCallback}.
@@ -63,6 +64,16 @@ void validateSuccessful() {
 		assertThat(entity).isEqualTo(coordinates);
 	}
 
+	@Test // GH-4914
+	void allowsChangingOrderDynamically() {
+
+		assertThat(callback).isInstanceOf(Ordered.class);
+		assertThat(callback.getOrder()).isEqualTo(100);
+
+		callback.setOrder(50);
+		assertThat(callback.getOrder()).isEqualTo(50);
+	}
+
 	record Coordinates(@NotNull @Min(0) Integer x, @NotNull @Min(0) Integer y) {
 
 		Document toDocument() {