Shippinno\Job\Application\Messaging\ConsumeStoredJobService

Complexity

Total Complexity

23

Size/Duplication

Total Lines	191
Duplicated Lines	10.99 %

Coupling/Cohesion

Components	1
Dependencies	19

Test Coverage

Coverage

16.13%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	19	3
F	execute()	21	103	18
A	delayMessage()	0	8	2

<?php

namespace Shippinno\Job\Application\Messaging;

use Closure;
use Enqueue\Sqs\SqsMessage;
use Interop\Queue\PsrContext;
use Interop\Queue\PsrMessage;
use Psr\Log\LoggerAwareTrait;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;
use Shippinno\Job\Application\Job\JobRunnerRegistry;
use Shippinno\Job\Domain\Model\AbandonedJobMessage;
use Shippinno\Job\Domain\Model\AbandonedJobMessageStore;
use Shippinno\Job\Domain\Model\JobFailedException;
use Shippinno\Job\Domain\Model\JobRunnerNotRegisteredException;
use Shippinno\Job\Domain\Model\JobSerializer;
use Shippinno\Job\Domain\Model\JobStore;
use Shippinno\Job\Domain\Model\StoredJobSerializer;

class ConsumeStoredJobService
{
    use LoggerAwareTrait;

    /**
     * @var StoredJobSerializer
     */
    private $storedJobSerializer;

    /**
     * @var PsrContext
     */
    private $context;

    /**
     * @var JobSerializer
     */
    private $jobSerializer;

    /**
     * @var JobRunnerRegistry
     */
    private $jobRunnerRegistry;

    /**
     * @var JobStore
     */
    private $jobStore;

    /**
     * @var AbandonedJobMessageStore
     */
    private $abandonedJobMessageStore;

    /**
     * @var JobFlightManager
     */
    private $jobFlightManager;

    /**
     * @param PsrContext $context
     * @param StoredJobSerializer $storedJobSerializer
     * @param JobSerializer $jobSerializer
     * @param JobRunnerRegistry $jobRunnerRegistry
     * @param JobStore $jobStore
     * @param AbandonedJobMessageStore $abandonedJobMessageStore
     * @param JobFlightManager|null $jobFlightManager
     * @param LoggerInterface|null $logger
     */
    public function __construct(
        PsrContext $context,
        StoredJobSerializer $storedJobSerializer,
        JobSerializer $jobSerializer,
        JobRunnerRegistry $jobRunnerRegistry,
        JobStore $jobStore,
        AbandonedJobMessageStore $abandonedJobMessageStore,
        JobFlightManager $jobFlightManager = null,
        LoggerInterface $logger = null
    ) {
        $this->context = $context;
        $this->storedJobSerializer = $storedJobSerializer;
        $this->jobSerializer = $jobSerializer;
        $this->jobRunnerRegistry = $jobRunnerRegistry;
        $this->jobStore = $jobStore;
        $this->abandonedJobMessageStore = $abandonedJobMessageStore;
        $this->jobFlightManager = $jobFlightManager ?: new NullJobFlightManager;
        $this->setLogger($logger ?: new NullLogger);
    }

