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

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