gbprod / elastica-provider-bundle

src/ElasticaProviderBundle/Command/ProvidingProgressBar.php

<?php

namespace GBProd\ElasticaProviderBundle\Command;

use GBProd\ElasticaProviderBundle\Event\HasFinishedProviding;
use GBProd\ElasticaProviderBundle\Event\HasProvidedDocument;
use GBProd\ElasticaProviderBundle\Event\HasStartedHandling;
use GBProd\ElasticaProviderBundle\Event\HasStartedProviding;
use Symfony\Component\Console\Helper\ProgressBar;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\EventDispatcher\EventDispatcherInterface;

/**
 * Progress bar for providing
 *
 * @author gbprod <contact@gb-prod.fr>
 */
class ProvidingProgressBar
{
    const PROGRESS_BAR_TEMPLATE = ' %current%/%max% [%bar%] %percent:3s%% %elapsed:6s%/%estimated:-6s% %memory:6s%';

    /**
     * @var OutputInterface
     */
    private $output;

    /**
     * @var ProgressBar
     *
     * public for test purpose
     */
    public $progressBar;

    /**
     * @param EventDispatcherInterface $dispatcher
     * @param OutputInterface          $output
     */
    public function __construct(
        EventDispatcherInterface $dispatcher,
        OutputInterface $output
    ) {
        $this->output = $output;

        $dispatcher->addListener(
            'elasticsearch.has_started_handling',
            [$this, 'onStartedHandling']
        );

        $dispatcher->addListener(
            'elasticsearch.has_started_providing',
            [$this, 'onStartedProviding']
        );

        $dispatcher->addListener(
            'elasticsearch.has_provided_document',
            [$this, 'onProvidedDocument']
        );

        $dispatcher->addListener(
            'elasticsearch.has_finished_providing',
            [$this, 'onFinishedProviding']
        );
    }

    public function onStartedHandling(HasStartedHandling $event)
    {
        $this->output->writeln(
            sprintf(
                '<info>Start running <comment>%d</comment> providers</info>',
                count($event->getEntries())
            )
        );
    }

    public function onStartedProviding(HasStartedProviding $event)
    {
        $this->output->writeln(sprintf(
            '<info> - Running <comment>%s</comment> provider into <comment>%s/%s</comment></info>',
            get_class($event->getEntry()->getProvider()),
            $event->getEntry()->getIndex(),
            $event->getEntry()->getType()
        ));

        $count = $event->getEntry()->getProvider()->count();
        if (null !== $count) {
            $this->progressBar = new ProgressBar($this->output, $count);
            $this->progressBar->setFormat(self::PROGRESS_BAR_TEMPLATE);
        }
    }

    public function onProvidedDocument(HasProvidedDocument $event)

The parameter $event is not used and could be removed.

    {
        if ($this->progressBar) {
            $this->progressBar->advance();
        }
    }

    public function onFinishedProviding(HasFinishedProviding $event)

The parameter $event is not used and could be removed.

    {
        $this->progressBar->finish();
        $this->output->writeln('');
        $this->progressBar = null;
    }
}