599f32c085
Ensure that the middlewares can also update the message within the envelope
78 lines
2.4 KiB
PHP
78 lines
2.4 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\Messenger;
|
|
|
|
use Symfony\Component\Messenger\Exception\InvalidArgumentException;
|
|
use Symfony\Component\Messenger\Middleware\MiddlewareInterface;
|
|
|
|
/**
|
|
* @author Samuel Roze <samuel.roze@gmail.com>
|
|
* @author Matthias Noback <matthiasnoback@gmail.com>
|
|
*/
|
|
class MessageBus implements MessageBusInterface
|
|
{
|
|
private $middlewareHandlers;
|
|
|
|
/**
|
|
* @var MiddlewareInterface[]|null
|
|
*/
|
|
private $indexedMiddlewareHandlers;
|
|
|
|
/**
|
|
* @param MiddlewareInterface[]|iterable $middlewareHandlers
|
|
*/
|
|
public function __construct(iterable $middlewareHandlers = array())
|
|
{
|
|
$this->middlewareHandlers = $middlewareHandlers;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function dispatch($message)
|
|
{
|
|
if (!\is_object($message)) {
|
|
throw new InvalidArgumentException(sprintf('Invalid type for message argument. Expected object, but got "%s".', \gettype($message)));
|
|
}
|
|
|
|
return \call_user_func($this->callableForNextMiddleware(0, Envelope::wrap($message)), $message);
|
|
}
|
|
|
|
private function callableForNextMiddleware(int $index, Envelope $currentEnvelope): callable
|
|
{
|
|
if (null === $this->indexedMiddlewareHandlers) {
|
|
$this->indexedMiddlewareHandlers = \is_array($this->middlewareHandlers) ? array_values($this->middlewareHandlers) : iterator_to_array($this->middlewareHandlers, false);
|
|
}
|
|
|
|
if (!isset($this->indexedMiddlewareHandlers[$index])) {
|
|
return function () {};
|
|
}
|
|
|
|
$middleware = $this->indexedMiddlewareHandlers[$index];
|
|
|
|
return function ($message) use ($middleware, $index, $currentEnvelope) {
|
|
if ($message instanceof Envelope) {
|
|
$currentEnvelope = $message;
|
|
} else {
|
|
$message = $currentEnvelope->withMessage($message);
|
|
}
|
|
|
|
if (!$middleware instanceof EnvelopeAwareInterface) {
|
|
// Do not provide the envelope if the middleware cannot read it:
|
|
$message = $message->getMessage();
|
|
}
|
|
|
|
return $middleware->handle($message, $this->callableForNextMiddleware($index + 1, $currentEnvelope));
|
|
};
|
|
}
|
|
}
|