Merge pull request #31 from jordandukart/dev_imt_70

Add sleep and wakeup to avoid some serialization pitfalls.

src/MigrateBatchExecutable.php

@@ -23,7 +23,10 @@
  */
 class MigrateBatchExecutable extends MigrateExecutable {
 
-  use DependencySerializationTrait;
+  use DependencySerializationTrait {
+    __sleep as traitSleep;
+    __wakeup as traitWakeup;
+  }
 
   // The name of our timer.
   const TIMER = 'dgi_migrate_iteration_timer';
@@ -69,6 +72,38 @@ public function __construct(MigrationInterface $migration, MigrateMessageInterfa
     }
   }
 
+  /**
+   * {@inheritdoc}
+   */
+  public function __sleep() {
+    $vars = $this->traitSleep();
+    $to_suppress = [
+      // XXX: Avoid serializing some things can't be natively serialized.
+      'source',
+    ];
+    foreach ($to_suppress as $value) {
+      $key = array_search($value, $vars);
+      if ($key !== FALSE) {
+        unset($vars[$key]);
+      }
+    }
+    // XXX: This unsets the protected sourcePlugin variable to avoid having
+    // it be serialized to the DB.
+    $this->migration->set('source', $this->migration->getSourceConfiguration());
+    return $vars;
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public function __wakeup() {
+    $this->traitWakeup();
+    // XXX: Re-add the listeners because the services get re-initialized.
+    foreach ($this->listeners as $event => $listener) {
+      $this->getEventDispatcher()->addListener($event, $listener);
+    }
+  }
+
   /**
    * Prepare a batch array for execution for the given migration.
    *