src/Core/Content/ProductExport/ScheduledTask/ProductExportGenerateTaskHandler.php line 41

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Content\ProductExport\ScheduledTask;
  5. use Shopware\Core\Content\ProductExport\ProductExportEntity;
  6. use Shopware\Core\Defaults;
  7. use Shopware\Core\Framework\Context;
  8. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepository;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\EqualsFilter;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\MultiFilter;
  12. use Shopware\Core\Framework\Log\Package;
  13. use Shopware\Core\Framework\MessageQueue\ScheduledTask\ScheduledTaskHandler;
  14. use Shopware\Core\Framework\Uuid\Uuid;
  15. use Shopware\Core\System\SalesChannel\Context\AbstractSalesChannelContextFactory;
  16. use Symfony\Component\Messenger\Attribute\AsMessageHandler;
  17. use Symfony\Component\Messenger\MessageBusInterface;
  19. /**
  20.  * @internal
  21.  */
  22. #[AsMessageHandler(handles: ProductExportGenerateTask::class)]
  23. #[Package('sales-channel')]
  24. final class ProductExportGenerateTaskHandler extends ScheduledTaskHandler
  25. {
  26.     /**
  27.      * @internal
  28.      */
  29.     public function __construct(
  30.         EntityRepository $scheduledTaskRepository,
  31.         private readonly AbstractSalesChannelContextFactory $salesChannelContextFactory,
  32.         private readonly EntityRepository $salesChannelRepository,
  33.         private readonly EntityRepository $productExportRepository,
  34.         private readonly MessageBusInterface $messageBus
  35.     ) {
  36.         parent::__construct($scheduledTaskRepository);
  37.     }
  39.     public function run(): void
  40.     {
  41.         $salesChannelIds = $this->fetchSalesChannelIds();
  43.         foreach ($salesChannelIds as $salesChannelId) {
  44.             $productExports = $this->fetchProductExports($salesChannelId);
  46.             if (\count($productExports) === 0) {
  47.                 continue;
  48.             }
  50.             $now = new \DateTimeImmutable('now');
  52.             foreach ($productExports as $productExport) {
  53.                 if (!$this->shouldBeRun($productExport, $now)) {
  54.                     continue;
  55.                 }
  57.                 $this->messageBus->dispatch(
  58.                     new ProductExportPartialGeneration($productExport->getId(), $salesChannelId)
  59.                 );
  60.             }
  61.         }
  62.     }
  64.     /**
  65.      * @return array<string>
  66.      */
  67.     private function fetchSalesChannelIds(): array
  68.     {
  69.         $criteria = new Criteria();
  70.         $criteria
  71.             ->addFilter(new EqualsFilter('typeId', Defaults::SALES_CHANNEL_TYPE_STOREFRONT))
  72.             ->addFilter(new EqualsFilter('active', true));
  74.         /**
  75.          * @var array<string>
  76.          */
  77.         return $this->salesChannelRepository
  78.             ->searchIds($criteria, Context::createDefaultContext())
  79.             ->getIds();
  80.     }
  82.     /**
  83.      * @return array<ProductExportEntity>
  84.      */
  85.     private function fetchProductExports(string $salesChannelId): array
  86.     {
  87.         $salesChannelContext = $this->salesChannelContextFactory->create(Uuid::randomHex(), $salesChannelId);
  89.         $criteria = new Criteria();
  90.         $criteria
  91.             ->addAssociation('salesChannel')
  92.             ->addFilter(
  93.                 new MultiFilter(
  94.                     'AND',
  95.                     [
  96.                         new EqualsFilter('generateByCronjob', true),
  97.                         new EqualsFilter('salesChannel.active', true),
  98.                     ]
  99.                 )
  100.             )
  101.             ->addFilter(
  102.                 new MultiFilter(
  103.                     'OR',
  104.                     [
  105.                         new EqualsFilter('storefrontSalesChannelId', $salesChannelId),
  106.                         new EqualsFilter('salesChannelDomain.salesChannel.id', $salesChannelId),
  107.                     ]
  108.                 )
  109.             );
  111.         /**
  112.          * @var array<ProductExportEntity>
  113.          */
  114.         return $this->productExportRepository->search($criteria, $salesChannelContext->getContext())->getElements();
  115.     }
  117.     private function shouldBeRun(ProductExportEntity $productExport, \DateTimeImmutable $now): bool
  118.     {
  119.         if ($productExport->getIsRunning()) {
  120.             return false;
  121.         }
  123.         if ($productExport->getGeneratedAt() === null) {
  124.             return true;
  125.         }
  127.         return $now->getTimestamp() - $productExport->getGeneratedAt()->getTimestamp() >= $productExport->getInterval();
  128.     }
  129. }