MigrationDrushCommandProgress.php in Migrate Tools 8.5

Namespace

Drupal\migrate_tools\EventSubscriber

File

src/EventSubscriber/MigrationDrushCommandProgress.php

<?php

namespace Drupal\migrate_tools\EventSubscriber;

use Drupal\migrate\Event\MigrateEvents;
use Drupal\migrate\Plugin\MigrationInterface;
use Psr\Log\LoggerInterface;
use Symfony\Component\Console\Helper\ProgressBar;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;

/**
 * Import and rollback progress bar.
 */
class MigrationDrushCommandProgress implements EventSubscriberInterface {

  /**
   * The logger service.
   *
   * @var \Psr\Log\LoggerInterface
   */
  protected $logger;

  /**
   * MigrationDrushCommandProgress constructor.
   *
   * @param \Psr\Log\LoggerInterface $logger
   *   The logger service.
   */
  public function __construct(LoggerInterface $logger) {
    $this->logger = $logger;
  }

  /**
   * The progress bar.
   *
   * @var \Symfony\Component\Console\Helper\ProgressBar
   */
  protected $symfonyProgressBar;

  /**
   * {@inheritdoc}
   */
  public static function getSubscribedEvents() {
    $events = [];
    $events[MigrateEvents::POST_ROW_SAVE][] = [
      'updateProgressBar',
      -10,
    ];
    $events[MigrateEvents::MAP_DELETE][] = [
      'updateProgressBar',
      -10,
    ];
    $events[MigrateEvents::POST_IMPORT][] = [
      'clearProgress',
      10,
    ];
    $events[MigrateEvents::POST_ROLLBACK][] = [
      'clearProgress',
      10,
    ];
    return $events;
  }

  /**
   * Initializes the progress bar.
   *
   * This must be called before the progress bar can be used.
   *
   * @param \Symfony\Component\Console\Output\OutputInterface $output
   *   The output.
   * @param \Drupal\migrate\Plugin\MigrationInterface $migration
   *   The migration.
   */
  public function initializeProgress(OutputInterface $output, MigrationInterface $migration) {

    // Don't display progress bar if explicitly disabled.
    if (!empty($migration->skipProgressBar)) {
      return;
    }

    // If the source is configured to skip counts, a progress bar is not
    // possible.
    if (!empty($migration
      ->getSourceConfiguration()['skip_count'])) {
      return;
    }
    try {

      // Clone so that any generators aren't initialized prematurely.
      $source = clone $migration
        ->getSourcePlugin();
      $this->symfonyProgressBar = new ProgressBar($output, $source
        ->count());
    } catch (\Exception $exception) {
      if (!empty($migration->continueOnFailure)) {
        $this->logger
          ->error($exception
          ->getMessage());
      }
      else {
        throw $exception;
      }
    }
  }

  /**
   * Event callback for advancing the progress bar.
   */
  public function updateProgressBar() {
    if ($this
      ->isProgressBar()) {
      $this->symfonyProgressBar
        ->advance();
    }
  }

  /**
   * Event callback for removing the progress bar after operation is finished.
   */
  public function clearProgress() {
    if ($this
      ->isProgressBar()) {
      $this->symfonyProgressBar
        ->clear();
    }
  }

  /**
   * Determine if a progress bar should be displayed.
   *
   * @return bool
   *   TRUE if a progress bar should be displayed, FALSE otherwise.
   */
  protected function isProgressBar() {

    // Can't do anything if the progress bar is not initialised; this probably
    // means we're not running as a Drush command and so we should do nothing.
    if (!$this->symfonyProgressBar) {
      return FALSE;
    }
    return TRUE;
  }

}

Classes

Name	Description
MigrationDrushCommandProgress	Import and rollback progress bar.