vendor/symfony/messenger/Middleware/SendMessageMiddleware.php line 71

Open in your IDE?
  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\Middleware;
  14. use Psr\EventDispatcher\EventDispatcherInterface;
  15. use Psr\Log\LoggerAwareTrait;
  16. use Psr\Log\NullLogger;
  17. use Symfony\Component\Messenger\Envelope;
  18. use Symfony\Component\Messenger\Event\SendMessageToTransportsEvent;
  19. use Symfony\Component\Messenger\Stamp\ReceivedStamp;
  20. use Symfony\Component\Messenger\Stamp\SentStamp;
  21. use Symfony\Component\Messenger\Transport\Sender\SendersLocatorInterface;
  23. /**
  24.  * @author Samuel Roze <samuel.roze@gmail.com>
  25.  * @author Tobias Schultze <http://tobion.de>
  26.  */
  27. class SendMessageMiddleware implements MiddlewareInterface
  28. {
  29.     use LoggerAwareTrait;
  31.     private SendersLocatorInterface $sendersLocator;
  32.     private ?EventDispatcherInterface $eventDispatcher;
  34.     public function __construct(SendersLocatorInterface $sendersLocator, EventDispatcherInterface $eventDispatcher = null)
  35.     {
  36.         $this->sendersLocator = $sendersLocator;
  37.         $this->eventDispatcher = $eventDispatcher;
  38.         $this->logger = new NullLogger();
  39.     }
  41.     /**
  42.      * {@inheritdoc}
  43.      */
  44.     public function handle(Envelope $envelope, StackInterface $stack): Envelope
  45.     {
  46.         $context = [
  47.             'class' => \get_class($envelope->getMessage()),
  48.         ];
  50.         $sender = null;
  52.         if ($envelope->all(ReceivedStamp::class)) {
  53.             // it's a received message, do not send it back
  54.             $this->logger->info('Received message {class}', $context);
  55.         } else {
  56.             $shouldDispatchEvent = true;
  57.             foreach ($this->sendersLocator->getSenders($envelope) as $alias => $sender) {
  58.                 if (null !== $this->eventDispatcher && $shouldDispatchEvent) {
  59.                     $event = new SendMessageToTransportsEvent($envelope);
  60.                     $this->eventDispatcher->dispatch($event);
  61.                     $envelope = $event->getEnvelope();
  62.                     $shouldDispatchEvent = false;
  63.                 }
  65.                 $this->logger->info('Sending message {class} with {alias} sender using {sender}', $context + ['alias' => $alias, 'sender' => \get_class($sender)]);
  66.                 $envelope = $sender->send($envelope->with(new SentStamp(\get_class($sender), \is_string($alias) ? $alias : null)));
  67.             }
  68.         }
  70.         if (null === $sender) {
  71.             return $stack->next()->handle($envelope, $stack);
  72.         }
  74.         // message should only be sent and not be handled by the next middleware
  75.         return $envelope;
  76.     }
  77. }