SERVER-43941: Add "errorLabels" field to failCommand failpoint

1 files changed, 15 insertions, 3 deletions

diff --git a/src/mongo/db/error_labels.cpp b/src/mongo/db/error_labels.cpp
index d56ba6dd82d..383f4f77551 100644
--- a/src/mongo/db/error_labels.cpp
+++ b/src/mongo/db/error_labels.cpp
@@ -28,6 +28,7 @@
  */
 
 #include "mongo/db/error_labels.h"
+#include "mongo/db/commands.h"
 
 namespace mongo {
 
@@ -94,17 +95,28 @@ bool ErrorLabelBuilder::_isCommitOrAbort() const {
         _commandName == "abortTransaction";
 }
 
-BSONObj getErrorLabels(const OperationSessionInfoFromClient& sessionOptions,
+BSONObj getErrorLabels(OperationContext* opCtx,
+                       const OperationSessionInfoFromClient& sessionOptions,
                        const std::string& commandName,
                        boost::optional<ErrorCodes::Error> code,
                        boost::optional<ErrorCodes::Error> wcCode,
                        bool isInternalClient) {
-    BSONArrayBuilder labelArray;
+    if (MONGO_unlikely(errorLabelsOverride(opCtx))) {
+        // This command was failed by a failCommand failpoint. Thus, we return the errorLabels
+        // specified in the failpoint to supress any other error labels that would otherwise be
+        // returned by the ErrorLabelBuilder.
+        if (errorLabelsOverride(opCtx).get().isEmpty()) {
+            return BSONObj();
+        } else {
+            return BSON(kErrorLabelsFieldName << errorLabelsOverride(opCtx).get());
+        }
+    }
 
+    BSONArrayBuilder labelArray;
     ErrorLabelBuilder labelBuilder(sessionOptions, commandName, code, wcCode, isInternalClient);
     labelBuilder.build(labelArray);
 
-    return (labelArray.arrSize() > 0) ? BSON("errorLabels" << labelArray.arr()) : BSONObj();
+    return (labelArray.arrSize() > 0) ? BSON(kErrorLabelsFieldName << labelArray.arr()) : BSONObj();
 }
 
 bool isTransientTransactionError(ErrorCodes::Error code,