vendor/pimcore/portal-engine/src/EventSubscriber/BatchTaskSubscriber.php line 99

Open in your IDE?
  1. <?php
  3. /**
  4.  * Pimcore
  5.  *
  6.  * This source file is available under following license:
  7.  * - Pimcore Commercial License (PCL)
  8.  *
  9.  *  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
  10.  *  @license    http://www.pimcore.org/license     PCL
  11.  */
  13. namespace Pimcore\Bundle\PortalEngineBundle\EventSubscriber;
  15. use Pimcore\Bundle\PortalEngineBundle\Entity\BatchTask;
  16. use Pimcore\Bundle\PortalEngineBundle\Message\BatchTask\Interfaces\BatchTaskMessageInterface;
  17. use Pimcore\Bundle\PortalEngineBundle\Message\BatchTask\Interfaces\SequentialBatchTaskMessageInterface;
  18. use Pimcore\Bundle\PortalEngineBundle\Message\BatchTask\Interfaces\SplittedBatchTaskMessageInterface;
  19. use Pimcore\Bundle\PortalEngineBundle\Message\BatchTask\Interfaces\TriggerFinishedMessageBatchTaskMessageInterface;
  20. use Pimcore\Bundle\PortalEngineBundle\Service\BatchTask\BatchTaskService;
  21. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  22. use Symfony\Component\HttpKernel\Event\TerminateEvent;
  23. use Symfony\Component\Messenger\Event\WorkerMessageFailedEvent;
  24. use Symfony\Component\Messenger\Event\WorkerMessageHandledEvent;
  25. use Symfony\Component\Messenger\MessageBusInterface;
  27. /**
  28.  * Class IndexUpdateListener
  29.  *
  30.  * @package Pimcore\Bundle\PortalEngineBundle\EventListener
  31.  */
  32. class BatchTaskSubscriber implements EventSubscriberInterface
  33. {
  34.     /**
  35.      * @var BatchTaskService
  36.      */
  37.     protected $batchTaskService;
  39.     /**
  40.      * @var MessageBusInterface
  41.      */
  42.     protected $messageBus;
  44.     /**
  45.      * @var BatchTask
  46.      */
  47.     protected $terminateBatchTask;
  49.     /**
  50.      * @param BatchTaskService $batchTaskService
  51.      * @param MessageBusInterface $messengerBusPortalEngine
  52.      */
  53.     public function __construct(BatchTaskService $batchTaskService, MessageBusInterface $messengerBusPortalEngine)
  54.     {
  55.         $this->batchTaskService = $batchTaskService;
  56.         $this->messageBus = $messengerBusPortalEngine;
  57.     }
  59.     public static function getSubscribedEvents()
  60.     {
  61.         return [
  62.             WorkerMessageFailedEvent::class => 'onBatchTaskMessageFailed',
  63.             WorkerMessageHandledEvent::class => 'onWorkerMessageHandled',
  64.             TerminateEvent::class => 'onTerminate',
  65.         ];
  66.     }
  68.     /**
  69.      * Mark batch tasks with failed items as finished as otherwise they will run forever.
  70.      *
  71.      * @param WorkerMessageFailedEvent $event
  72.      *
  73.      * @throws \Doctrine\DBAL\DBALException
  74.      */
  75.     public function onBatchTaskMessageFailed(WorkerMessageFailedEvent $event)
  76.     {
  77.         $message = $event->getEnvelope()->getMessage();
  78.         if (!$message instanceof BatchTaskMessageInterface) {
  79.             return;
  80.         }
  82.         if ($event->willRetry()) {
  83.             return;
  84.         }
  86.         if (!$batchTask = $this->batchTaskService->getTaskById($message->getTaskId())) {
  87.             return;
  88.         }
  90.         foreach (array_keys($message->getItems()) as $itemIndex) {
  91.             if (!$this->batchTaskService->isItemIndexProcessed($batchTask, $itemIndex)) {
  92.                 $this->batchTaskService->markItemIndexAsProcessed($batchTask, $itemIndex);
  93.             }
  94.         }
  96.         $this->checkBatchTaskFinished($batchTask, $message);
  97.     }
  99.     public function onWorkerMessageHandled(WorkerMessageHandledEvent $event)
  100.     {
  101.         $message = $event->getEnvelope()->getMessage();
  102.         if (!$message instanceof BatchTaskMessageInterface) {
  103.             return;
  104.         }
  106.         if ($message instanceof SplittedBatchTaskMessageInterface) {
  107.             return;
  108.         }
  110.         if (!$batchTask = $this->batchTaskService->getTaskById($message->getTaskId())) {
  111.             return;
  112.         }
  114.         if ($message instanceof SequentialBatchTaskMessageInterface && $message->hasRemainingItems()) {
  115.             $remainingMessage = $message->createRemainingMessage($this->batchTaskService);
  116.             $this->messageBus->dispatch($remainingMessage);
  118.             return;
  119.         }
  121.         $this->checkBatchTaskFinished($batchTask, $message);
  122.     }
  124.     public function onTerminate(TerminateEvent $event)
  125.     {
  126.         $this->batchTaskService->terminateBatchTask();
  127.     }
  129.     public function setTerminateBatchTask(BatchTask $batchTask)
  130.     {
  131.         $this->terminateBatchTask = $batchTask;
  132.     }
  134.     protected function checkBatchTaskFinished(BatchTask $batchTask, BatchTaskMessageInterface $message)
  135.     {
  136.         $this->batchTaskService->checkBatchTaskFinished($batchTask);
  138.         if ($message instanceof TriggerFinishedMessageBatchTaskMessageInterface) {
  139.             $finishedMessage = $message->createFinishedMessage();
  140.             $this->messageBus->dispatch($finishedMessage);
  141.         }
  142.     }
  143. }