Update Mgrate batch executable class to handle source url correctly.

LOBsTerr · Jan 14, 2022 · 31bc44a · 31bc44a
1 parent 3e56f9b
commit 31bc44a
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/src/MigrateBatchExecutable.php b/src/MigrateBatchExecutable.php
@@ -31,6 +31,13 @@ class MigrateBatchExecutable extends MigrateExecutable {
      */
     protected $checkDependencies = 0;
 
+    /**
+     * Source URL.
+     *
+     * @var string
+     */
+    protected $sourceUrl = '';
+
     /**
      * The current batch context.
      *
@@ -129,7 +136,7 @@ public function batchImport() {
      */
     protected function batchOperations(array $migrations, $operation, array $options = []) {
         $operations = [];
-        foreach ($migrations as $id => $migration) {
+        foreach ($migrations as $migration) {
 
             if (!empty($options['update'])) {
                 $migration->getIdMap()->prepareUpdate();
@@ -148,7 +155,7 @@ protected function batchOperations(array $migrations, $operation, array $options
                 $dependencies = $migration->getMigrationDependencies();
                 if (!empty($dependencies['required'])) {
                     $required_migrations = $this->migrationPluginManager->createInstances($dependencies['required']);
-                    // For dependent migrations will need to be migrate all items.
+                    // For dependent migrations will need to be migrated all items.
                     $operations = array_merge($operations, $this->batchOperations($required_migrations, $operation, [
                         'limit' => 0,
                         'update' => $options['update'],
@@ -158,7 +165,7 @@ protected function batchOperations(array $migrations, $operation, array $options
             }
 
             $operations[] = [
-                '\Drupal\migrate_tools\MigrateBatchExecutable::batchProcessImport',
+                '\Drupal\newsroom_connector\MigrateBatchExecutable::batchProcessImport',
                 [$migration->id(), $options],
             ];
         }
@@ -196,6 +203,12 @@ public static function batchProcessImport($migration_id, array $options, &$conte
             $options['limit'] = $options['limit'] - $context['results'][$migration->id()]['@numitems'];
         }
 
+        if (!empty($options['source_url'])) {
+            $source = $migration->getSourceConfiguration();
+            $source['urls'] = $options['source_url'];
+            $migration->set('source', $source);
+        }
+
         $executable = new MigrateBatchExecutable($migration, $message, $options);
 
         if (empty($context['sandbox']['total'])) {