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

Open in your IDE?
  1. <?php
  3. /*
  4. * This file is part of the Symfony package.
  5. *
  6. * (c) Fabien Potencier <[email protected]>
  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 <[email protected]>
  22. */
  23. class StopWorkerOnCustomStopExceptionListener implements EventSubscriberInterface
  24. {
  25. private $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. }