SERVER-61121 Make TransactionMetricsObserver support TxnNumberAndRetryCounter

2 files changed, 8 insertions, 5 deletions

diff --git a/src/mongo/db/stats/single_transaction_stats.cpp b/src/mongo/db/stats/single_transaction_stats.cpp
index 386a2568d45..3234998fb28 100644
--- a/src/mongo/db/stats/single_transaction_stats.cpp
+++ b/src/mongo/db/stats/single_transaction_stats.cpp
@@ -116,7 +116,7 @@ void SingleTransactionStats::report(BSONObjBuilder* builder,
                                     TickSource* tickSource,
                                     TickSource::Tick curTick) const {
     BSONObjBuilder parametersBuilder(builder->subobjStart("parameters"));
-    parametersBuilder.append("txnNumber", _txnNumber);
+    parametersBuilder.append("txnNumber", _txnNumberAndRetryCounter.getTxnNumber());
 
     if (!isForMultiDocumentTransaction()) {
         // For retryable writes, we only include the txnNumber.
@@ -124,6 +124,7 @@ void SingleTransactionStats::report(BSONObjBuilder* builder,
         return;
     }
 
+    parametersBuilder.append("txnRetryCounter", *_txnNumberAndRetryCounter.getTxnRetryCounter());
     parametersBuilder.append("autocommit", *_autoCommit);
     readConcernArgs.appendInfo(&parametersBuilder);
     parametersBuilder.done();
diff --git a/src/mongo/db/stats/single_transaction_stats.h b/src/mongo/db/stats/single_transaction_stats.h
index 51524c8d01c..fa987e08a02 100644
--- a/src/mongo/db/stats/single_transaction_stats.h
+++ b/src/mongo/db/stats/single_transaction_stats.h
@@ -62,8 +62,10 @@ public:
         }
     };
 
-    SingleTransactionStats() : _txnNumber(kUninitializedTxnNumber){};
-    SingleTransactionStats(TxnNumber txnNumber) : _txnNumber(txnNumber){};
+    SingleTransactionStats()
+        : _txnNumberAndRetryCounter(kUninitializedTxnNumber, kUninitializedTxnRetryCounter){};
+    SingleTransactionStats(TxnNumberAndRetryCounter txnNumberAndRetryCounter)
+        : _txnNumberAndRetryCounter(txnNumberAndRetryCounter){};
 
     /**
      * Sets the transaction's start time, only if it hasn't already been set.
@@ -226,8 +228,8 @@ public:
                 TickSource::Tick curTick) const;
 
 private:
-    // The transaction number of the transaction.
-    TxnNumber _txnNumber;
+    // The struct containing the transaction number and transaction retry counter.
+    TxnNumberAndRetryCounter _txnNumberAndRetryCounter;
 
     // Unset for retryable write, 'false' for multi-document transaction.  Value 'true' is
     // for future use.