SERVER-29908 Move more libraries from sharding into sharding_api_d/sharding_runtime_d

1 files changed, 14 insertions, 2 deletions

diff --git a/src/mongo/db/s/active_migrations_registry.cpp b/src/mongo/db/s/active_migrations_registry.cpp
index e98497d90f9..6c07939d7d7 100644
--- a/src/mongo/db/s/active_migrations_registry.cpp
+++ b/src/mongo/db/s/active_migrations_registry.cpp
@@ -30,14 +30,18 @@
 
 #include "mongo/db/s/active_migrations_registry.h"
 
-#include "mongo/base/status_with.h"
 #include "mongo/db/catalog_raii.h"
 #include "mongo/db/s/collection_sharding_state.h"
 #include "mongo/db/s/migration_session_id.h"
 #include "mongo/db/s/migration_source_manager.h"
-#include "mongo/util/assert_util.h"
+#include "mongo/db/service_context.h"
 
 namespace mongo {
+namespace {
+
+const auto getRegistry = ServiceContext::declareDecoration<ActiveMigrationsRegistry>();
+
+}  // namespace
 
 ActiveMigrationsRegistry::ActiveMigrationsRegistry() = default;
 
@@ -45,6 +49,14 @@ ActiveMigrationsRegistry::~ActiveMigrationsRegistry() {
     invariant(!_activeMoveChunkState);
 }
 
+ActiveMigrationsRegistry& ActiveMigrationsRegistry::get(ServiceContext* service) {
+    return getRegistry(service);
+}
+
+ActiveMigrationsRegistry& ActiveMigrationsRegistry::get(OperationContext* opCtx) {
+    return get(opCtx->getServiceContext());
+}
+
 StatusWith<ScopedDonateChunk> ActiveMigrationsRegistry::registerDonateChunk(
     const MoveChunkRequest& args) {
     stdx::lock_guard<stdx::mutex> lk(_mutex);