SERVER-29417 Attach logical session ids to cursors

1 files changed, 17 insertions, 0 deletions

diff --git a/src/mongo/db/cursor_manager.h b/src/mongo/db/cursor_manager.h
index 40408c48830..0b243e21996 100644
--- a/src/mongo/db/cursor_manager.h
+++ b/src/mongo/db/cursor_manager.h
@@ -36,6 +36,7 @@
 #include "mongo/db/record_id.h"
 #include "mongo/platform/unordered_map.h"
 #include "mongo/platform/unordered_set.h"
+#include "mongo/stdx/unordered_set.h"
 #include "mongo/util/concurrency/mutex.h"
 #include "mongo/util/duration.h"
 
@@ -78,6 +79,12 @@ public:
     static constexpr Minutes kDefaultCursorTimeoutMinutes{10};
     using RegistrationToken = Partitioned<unordered_set<PlanExecutor*>>::PartitionId;
 
+    /**
+     * Appends the sessions that have open cursors on the global cursor manager and across
+     * all collection-level cursor managers to the given set of lsids.
+     */
+    static void appendAllActiveSessions(OperationContext* opCtx, LogicalSessionIdSet* lsids);
+
     CursorManager(NamespaceString nss);
 
     /**
@@ -159,6 +166,16 @@ public:
     void getCursorIds(std::set<CursorId>* openCursors) const;
 
     /**
+     * Appends sessions that have open cursors in this cursor manager to the given set of lsids.
+     */
+    void appendActiveSessions(LogicalSessionIdSet* lsids) const;
+
+    /*
+     * Returns a list of all open cursors for the given session.
+     */
+    stdx::unordered_set<CursorId> getCursorsForSession(LogicalSessionId lsid) const;
+
+    /**
      * Returns the number of ClientCursors currently registered. Excludes any registered bare
      * PlanExecutors.
      */