GBProd\ElasticsearchDataProviderBundle\Command\ProvidingProgressBar

Complexity

Total Complexity

7

Size/Duplication

Total Lines	87
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	7

Test Coverage

Coverage

100%

Importance

Changes	9
Bugs	1	Features	4

5 Methods

Rating	Name	Duplication	Size	Complexity
A	onStartedHandling()	0	9	1
A	onStartedProviding()	0	15	2
A	onFinishedProviding()	0	6	1
B	__construct()	0	26	1
A	onProvidedDocument()	0	6	2

<?php

namespace GBProd\ElasticsearchDataProviderBundle\Command;

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

/**
 * 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)
    {
        if ($this->progressBar) {
            $this->progressBar->advance();
        }
    }

    public function onFinishedProviding(HasFinishedProviding $event)
    {
        $this->progressBar->finish();
        $this->output->writeln('');
        $this->progressBar = null;
    }
}