SERVER-29531 Handle rollbacks in SessionTransactionTable

1 files changed, 11 insertions, 0 deletions

diff --git a/src/mongo/db/session_catalog.h b/src/mongo/db/session_catalog.h
index 3d77f64a249..c6a67f6c44c 100644
--- a/src/mongo/db/session_catalog.h
+++ b/src/mongo/db/session_catalog.h
@@ -60,6 +60,12 @@ public:
     static void create(ServiceContext* service);
 
     /**
+     * Resets the transaction table on the specified service context to an uninitialized state.
+     * Meant only for testing.
+     */
+    static void reset_forTest(ServiceContext* service);
+
+    /**
      * Retrieves the session transaction table associated with the service or operation context.
      * Must only be called after 'create' has been called.
      */
@@ -86,6 +92,11 @@ public:
      */
     ScopedSession checkOutSession(OperationContext* opCtx);
 
+    /**
+     * Clears the entire transaction table. Invoked after rollback.
+     */
+    void clearTransactionTable();
+
 private:
     struct SessionRuntimeInfo {
         SessionRuntimeInfo(LogicalSessionId lsid) : txnState(std::move(lsid)) {}