vendor/symfony/messenger/EventListener/StopWorkerOnCustomStopExceptionListener.php line 43

  1. <?php
  3. /*
  4.  * This file is part of the Symfony package.
  5.  *
  6.  * (c) Fabien Potencier <fabien@symfony.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Symfony\Component\Messenger\EventListener;
  14. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  15. use Symfony\Component\Messenger\Event\WorkerMessageFailedEvent;
  16. use Symfony\Component\Messenger\Event\WorkerRunningEvent;
  17. use Symfony\Component\Messenger\Exception\HandlerFailedException;
  18. use Symfony\Component\Messenger\Exception\StopWorkerExceptionInterface;
  20. /**
  21.  * @author Grégoire Pineau <lyrixx@lyrixx.info>
  22.  */
  23. class StopWorkerOnCustomStopExceptionListener implements EventSubscriberInterface
  24. {
  25.     private bool $stop = false;
  27.     public function onMessageFailed(WorkerMessageFailedEvent $event): void
  28.     {
  29.         $th = $event->getThrowable();
  30.         if ($th instanceof StopWorkerExceptionInterface) {
  31.             $this->stop = true;
  32.         }
  33.         if ($th instanceof HandlerFailedException) {
  34.             foreach ($th->getNestedExceptions() as $e) {
  35.                 if ($e instanceof StopWorkerExceptionInterface) {
  36.                     $this->stop = true;
  37.                     break;
  38.                 }
  39.             }
  40.         }
  41.     }
  43.     public function onWorkerRunning(WorkerRunningEvent $event): void
  44.     {
  45.         if ($this->stop) {
  46.             $event->getWorker()->stop();
  47.         }
  48.     }
  50.     public static function getSubscribedEvents(): array
  51.     {
  52.         return [
  53.             WorkerMessageFailedEvent::class => 'onMessageFailed',
  54.             WorkerRunningEvent::class => 'onWorkerRunning',
  55.         ];
  56.     }
  57. }