PYTHON-2984 Fix retry behavior for bulk write writeConcernError (#800)

(cherry picked from commit 2af521e)

Author: ShaneHarvey

doc/changelog.rst

@@ -1,6 +1,21 @@
 Changelog
 =========
 
+Changes in Version 3.12.2
+-------------------------
+
+Issues Resolved
+...............
+
+Version 3.12.2 fixes a bug that prevented PyMongo from retrying bulk writes
+after a ``writeConcernError`` on MongoDB 4.4+ (`PYTHON-2984`_).
+
+See the `PyMongo 3.12.2 release notes in JIRA`_ for the list of resolved issues
+in this release.
+
+.. _PYTHON-2984: https://jira.mongodb.org/browse/PYTHON-2984
+.. _PyMongo 3.12.2 release notes in JIRA:https://jira.mongodb.org/secure/ReleaseNote.jspa?projectId=10004&version=32310
+
 Changes in Version 3.12.1
 -------------------------
 

pymongo/bulk.py

@@ -28,7 +28,7 @@
                             validate_is_document_type,
                             validate_ok_for_replace,
                             validate_ok_for_update)
-from pymongo.helpers import _RETRYABLE_ERROR_CODES
+from pymongo.helpers import _RETRYABLE_ERROR_CODES, _get_wce_doc
 from pymongo.collation import validate_collation_or_none
 from pymongo.errors import (BulkWriteError,
                             ConfigurationError,
@@ -126,9 +126,9 @@ def _merge_command(run, full_result, offset, result):
             replacement[_UOP] = run.ops[idx]
             full_result["writeErrors"].append(replacement)
 
-    wc_error = result.get("writeConcernError")
-    if wc_error:
-        full_result["writeConcernErrors"].append(wc_error)
+    wce = _get_wce_doc(result)
+    if wce:
+        full_result["writeConcernErrors"].append(wce)
 
 
 def _raise_bulk_write_error(full_result):

pymongo/helpers.py

@@ -220,6 +220,18 @@ def _raise_write_concern_error(error):
         error.get("errmsg"), error.get("code"), error)
 
 
+def _get_wce_doc(result):
+    """Return the writeConcernError or None."""
+    wce = result.get("writeConcernError")
+    if wce:
+        # The server reports errorLabels at the top level but it's more
+        # convenient to attach it to the writeConcernError doc itself.
+        error_labels = result.get("errorLabels")
+        if error_labels:
+            wce["errorLabels"] = error_labels
+    return wce
+
+
 def _check_write_command_response(result):
     """Backward compatibility helper for write command error handling.
     """
@@ -228,12 +240,9 @@ def _check_write_command_response(result):
     if write_errors:
         _raise_last_write_error(write_errors)
 
-    error = result.get("writeConcernError")
-    if error:
-        error_labels = result.get("errorLabels")
-        if error_labels:
-            error.update({'errorLabels': error_labels})
-        _raise_write_concern_error(error)
+    wce = _get_wce_doc(result)
+    if wce:
+        _raise_write_concern_error(wce)
 
 
 def _raise_last_error(bulk_write_result):

test/retryable_writes/bulkWrite-errorLabels.json renamed to test/retryable_writes/legacy/bulkWrite-errorLabels.json

@@ -119,12 +119,12 @@
           "failCommands": [
             "insert"
           ],
+          "errorLabels": [
+            "RetryableWriteError"
+          ],
           "writeConcernError": {
             "code": 91,
-            "errmsg": "Replication is being shut down",
-            "errorLabels": [
-              "RetryableWriteError"
-            ]
+            "errmsg": "Replication is being shut down"
           }
         }
       },

test/retryable_writes/bulkWrite-serverErrors.json renamed to test/retryable_writes/legacy/bulkWrite-serverErrors.json

@@ -75,12 +75,12 @@
           "failCommands": [
             "delete"
           ],
+          "errorLabels": [
+            "RetryableWriteError"
+          ],
           "writeConcernError": {
             "code": 91,
-            "errmsg": "Replication is being shut down",
-            "errorLabels": [
-              "RetryableWriteError"
-            ]
+            "errmsg": "Replication is being shut down"
           }
         }
       },