vendor/3slab/vdm-library-bundle/EventSubscriber/StopWorker/StopWorkerAfterHandleStampSubscriber.php line 86

Open in your IDE?
  1. <?php
  3. /**
  4.  * @package    3slab/VdmLibraryBundle
  5.  * @copyright  2020 Suez Smart Solutions 3S.lab
  6.  * @license    https://github.com/3slab/VdmLibraryBundle/blob/master/LICENSE
  7.  */
  9. namespace Vdm\Bundle\LibraryBundle\EventSubscriber\StopWorker;
  11. use Psr\Log\LoggerInterface;
  12. use Psr\Log\NullLogger;
  13. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  14. use Symfony\Component\Messenger\Event\AbstractWorkerMessageEvent;
  15. use Symfony\Component\Messenger\Event\WorkerMessageFailedEvent;
  16. use Symfony\Component\Messenger\Event\WorkerMessageHandledEvent;
  17. use Vdm\Bundle\LibraryBundle\Event\CollectWorkerMessageFailedEvent;
  18. use Vdm\Bundle\LibraryBundle\Event\CollectWorkerMessageHandledEvent;
  19. use Vdm\Bundle\LibraryBundle\Service\StopWorkerService;
  20. use Vdm\Bundle\LibraryBundle\Stamp\StopAfterHandleStamp;
  22. /**
  23.  * Class StopWorkerAfterHandleStampSubscriber
  24.  *
  25.  * @package Vdm\Bundle\LibraryBundle\EventSubscriber\StopWorker
  26.  */
  27. class StopWorkerAfterHandleStampSubscriber implements EventSubscriberInterface
  28. {
  29.     /**
  30.      * @var StopWorkerService $stopWorker
  31.      */
  32.     private $stopWorker;
  34.     /**
  35.      * @var LoggerInterface
  36.      */
  37.     private $logger;
  39.     /**
  40.      * StopWorkerAfterHandleStampSubscriber constructor.
  41.      *
  42.      * @param StopWorkerService $stopWorker
  43.      * @param LoggerInterface|null $vdmLogger
  44.      */
  45.     public function __construct(StopWorkerService $stopWorker, LoggerInterface $vdmLogger = null)
  46.     {
  47.         $this->stopWorker = $stopWorker;
  48.         $this->logger = $vdmLogger ?? new NullLogger();
  49.     }
  51.     /**
  52.      * Method executed on WorkerMessageFailedEvent event
  53.      *
  54.      * @param WorkerMessageFailedEvent $event
  55.      */
  56.     public function onWorkerMessageFailedEvent(WorkerMessageFailedEvent $event)
  57.     {
  58.         $this->hasStopAfterHandlerStamp($event, 'WorkerMessageFailedEvent');
  59.     }
  61.     /**
  62.      * Method executed on WorkerMessageHandledEvent event
  63.      *
  64.      * @param WorkerMessageHandledEvent $event
  65.      */
  66.     public function onWorkerMessageHandledEvent(WorkerMessageHandledEvent $event)
  67.     {
  68.         $this->hasStopAfterHandlerStamp($event, 'WorkerMessageHandledEvent');
  69.     }
  71.     /**
  72.      * Method executed on CollectWorkerMessageFailedEvent event
  73.      *
  74.      * @param CollectWorkerMessageFailedEvent $event
  75.      */
  76.     public function onCollectWorkerMessageFailedEvent(CollectWorkerMessageFailedEvent $event)
  77.     {
  78.         $this->hasStopAfterHandlerStamp($event, 'CollectWorkerMessageFailedEvent');
  79.     }
  81.     /**
  82.      * Method executed on CollectWorkerMessageHandledEvent event
  83.      *
  84.      * @param CollectWorkerMessageHandledEvent $event
  85.      */
  86.     public function onCollectWorkerMessageHandledEvent(CollectWorkerMessageHandledEvent $event)
  87.     {
  88.         $this->hasStopAfterHandlerStamp($event, 'CollectWorkerMessageHandledEvent');
  89.     }
  91.     /**
  92.      * @param AbstractWorkerMessageEvent $event
  93.      * @param string $eventName
  94.      */
  95.     protected function hasStopAfterHandlerStamp(AbstractWorkerMessageEvent $event, string $eventName)
  96.     {
  97.         $stamps = $event->getEnvelope()->all();
  98.         if (in_array(StopAfterHandleStamp::class, array_keys($stamps))) {
  99.             $this->logger->debug(
  100.                 'StopAfterHandleStamp detected on envelop during event {eventName} so we schedule to stop the worker',
  101.                 ['eventName' => $eventName]
  102.             );
  103.             $this->stopWorker->setFlag(true);
  104.         }
  105.     }
  107.     /**
  108.      * {@inheritDoc}
  109.      * @codeCoverageIgnore
  110.      */
  111.     public static function getSubscribedEvents(): array
  112.     {
  113.         return [
  114.             CollectWorkerMessageFailedEvent::class => 'onCollectWorkerMessageFailedEvent',
  115.             CollectWorkerMessageHandledEvent::class => 'onCollectWorkerMessageHandledEvent',
  116.             WorkerMessageFailedEvent::class => 'onWorkerMessageFailedEvent',
  117.             WorkerMessageHandledEvent::class => 'onWorkerMessageHandledEvent',
  118.         ];
  119.     }
  120. }