SERVER-59342 Add killSessions and failover support to update shard key fsm testing

1 files changed, 10 insertions, 0 deletions

diff --git a/src/mongo/db/transaction_participant.cpp b/src/mongo/db/transaction_participant.cpp
index ae2a12fe2e7..f932c572fbf 100644
--- a/src/mongo/db/transaction_participant.cpp
+++ b/src/mongo/db/transaction_participant.cpp
@@ -710,6 +710,16 @@ void TransactionParticipant::Participant::_beginOrContinueRetryableWrite(
         retryableWriteTxnParticipantCatalog.addParticipant(*this);
     } else {
         // Retrying a retryable write.
+
+        // If this retryable write's transaction id has been converted to a transaction, and that
+        // transaction is in prepare, wait for it to exit prepare before throwing
+        // IncompleteTransactionHistory so the error response's operationTime is inclusive of the
+        // transaction's 2PC decision, guaranteeing causally consistent sessions will always read
+        // the transaction's writes.
+        uassert(ErrorCodes::PreparedTransactionInProgress,
+                "Retryable write that has been converted to a transaction is in prepare",
+                !o().txnState.isInSet(TransactionState::kPrepared));
+
         uassert(ErrorCodes::IncompleteTransactionHistory,
                 "Cannot retry a retryable write that has been converted into a transaction",
                 o().txnState.isInRetryableWriteMode());