SERVER-31289 Add test only $listLocalCursors

In order to make the kill sessions test a bit more reliable, adding a
test only agg stage that returns minimal info about cursors.  That way
we can see if we killed our specific cursors, rather than looking at
global counts.

1 files changed, 5 insertions, 0 deletions

diff --git a/src/mongo/s/server.cpp b/src/mongo/s/server.cpp
index e2e87e187d2..54e537e2e83 100644
--- a/src/mongo/s/server.cpp
+++ b/src/mongo/s/server.cpp
@@ -51,6 +51,8 @@
 #include "mongo/db/auth/user_cache_invalidator_job.h"
 #include "mongo/db/client.h"
 #include "mongo/db/ftdc/ftdc_mongos.h"
+#include "mongo/db/generic_cursor_manager.h"
+#include "mongo/db/generic_cursor_manager_mongos.h"
 #include "mongo/db/initialize_server_global_state.h"
 #include "mongo/db/kill_sessions.h"
 #include "mongo/db/lasterror.h"
@@ -439,6 +441,9 @@ static ExitCode runMongosServer() {
         getGlobalServiceContext(),
         std::make_shared<SessionKiller>(getGlobalServiceContext(), killSessionsRemote));
 
+    GenericCursorManager::set(getGlobalServiceContext(),
+                              stdx::make_unique<GenericCursorManagerMongos>());
+
     // Set up the logical session cache
     LogicalSessionCache::set(getGlobalServiceContext(), makeLogicalSessionCacheS());