Translate transient WriteConflict to ConcurrencyFailureException

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

@@ -20,6 +20,7 @@
 import com.mongodb.ClientBulkWriteException;
 import org.bson.BsonInvalidOperationException;
 import org.jspecify.annotations.Nullable;
+import org.springframework.dao.ConcurrencyFailureException;
 import org.springframework.dao.DataAccessException;
 import org.springframework.dao.DataAccessResourceFailureException;
 import org.springframework.dao.DataIntegrityViolationException;
@@ -50,6 +51,7 @@
  * @author Michal Vich
  * @author Christoph Strobl
  * @author Brice Vandeputte
+ * @author Seonwoo Jung
  */
 public class MongoExceptionTranslator implements PersistenceExceptionTranslator {
 
@@ -120,6 +122,10 @@ DataAccessException doTranslateException(RuntimeException ex) {
 				}
 			}
 
+			if (isTransientFailure(ex)) {
+				return new ConcurrencyFailureException(ex.getMessage(), ex);
+			}
+
 			return new DataIntegrityViolationException(ex.getMessage(), ex);
 		}
 
@@ -142,6 +148,9 @@ DataAccessException doTranslateException(RuntimeException ex) {
 				return new PermissionDeniedDataAccessException(ex.getMessage(), ex);
 			}
 			if (MongoDbErrorCodes.isDataIntegrityViolationError(mongoException)) {
+				if (isTransientFailure(mongoException)) {
+					return new ConcurrencyFailureException(mongoException.getMessage(), mongoException);
+				}
 				return new DataIntegrityViolationException(mongoException.getMessage(), mongoException);
 			}
 			if (MongoDbErrorCodes.isClientSessionFailure(mongoException)) {

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

@@ -24,6 +24,7 @@
 import org.mockito.Mockito;
 
 import org.springframework.core.NestedRuntimeException;
+import org.springframework.dao.ConcurrencyFailureException;
 import org.springframework.dao.DataAccessException;
 import org.springframework.dao.DataAccessResourceFailureException;
 import org.springframework.dao.DuplicateKeyException;
@@ -39,7 +40,9 @@
 import com.mongodb.MongoSocketException;
 import com.mongodb.MongoSocketReadTimeoutException;
 import com.mongodb.MongoSocketWriteException;
+import com.mongodb.MongoWriteException;
 import com.mongodb.ServerAddress;
+import com.mongodb.WriteError;
 
 /**
  * Unit tests for {@link MongoExceptionTranslator}.
@@ -48,6 +51,7 @@
  * @author Oliver Gierke
  * @author Christoph Strobl
  * @author Brice Vandeputte
+ * @author Seonwoo Jung
  */
 class MongoExceptionTranslatorUnitTests {
 
@@ -193,6 +197,42 @@ public void translateMongoExceptionWithTransientLabel() {
 		expectExceptionWithCauseMessage(translatedException, UncategorizedMongoDbException.class);
 	}
 
+	@Test // GH-5150
+	void translateTransientWriteConflictMongoException() {
+
+		MongoException source = new MongoException(112, "WriteConflict");
+		source.addLabel(MongoException.TRANSIENT_TRANSACTION_ERROR_LABEL);
+
+		DataAccessException translated = translator.translateExceptionIfPossible(source);
+
+		expectExceptionWithCauseMessage(translated, ConcurrencyFailureException.class, "WriteConflict");
+		assertThat(translator.isTransientFailure(translated)).isTrue();
+	}
+
+	@Test // GH-5150
+	void translateTransientMongoWriteException() {
+
+		WriteError writeError = new WriteError(112, "WriteConflict", new BsonDocument());
+		MongoWriteException source = new MongoWriteException(writeError, new ServerAddress());
+		source.addLabel(MongoException.TRANSIENT_TRANSACTION_ERROR_LABEL);
+
+		DataAccessException translated = translator.translateExceptionIfPossible(source);
+
+		expectExceptionWithCauseMessage(translated, ConcurrencyFailureException.class, "WriteConflict");
+		assertThat(translator.isTransientFailure(translated)).isTrue();
+	}
+
+	@Test // GH-5150
+	void translateNonTransientWriteConflictMongoException() {
+
+		MongoException source = new MongoException(112, "WriteConflict");
+
+		DataAccessException translated = translator.translateExceptionIfPossible(source);
+
+		expectExceptionWithCauseMessage(translated,
+				org.springframework.dao.DataIntegrityViolationException.class, "WriteConflict");
+	}
+
 	private void checkTranslatedMongoException(Class<? extends Exception> clazz, int code) {
 
 		DataAccessException translated = translator.translateExceptionIfPossible(new MongoException(code, ""));