SERVER-28918 Make CRUD paths retryable

1 files changed, 17 insertions, 0 deletions

diff --git a/src/mongo/db/session.cpp b/src/mongo/db/session.cpp
index 844f80f81d1..0811996eb8f 100644
--- a/src/mongo/db/session.cpp
+++ b/src/mongo/db/session.cpp
@@ -179,4 +179,21 @@ TransactionHistoryIterator Session::getWriteHistory(OperationContext* opCtx) con
     return TransactionHistoryIterator(getLastWriteOpTimeTs());
 }
 
+boost::optional<repl::OplogEntry> Session::checkStatementExecuted(OperationContext* opCtx,
+                                                                  StmtId stmtId) {
+    if (!opCtx->getTxnNumber()) {
+        return boost::none;
+    }
+
+    auto it = getWriteHistory(opCtx);
+    while (it.hasNext()) {
+        auto entry = it.next(opCtx);
+        if (entry.getStatementId() == stmtId) {
+            return entry;
+        }
+    }
+
+    return boost::none;
+}
+
 }  // namespace mongo