vendor/symfony/messenger/MessageBus.php line 62

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;
  14. use Symfony\Component\Messenger\Middleware\MiddlewareInterface;
  15. use Symfony\Component\Messenger\Middleware\StackMiddleware;
  17. /**
  18.  * @author Samuel Roze <samuel.roze@gmail.com>
  19.  * @author Matthias Noback <matthiasnoback@gmail.com>
  20.  * @author Nicolas Grekas <p@tchwork.com>
  21.  */
  22. class MessageBus implements MessageBusInterface
  23. {
  24.     private $middlewareAggregate;
  26.     /**
  27.      * @param iterable<mixed, MiddlewareInterface> $middlewareHandlers
  28.      */
  29.     public function __construct(iterable $middlewareHandlers = [])
  30.     {
  31.         if ($middlewareHandlers instanceof \IteratorAggregate) {
  32.             $this->middlewareAggregate = $middlewareHandlers;
  33.         } elseif (\is_array($middlewareHandlers)) {
  34.             $this->middlewareAggregate = new \ArrayObject($middlewareHandlers);
  35.         } else {
  36.             // $this->middlewareAggregate should be an instance of IteratorAggregate.
  37.             // When $middlewareHandlers is an Iterator, we wrap it to ensure it is lazy-loaded and can be rewound.
  38.             $this->middlewareAggregate = new class($middlewareHandlers) implements \IteratorAggregate {
  39.                 private $middlewareHandlers;
  40.                 private $cachedIterator;
  42.                 public function __construct(\Traversable $middlewareHandlers)
  43.                 {
  44.                     $this->middlewareHandlers = $middlewareHandlers;
  45.                 }
  47.                 public function getIterator(): \Traversable
  48.                 {
  49.                     if (null === $this->cachedIterator) {
  50.                         $this->cachedIterator = new \ArrayObject(iterator_to_array($this->middlewareHandlers, false));
  51.                     }
  53.                     return $this->cachedIterator;
  54.                 }
  55.             };
  56.         }
  57.     }
  59.     /**
  60.      * {@inheritdoc}
  61.      */
  62.     public function dispatch(object $message, array $stamps = []): Envelope
  63.     {
  64.         $envelope = Envelope::wrap($message, $stamps);
  65.         $middlewareIterator = $this->middlewareAggregate->getIterator();
  67.         while ($middlewareIterator instanceof \IteratorAggregate) {
  68.             $middlewareIterator = $middlewareIterator->getIterator();
  69.         }
  70.         $middlewareIterator->rewind();
  72.         if (!$middlewareIterator->valid()) {
  73.             return $envelope;
  74.         }
  75.         $stack = new StackMiddleware($middlewareIterator);
  77.         return $middlewareIterator->current()->handle($envelope, $stack);
  78.     }
  79. }