vendor/symfony/messenger/MessageBus.php line 70

  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 \IteratorAggregate $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 \Traversable $middlewareHandlers;
  40.                 private \ArrayObject $cachedIterator;
  42.                 public function __construct(\Traversable $middlewareHandlers)
  43.                 {
  44.                     $this->middlewareHandlers = $middlewareHandlers;
  45.                 }
  47.                 public function getIterator(): \Traversable
  48.                 {
  49.                     return $this->cachedIterator ??= new \ArrayObject(iterator_to_array($this->middlewareHandlers, false));
  50.                 }
  51.             };
  52.         }
  53.     }
  55.     public function dispatch(object $message, array $stamps = []): Envelope
  56.     {
  57.         $envelope = Envelope::wrap($message, $stamps);
  58.         $middlewareIterator = $this->middlewareAggregate->getIterator();
  60.         while ($middlewareIterator instanceof \IteratorAggregate) {
  61.             $middlewareIterator = $middlewareIterator->getIterator();
  62.         }
  63.         $middlewareIterator->rewind();
  65.         if (!$middlewareIterator->valid()) {
  66.             return $envelope;
  67.         }
  68.         $stack = new StackMiddleware($middlewareIterator);
  70.         return $middlewareIterator->current()->handle($envelope, $stack);
  71.     }
  72. }