SERVER-58504 Abort in-progress transactions when a load-balanced client disconnects from mongos

1 files changed, 9 insertions, 0 deletions

diff --git a/src/mongo/s/transaction_router.h b/src/mongo/s/transaction_router.h
index e0f0a1eb8e0..687ea19f51e 100644
--- a/src/mongo/s/transaction_router.h
+++ b/src/mongo/s/transaction_router.h
@@ -532,6 +532,15 @@ public:
             return o().metricsTracker->getTimingStats();
         }
 
+        /**
+         * Returns if the router is not currently tracking an active transaction.
+         */
+        bool isTrackingOver() {
+            if (o().metricsTracker)
+                return o().metricsTracker->isTrackingOver();
+            return true;
+        }
+
     private:
         /**
          * Resets the router's state. Used when the router sees a new transaction for the first