Darkilliant\MqProcessBundle\Step\SendMessageStep

Complexity

Total Complexity

11

Size/Duplication

Total Lines	76
Duplicated Lines	0 %

Test Coverage

Coverage

0%

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	getPublisherName()	0	5	1
A	configureOptionResolver()	0	9	1
A	finalize()	0	12	3
A	execute()	0	5	1
A	__construct()	0	4	1
A	count()	0	7	2
A	getProgress()	0	3	1
A	getPublisher()	0	3	1

<?php

namespace Darkilliant\MqProcessBundle\Step;

use Darkilliant\MqProcessBundle\Message\Publisher\MessagePublisherFactory;
use Darkilliant\ProcessBundle\ProcessNotifier\ProgressBarProcessNotifier;
use Darkilliant\ProcessBundle\State\ProcessState;
use Darkilliant\ProcessBundle\Step\AbstractConfigurableStep;
use Darkilliant\ProcessBundle\Step\MonitorableStepInterface;
use Symfony\Component\OptionsResolver\OptionsResolver;

class SendMessageStep extends AbstractConfigurableStep implements MonitorableStepInterface
{
    /** @var MessagePublisherFactory */
    private $publisherFactory;

    /** @var ProgressBarProcessNotifier */
    private $notifier;

    private $count;

    private $countMessageReceived = 0;

    public function __construct(MessagePublisherFactory $publisherFactory, ProgressBarProcessNotifier $notifier)
    {
        $this->publisherFactory = $publisherFactory;
        $this->notifier = $notifier;
    }

    public function configureOptionResolver(OptionsResolver $resolver): OptionsResolver
    {
        $resolver->setRequired(['queue', 'exchange', 'client', 'wait_consume_on_finalize', 'batch_count', 'persistant']);
        $resolver->setDefault('client', 'amqp_lib');
        $resolver->setDefault('wait_consume_on_finalize', false);
        $resolver->setDefault('persistant', false);
        $resolver->setDefault('batch_count', 1);

        return parent::configureOptionResolver($resolver); // TODO: Change the autogenerated stub
    }

    public function execute(ProcessState $state)
    {
        $this->getPublisher($state)->publish(
            json_encode($state->getData()),
            $state->getOptions()['batch_count']
        );
    }

    public function count(ProcessState $state)
    {
        if (null === $this->count) {
            $this->count = $this->getPublisher($state)->countMessages();
        }

        return $this->count;
    }

    public function getProgress(ProcessState $state)
    {
        return $this->countMessageReceived = $this->count - $this->getPublisher($state)->countMessages();
    }

    public function finalize(ProcessState $state)
    {
        $this->getPublisher($state)->finalize();

        if (!$state->getOptions()['wait_consume_on_finalize']) {
            return;
        }

        $this->notifier->onStartIterableProcess($state, $this);
        while ($this->countMessageReceived < $this->count) {
            $this->notifier->onUpdateIterableProcess($state, $this);
            sleep(1);
        }
    }

    private function getPublisherName(ProcessState $state)
    {
        $options = $state->getOptions();

        return $options['exchange'].'_'.$options['queue'];
    }

    private function getPublisher(ProcessState $state)
    {
        return $this->publisherFactory->factory($this->getPublisherName($state), $state->getOptions());
    }
}