DailyFeedCronModule.php

<?php

namespace Shopsys\FrameworkBundle\Model\Feed;

use Shopsys\FrameworkBundle\Component\Domain\Domain;
use Shopsys\FrameworkBundle\Component\Setting\Setting;
use Shopsys\Plugin\Cron\IteratedCronModuleInterface;
use Symfony\Bridge\Monolog\Logger;

class DailyFeedCronModule implements IteratedCronModuleInterface
{
    /**
     * @var \Shopsys\FrameworkBundle\Model\Feed\FeedFacade
     */
    protected $feedFacade;

    /**
     * @var \Shopsys\FrameworkBundle\Component\Domain\Domain
     */
    protected $domain;

    /**
     * @var \Shopsys\FrameworkBundle\Component\Setting\Setting
     */
    protected $setting;

    /**
     * @var \Symfony\Bridge\Monolog\Logger
     */
    protected $logger;

    /**
     * @var \Shopsys\FrameworkBundle\Model\Feed\FeedExportCreationDataQueue
     */
    protected $feedExportCreationDataQueue;

    /**
     * @var \Shopsys\FrameworkBundle\Model\Feed\FeedExport|null
     */
    protected $currentFeedExport;

    /**
     * @param \Shopsys\FrameworkBundle\Model\Feed\FeedFacade $feedFacade
     * @param \Shopsys\FrameworkBundle\Component\Domain\Domain $domain
     * @param \Shopsys\FrameworkBundle\Component\Setting\Setting $setting
     */
    public function __construct(FeedFacade $feedFacade, Domain $domain, Setting $setting)
    {
        $this->feedFacade = $feedFacade;
        $this->domain = $domain;
        $this->setting = $setting;
        $this->feedExportCreationDataQueue = new FeedExportCreationDataQueue(
            $this->feedFacade->getFeedNames('daily'),
            $this->domain->getAll()
        );
    }

    /**
     * {@inheritdoc}
     */
    public function setLogger(Logger $logger): void
    {
        $this->logger = $logger;
    }

    /**
     * {@inheritdoc}
     */
    public function iterate(): bool
    {
        if ($this->feedExportCreationDataQueue->isEmpty()) {
            $this->logger->addDebug('Queue is empty, no feeds to process.');

            return false;
        }

        if ($this->currentFeedExport === null) {
            $this->currentFeedExport = $this->createCurrentFeedExport();
        }

        $this->currentFeedExport->generateBatch();

        if ($this->currentFeedExport->isFinished()) {
            $feedInfo = $this->currentFeedExport->getFeedInfo();
            $domainConfig = $this->currentFeedExport->getDomainConfig();

            $this->logger->addDebug(sprintf(
                'Feed "%s" generated on domain "%s" into "%s".',
                $feedInfo->getName(),
                $domainConfig->getName(),
                $this->feedFacade->getFeedFilepath($feedInfo, $domainConfig)
            ));

            $this->currentFeedExport = null;

            $existsNext = $this->feedExportCreationDataQueue->next();
            if ($existsNext === true) {
                $this->currentFeedExport = $this->createCurrentFeedExport();
            }

            return $existsNext;
        }

        return true;
    }

    /**
     * {@inheritdoc}
     */
    public function sleep(): void
    {
        $lastSeekId = $this->currentFeedExport !== null ? $this->currentFeedExport->getLastSeekId() : null;

        if ($lastSeekId !== null) {
            $this->currentFeedExport->sleep();
        }

        $currentFeedName = $this->feedExportCreationDataQueue->getCurrentFeedName();
        $currentDomain = $this->feedExportCreationDataQueue->getCurrentDomain();

        $this->setting->set(Setting::FEED_NAME_TO_CONTINUE, $currentFeedName);
        $this->setting->set(Setting::FEED_DOMAIN_ID_TO_CONTINUE, $currentDomain->getId());
        $this->setting->set(Setting::FEED_ITEM_ID_TO_CONTINUE, $lastSeekId);

        $this->logger->addDebug(sprintf(
            'Going to sleep... Will continue with feed "%s" on "%s", processing from ID %d.',
            $currentFeedName,
            $currentDomain->getName(),
            $lastSeekId
        ));
    }

    /**
     * {@inheritdoc}
     */
    public function wakeUp(): void
    {
        $feedNameToContinue = $this->setting->get(Setting::FEED_NAME_TO_CONTINUE);
        $domainIdToContinue = $this->setting->get(Setting::FEED_DOMAIN_ID_TO_CONTINUE);
        if ($feedNameToContinue !== null && $domainIdToContinue !== null) {
            $queue = $this->feedExportCreationDataQueue;
            while ($feedNameToContinue !== $queue->getCurrentFeedName() || $domainIdToContinue !== $queue->getCurrentDomain()->getId()) {
                $queue->next();
            }
        }

        $lastSeekId = $this->setting->get(Setting::FEED_ITEM_ID_TO_CONTINUE);
        $this->currentFeedExport = $this->createCurrentFeedExport($lastSeekId);
        $this->currentFeedExport->wakeUp();

        $this->logger->addDebug(sprintf(
            'Waking up... Continuing with feed "%s" on "%s", processing from ID %d.',
            $this->feedExportCreationDataQueue->getCurrentFeedName(),
            $this->feedExportCreationDataQueue->getCurrentDomain()->getName(),
            $this->currentFeedExport->getLastSeekId()
        ));
    }

    /**
     * @param int|null $lastSeekId
     * @return \Shopsys\FrameworkBundle\Model\Feed\FeedExport
     */
    protected function createCurrentFeedExport(int $lastSeekId = null): FeedExport
    {
        return $this->feedFacade->createFeedExport(
            $this->feedExportCreationDataQueue->getCurrentFeedName(),
            $this->feedExportCreationDataQueue->getCurrentDomain(),
            $lastSeekId
        );
    }
}