    /**
     * @param string $queueName
     * @param Closure|null $persist
     */
    public function execute(string $queueName, Closure $persist = null, Closure $clear = null): void
    {
        $consumer = $this->context->createConsumer($this->context->createQueue($queueName));
        $message = $consumer->receive(5000);
        if (null === $message) {
            return;
        }
        $storedJob = $this->storedJobSerializer->deserialize($message->getBody());
        if (null === $message->getMessageId()) {
            $this->logger->alert('Message without ID. Filling it.', ['message' => $message->getBody()]);
            $message->setMessageId($storedJob->id());
        }
        $this->jobFlightManager->arrived($message->getMessageId());
        $job = $this->jobSerializer->deserialize($storedJob->body(), $storedJob->name());
        try {
            $jobRunner = $this->jobRunnerRegistry->get(get_class($job));
        } catch (JobRunnerNotRegisteredException $e) {
            $this->abandonedJobMessageStore->add(
                new AbandonedJobMessage($queueName, $message->getBody(), $e->__toString())
            );
            $this->logger->alert(
                'No JobRunner is registered. Message is abandoned. Rejecting the message.',
                ['message' => $message->getBody()]
            );
            $this->jobFlightManager->rejected($message->getMessageId());
            $consumer->reject($message);
            return;
        }
        try {
            $jobRunner->run($job);
            if (!is_null($persist) && !$persist()) {

This code seems to be duplicated across your project.

                $this->logger->info(
                    'Persistence failed after the job. Requeueing the message.',
                    ['message' => $message->getBody()]
                );
                $newMessageId = $message->getMessageId() . '+';
                $this->jobFlightManager->requeued($message->getMessageId(), $newMessageId);
                $message->setMessageId($newMessageId);
                $consumer->reject($message, true);
                return;
            }
            $dependentJobs = $job->dependentJobs();
            if (count($dependentJobs) > 0) {
                foreach ($dependentJobs as $dependentJob) {
                    $this->jobStore->append($dependentJob);
                }
            }
            $this->logger->info('Acknowledging message.', ['message' => $message->getBody()]);
            $this->jobFlightManager->acknowledged($message->getMessageId());
            $consumer->acknowledge($message);
        } catch (JobFailedException $e) {
            if ($job->isExpendable()) {
                $this->logger->debug(
                    'Expendable job failed. Acknowledging and letting it go.',
                    ['message' => $message->getBody()]
                );
                $this->jobFlightManager->letGo($message->getMessageId());
                $consumer->acknowledge($message);
                return;
            }
            $attempts = $message->getProperty('attempts', 0) + 1;
            if ($attempts >= $job->maxAttempts()) {
                if (!is_null($clear)) {
                    $clear();
                }
                $this->abandonedJobMessageStore->add(
                    new AbandonedJobMessage($queueName, $message->getBody(), $e->__toString())
                );
                $this->logger->info(
                    'Rejecting the message reaching the max attempts.',
                    ['message' => $message->getBody()]
                );
                $this->jobFlightManager->rejected($message->getMessageId());
                $consumer->reject($message);
                if (!is_null($persist) && !$persist()) {

This code seems to be duplicated across your project.

                    $this->logger->info(
                        'Failed after the job. Requeueing the message.',
                        ['message' => $message->getBody()]
                    );
                    $newMessageId = $message->getMessageId() . '+';
                    $this->jobFlightManager->requeued($message->getMessageId(), $newMessageId);
                    $message->setMessageId($newMessageId);
                    $consumer->reject($message, true);
                }
                return;
            }
            $message->setProperty('attempts', $attempts);
            if ($job->reattemptDelay() > 0) {
                $message = $this->delayMessage($message, $job->reattemptDelay());
            }
            if (method_exists($message, 'setMessageDeduplicationId')) {
                $message->setMessageDeduplicationId(uniqid());
            }
            if (method_exists($message, 'setMessageGroupId')) {
                $message->setMessageGroupId(is_null($storedJob->fifoGroupId()) ? uniqid() : $storedJob->fifoGroupId());
            }
            $this->logger->info('Requeueing the message.', ['message' => $message->getBody()]);
            $newMessageId = $message->getMessageId() . '+';
            $this->jobFlightManager->requeued($message->getMessageId(), $newMessageId);
            $message->setMessageId($newMessageId);
            $consumer->reject($message, true);
        }
    }

    /**
     * @param PsrMessage $message
     * @param int $delay
     * @return PsrMessage
     */
    protected function delayMessage(PsrMessage $message, int $delay)
    {
        if ($message instanceof SqsMessage) {
            $message->setDelaySeconds($delay);
        }

        return $message;
    }
}