SERVER-50066 Rehost tenant migration donor's logic on Primary-Only Service

1 files changed, 19 insertions, 8 deletions

diff --git a/src/mongo/db/commands/tenant_migration_donor_cmds.cpp b/src/mongo/db/commands/tenant_migration_donor_cmds.cpp
index 16e773b2e0f..ec9c2a74946 100644
--- a/src/mongo/db/commands/tenant_migration_donor_cmds.cpp
+++ b/src/mongo/db/commands/tenant_migration_donor_cmds.cpp
@@ -29,6 +29,8 @@
 
 #include "mongo/db/commands.h"
 #include "mongo/db/commands/tenant_migration_donor_cmds_gen.h"
+#include "mongo/db/repl/primary_only_service.h"
+#include "mongo/db/repl/tenant_migration_donor_service.h"
 #include "mongo/db/repl/tenant_migration_donor_util.h"
 
 namespace mongo {
@@ -42,19 +44,28 @@ public:
 
     public:
         using InvocationBase::InvocationBase;
+
         void typedRun(OperationContext* opCtx) {
             const RequestType& requestBody = request();
 
-            const TenantMigrationDonorDocument donorStateDoc(
-                requestBody.getMigrationId(),
-                requestBody.getRecipientConnectionString().toString(),
-                requestBody.getDatabasePrefix().toString(),
-                TenantMigrationDonorStateEnum::kDataSync);
-
-            tenant_migration_donor::startMigration(opCtx, donorStateDoc);
+            const auto donorStateDoc =
+                TenantMigrationDonorDocument(requestBody.getMigrationId(),
+                                             requestBody.getRecipientConnectionString().toString(),
+                                             requestBody.getReadPreference(),
+                                             requestBody.getDatabasePrefix().toString(),
+                                             TenantMigrationDonorStateEnum::kDataSync)
+                    .toBSON();
+
+            auto donorService =
+                repl::PrimaryOnlyServiceRegistry::get(opCtx->getServiceContext())
+                    ->lookupServiceByName(TenantMigrationDonorService::kServiceName);
+            auto donor =
+                TenantMigrationDonorService::Instance::getOrCreate(donorService, donorStateDoc);
+            uassertStatusOK(donor->checkIfOptionsConflict(donorStateDoc));
+
+            donor->onCompletion().get();
         }
 
-
         void doCheckAuthorization(OperationContext* opCtx) const {}
 
     private: