Add top-level error labels to write concern error (mongodb#766)

This will allow the driver to properly retry writes that the server has labeled as a RetryableWriteError. JAVA-4244
jyemin · Jul 27, 2021 · 81cc1a7 · 81cc1a7
1 parent c3c0a2c
commit 81cc1a7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/driver-core/src/main/com/mongodb/internal/connection/ProtocolHelper.java b/driver-core/src/main/com/mongodb/internal/connection/ProtocolHelper.java
@@ -243,7 +243,14 @@ public static MongoException createSpecialException(final BsonDocument response,
         } else if (errorMessage.contains("not master") || NOT_MASTER_CODES.contains(errorCode)) {
             return new MongoNotPrimaryException(response, serverAddress);
         } else if (response.containsKey("writeConcernError")) {
-            return createSpecialException(response.getDocument("writeConcernError"), serverAddress, "errmsg");
+            MongoException writeConcernException = createSpecialException(response.getDocument("writeConcernError"), serverAddress,
+                    "errmsg");
+            if (writeConcernException != null && response.isArray("errorLabels")) {
+                for (BsonValue errorLabel : response.getArray("errorLabels")) {
+                    writeConcernException.addLabel(errorLabel.asString().getValue());
+                }
+            }
+            return writeConcernException;
         } else {
             return null;
         }