1 files changed, 12 insertions, 0 deletions
diff --git a/src/mongo/db/commands/bulk_write_common.cpp b/src/mongo/db/commands/bulk_write_common.cpp
index 691932dfc17..2b39688dc1b 100644
--- a/src/mongo/db/commands/bulk_write_common.cpp
+++ b/src/mongo/db/commands/bulk_write_common.cpp
@@ -129,5 +129,17 @@ std::vector<Privilege> getPrivileges(const BulkWriteCommandRequest& req) {
     return privileges;
 }
 
+int32_t getStatementId(const BulkWriteCommandRequest& req, size_t currentOpIdx) {
+    auto stmtId = req.getStmtId();
+    auto stmtIds = req.getStmtIds();
+
+    if (stmtIds) {
+        return stmtIds->at(currentOpIdx);
+    }
+
+    int32_t firstStmtId = stmtId ? *stmtId : 0;
+    return firstStmtId + currentOpIdx;
+}
+
 }  // namespace bulk_write_common
 }  // namespace mongo