SERVER-21093: Make ordered bulk insert operation error out properly

2 files changed, 10 insertions, 4 deletions

diff --git a/src/mongo/db/commands/write_commands/batch_executor.cpp b/src/mongo/db/commands/write_commands/batch_executor.cpp
index 753b6917b3d..184807edea6 100644
--- a/src/mongo/db/commands/write_commands/batch_executor.cpp
+++ b/src/mongo/db/commands/write_commands/batch_executor.cpp
@@ -753,7 +753,8 @@ static void normalizeInserts(const BatchedCommandRequest& request,
 static void insertOne(WriteBatchExecutor::ExecInsertsState* state, WriteOpResult* result);
 
 // Loops over the specified subset of the batch, processes one document at a time.
-void WriteBatchExecutor::insertMany(WriteBatchExecutor::ExecInsertsState* state,
+// Returns a true to discontinue the insert, or false if not.
+bool WriteBatchExecutor::insertMany(WriteBatchExecutor::ExecInsertsState* state,
                                     size_t startIndex,
                                     size_t endIndex,
                                     CurOp* currentOp,
@@ -791,11 +792,12 @@ void WriteBatchExecutor::insertMany(WriteBatchExecutor::ExecInsertsState* state,
             CurOp* const currentOp = CurOp::get(_txn);
             logCurOpError(currentOp, error);
             if (ordered)
-                break;
+                return true;
         } else {
             _le->recordInsert(nInserted);
         }
     }
+    return false;
 }
 
 // Instantiates an ExecInsertsState, which represents all of the state for the batch.
@@ -832,7 +834,8 @@ void WriteBatchExecutor::execInserts(const BatchedCommandRequest& request,
 
         if ((chunkCount >= chunkMaxCount) || (chunkBytes >= insertVectorMaxBytes) ||
             (i == maxIndex)) {
-            insertMany(&state, startIndex, i + 1, currentOp, errors, request.getOrdered());
+            bool stop;
+            stop = insertMany(&state, startIndex, i + 1, currentOp, errors, request.getOrdered());
             startIndex = i + 1;
             chunkCount = 0;
             chunkBytes = 0;
@@ -846,6 +849,8 @@ void WriteBatchExecutor::execInserts(const BatchedCommandRequest& request,
                 // Since the lock manager guarantees FIFO queues waiting on locks,
                 // there is no need to explicitly sleep or give up control of the processor here.
             }
+            if (stop)
+                break;
         }
     }
 }
diff --git a/src/mongo/db/commands/write_commands/batch_executor.h b/src/mongo/db/commands/write_commands/batch_executor.h
index 78ab8028449..945adfca021 100644
--- a/src/mongo/db/commands/write_commands/batch_executor.h
+++ b/src/mongo/db/commands/write_commands/batch_executor.h
@@ -85,8 +85,9 @@ private:
 
     /**
      * Inserts a subset of an insert batch.
+     * Returns a true to discontinue the insert, or false if not.
      */
-    void insertMany(WriteBatchExecutor::ExecInsertsState* state,
+    bool insertMany(WriteBatchExecutor::ExecInsertsState* state,
                     size_t startIndex,
                     size_t endIndex,
                     CurOp* currentOp,