vendor/symfony/messenger/DependencyInjection/MessengerPass.php line 120

Open in your IDE?
  1. <?php
  2. /*
  3.  * This file is part of the Symfony package.
  4.  *
  5.  * (c) Fabien Potencier <fabien@symfony.com>
  6.  *
  7.  * For the full copyright and license information, please view the LICENSE
  8.  * file that was distributed with this source code.
  9.  */
  10. namespace Symfony\Component\Messenger\DependencyInjection;
  11. use Symfony\Component\DependencyInjection\Argument\IteratorArgument;
  12. use Symfony\Component\DependencyInjection\ChildDefinition;
  13. use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
  14. use Symfony\Component\DependencyInjection\Compiler\ServiceLocatorTagPass;
  15. use Symfony\Component\DependencyInjection\ContainerBuilder;
  16. use Symfony\Component\DependencyInjection\Definition;
  17. use Symfony\Component\DependencyInjection\Exception\OutOfBoundsException;
  18. use Symfony\Component\DependencyInjection\Exception\RuntimeException;
  19. use Symfony\Component\DependencyInjection\Reference;
  20. use Symfony\Component\Messenger\Handler\HandlerDescriptor;
  21. use Symfony\Component\Messenger\Handler\HandlersLocator;
  22. use Symfony\Component\Messenger\Handler\MessageSubscriberInterface;
  23. use Symfony\Component\Messenger\TraceableMessageBus;
  24. use Symfony\Component\Messenger\Transport\Receiver\ReceiverInterface;
  25. /**
  26.  * @author Samuel Roze <samuel.roze@gmail.com>
  27.  */
  28. class MessengerPass implements CompilerPassInterface
  29. {
  30.     /**
  31.      * {@inheritdoc}
  32.      */
  33.     public function process(ContainerBuilder $container)
  34.     {
  35.         $busIds = [];
  36.         foreach ($container->findTaggedServiceIds('messenger.bus') as $busId => $tags) {
  37.             $busIds[] = $busId;
  38.             if ($container->hasParameter($busMiddlewareParameter $busId.'.middleware')) {
  39.                 $this->registerBusMiddleware($container$busId$container->getParameter($busMiddlewareParameter));
  40.                 $container->getParameterBag()->remove($busMiddlewareParameter);
  41.             }
  42.             if ($container->hasDefinition('data_collector.messenger')) {
  43.                 $this->registerBusToCollector($container$busId);
  44.             }
  45.         }
  46.         if ($container->hasDefinition('messenger.receiver_locator')) {
  47.             $this->registerReceivers($container$busIds);
  48.         }
  49.         $this->registerHandlers($container$busIds);
  50.     }
  51.     private function registerHandlers(ContainerBuilder $container, array $busIds)
  52.     {
  53.         $definitions = [];
  54.         $handlersByBusAndMessage = [];
  55.         $handlerToOriginalServiceIdMapping = [];
  56.         foreach ($container->findTaggedServiceIds('messenger.message_handler'true) as $serviceId => $tags) {
  57.             foreach ($tags as $tag) {
  58.                 if (isset($tag['bus']) && !\in_array($tag['bus'], $busIdstrue)) {
  59.                     throw new RuntimeException(sprintf('Invalid handler service "%s": bus "%s" specified on the tag "messenger.message_handler" does not exist (known ones are: "%s").'$serviceId$tag['bus'], implode('", "'$busIds)));
  60.                 }
  61.                 $className $this->getServiceClass($container$serviceId);
  62.                 $r $container->getReflectionClass($className);
  63.                 if (null === $r) {
  64.                     throw new RuntimeException(sprintf('Invalid service "%s": class "%s" does not exist.'$serviceId$className));
  65.                 }
  66.                 if (isset($tag['handles'])) {
  67.                     $handles = isset($tag['method']) ? [$tag['handles'] => $tag['method']] : [$tag['handles']];
  68.                 } else {
  69.                     $handles $this->guessHandledClasses($r$serviceId$tag['method'] ?? '__invoke');
  70.                 }
  71.                 $message null;
  72.                 $handlerBuses = (array) ($tag['bus'] ?? $busIds);
  73.                 foreach ($handles as $message => $options) {
  74.                     $buses $handlerBuses;
  75.                     if (\is_int($message)) {
  76.                         if (\is_string($options)) {
  77.                             $message $options;
  78.                             $options = [];
  79.                         } else {
  80.                             throw new RuntimeException(sprintf('The handler configuration needs to return an array of messages or an associated array of message and configuration. Found value of type "%s" at position "%d" for service "%s".'get_debug_type($options), $message$serviceId));
  81.                         }
  82.                     }
  83.                     if (\is_string($options)) {
  84.                         $options = ['method' => $options];
  85.                     }
  86.                     if (!isset($options['from_transport']) && isset($tag['from_transport'])) {
  87.                         $options['from_transport'] = $tag['from_transport'];
  88.                     }
  89.                     $priority $tag['priority'] ?? $options['priority'] ?? 0;
  90.                     $method $options['method'] ?? '__invoke';
  91.                     if (isset($options['bus'])) {
  92.                         if (!\in_array($options['bus'], $busIds)) {
  93.                             $messageLocation = isset($tag['handles']) ? 'declared in your tag attribute "handles"' : ($r->implementsInterface(MessageSubscriberInterface::class) ? sprintf('returned by method "%s::getHandledMessages()"'$r->getName()) : sprintf('used as argument type in method "%s::%s()"'$r->getName(), $method));
  94.                             throw new RuntimeException(sprintf('Invalid configuration "%s" for message "%s": bus "%s" does not exist.'$messageLocation$message$options['bus']));
  95.                         }
  96.                         $buses = [$options['bus']];
  97.                     }
  98.                     if ('*' !== $message && !class_exists($message) && !interface_exists($messagefalse)) {
  99.                         $messageLocation = isset($tag['handles']) ? 'declared in your tag attribute "handles"' : ($r->implementsInterface(MessageSubscriberInterface::class) ? sprintf('returned by method "%s::getHandledMessages()"'$r->getName()) : sprintf('used as argument type in method "%s::%s()"'$r->getName(), $method));
  100.                         throw new RuntimeException(sprintf('Invalid handler service "%s": class or interface "%s" "%s" not found.'$serviceId$message$messageLocation));
  101.                     }
  102.                     if (!$r->hasMethod($method)) {
  103.                         throw new RuntimeException(sprintf('Invalid handler service "%s": method "%s::%s()" does not exist.'$serviceId$r->getName(), $method));
  104.                     }
  105.                     if ('__invoke' !== $method) {
  106.                         $wrapperDefinition = (new Definition('callable'))->addArgument([new Reference($serviceId), $method])->setFactory('Closure::fromCallable');
  107.                         $definitions[$definitionId '.messenger.method_on_object_wrapper.'.ContainerBuilder::hash($message.':'.$priority.':'.$serviceId.':'.$method)] = $wrapperDefinition;
  108.                     } else {
  109.                         $definitionId $serviceId;
  110.                     }
  111.                     $handlerToOriginalServiceIdMapping[$definitionId] = $serviceId;
  112.                     foreach ($buses as $handlerBus) {
  113.                         $handlersByBusAndMessage[$handlerBus][$message][$priority][] = [$definitionId$options];
  114.                     }
  115.                 }
  116.                 if (null === $message) {
  117.                     throw new RuntimeException(sprintf('Invalid handler service "%s": method "%s::getHandledMessages()" must return one or more messages.'$serviceId$r->getName()));
  118.                 }
  119.             }
  120.         }
  121.         foreach ($handlersByBusAndMessage as $bus => $handlersByMessage) {
  122.             foreach ($handlersByMessage as $message => $handlersByPriority) {
  123.                 krsort($handlersByPriority);
  124.                 $handlersByBusAndMessage[$bus][$message] = array_merge(...$handlersByPriority);
  125.             }
  126.         }
  127.         $handlersLocatorMappingByBus = [];
  128.         foreach ($handlersByBusAndMessage as $bus => $handlersByMessage) {
  129.             foreach ($handlersByMessage as $message => $handlers) {
  130.                 $handlerDescriptors = [];
  131.                 foreach ($handlers as $handler) {
  132.                     $definitions[$definitionId '.messenger.handler_descriptor.'.ContainerBuilder::hash($bus.':'.$message.':'.$handler[0])] = (new Definition(HandlerDescriptor::class))->setArguments([new Reference($handler[0]), $handler[1]]);
  133.                     $handlerDescriptors[] = new Reference($definitionId);
  134.                 }
  135.                 $handlersLocatorMappingByBus[$bus][$message] = new IteratorArgument($handlerDescriptors);
  136.             }
  137.         }
  138.         $container->addDefinitions($definitions);
  139.         foreach ($busIds as $bus) {
  140.             $container->register($locatorId $bus.'.messenger.handlers_locator'HandlersLocator::class)
  141.                 ->setArgument(0$handlersLocatorMappingByBus[$bus] ?? [])
  142.             ;
  143.             if ($container->has($handleMessageId $bus.'.middleware.handle_message')) {
  144.                 $container->getDefinition($handleMessageId)
  145.                     ->replaceArgument(0, new Reference($locatorId))
  146.                 ;
  147.             }
  148.         }
  149.         if ($container->hasDefinition('console.command.messenger_debug')) {
  150.             $debugCommandMapping $handlersByBusAndMessage;
  151.             foreach ($busIds as $bus) {
  152.                 if (!isset($debugCommandMapping[$bus])) {
  153.                     $debugCommandMapping[$bus] = [];
  154.                 }
  155.                 foreach ($debugCommandMapping[$bus] as $message => $handlers) {
  156.                     foreach ($handlers as $key => $handler) {
  157.                         $debugCommandMapping[$bus][$message][$key][0] = $handlerToOriginalServiceIdMapping[$handler[0]];
  158.                     }
  159.                 }
  160.             }
  161.             $container->getDefinition('console.command.messenger_debug')->replaceArgument(0$debugCommandMapping);
  162.         }
  163.     }
  164.     private function guessHandledClasses(\ReflectionClass $handlerClassstring $serviceIdstring $methodName): iterable
  165.     {
  166.         if ($handlerClass->implementsInterface(MessageSubscriberInterface::class)) {
  167.             return $handlerClass->getName()::getHandledMessages();
  168.         }
  169.         try {
  170.             $method $handlerClass->getMethod($methodName);
  171.         } catch (\ReflectionException) {
  172.             throw new RuntimeException(sprintf('Invalid handler service "%s": class "%s" must have an "%s()" method.'$serviceId$handlerClass->getName(), $methodName));
  173.         }
  174.         if (=== $method->getNumberOfRequiredParameters()) {
  175.             throw new RuntimeException(sprintf('Invalid handler service "%s": method "%s::%s()" requires at least one argument, first one being the message it handles.'$serviceId$handlerClass->getName(), $methodName));
  176.         }
  177.         $parameters $method->getParameters();
  178.         /** @var \ReflectionNamedType|\ReflectionUnionType|null */
  179.         $type $parameters[0]->getType();
  180.         if (!$type) {
  181.             throw new RuntimeException(sprintf('Invalid handler service "%s": argument "$%s" of method "%s::%s()" must have a type-hint corresponding to the message class it handles.'$serviceId$parameters[0]->getName(), $handlerClass->getName(), $methodName));
  182.         }
  183.         if ($type instanceof \ReflectionUnionType) {
  184.             $types = [];
  185.             $invalidTypes = [];
  186.             foreach ($type->getTypes() as $type) {
  187.                 if (!$type->isBuiltin()) {
  188.                     $types[] = (string) $type;
  189.                 } else {
  190.                     $invalidTypes[] = (string) $type;
  191.                 }
  192.             }
  193.             if ($types) {
  194.                 return ('__invoke' === $methodName) ? $types array_fill_keys($types$methodName);
  195.             }
  196.             throw new RuntimeException(sprintf('Invalid handler service "%s": type-hint of argument "$%s" in method "%s::__invoke()" must be a class , "%s" given.'$serviceId$parameters[0]->getName(), $handlerClass->getName(), implode('|'$invalidTypes)));
  197.         }
  198.         if ($type->isBuiltin()) {
  199.             throw new RuntimeException(sprintf('Invalid handler service "%s": type-hint of argument "$%s" in method "%s::%s()" must be a class , "%s" given.'$serviceId$parameters[0]->getName(), $handlerClass->getName(), $methodName$type instanceof \ReflectionNamedType $type->getName() : (string) $type));
  200.         }
  201.         return ('__invoke' === $methodName) ? [$type->getName()] : [$type->getName() => $methodName];
  202.     }
  203.     private function registerReceivers(ContainerBuilder $container, array $busIds)
  204.     {
  205.         $receiverMapping = [];
  206.         $failureTransportsMap = [];
  207.         if ($container->hasDefinition('console.command.messenger_failed_messages_retry')) {
  208.             $commandDefinition $container->getDefinition('console.command.messenger_failed_messages_retry');
  209.             $globalReceiverName $commandDefinition->getArgument(0);
  210.             if (null !== $globalReceiverName) {
  211.                 if ($container->hasAlias('messenger.failure_transports.default')) {
  212.                     $failureTransportsMap[$globalReceiverName] = new Reference('messenger.failure_transports.default');
  213.                 } else {
  214.                     $failureTransportsMap[$globalReceiverName] = new Reference('messenger.transport.'.$globalReceiverName);
  215.                 }
  216.             }
  217.         }
  218.         foreach ($container->findTaggedServiceIds('messenger.receiver') as $id => $tags) {
  219.             $receiverClass $this->getServiceClass($container$id);
  220.             if (!is_subclass_of($receiverClassReceiverInterface::class)) {
  221.                 throw new RuntimeException(sprintf('Invalid receiver "%s": class "%s" must implement interface "%s".'$id$receiverClassReceiverInterface::class));
  222.             }
  223.             $receiverMapping[$id] = new Reference($id);
  224.             foreach ($tags as $tag) {
  225.                 if (isset($tag['alias'])) {
  226.                     $receiverMapping[$tag['alias']] = $receiverMapping[$id];
  227.                     if ($tag['is_failure_transport'] ?? false) {
  228.                         $failureTransportsMap[$tag['alias']] = $receiverMapping[$id];
  229.                     }
  230.                 }
  231.             }
  232.         }
  233.         $receiverNames = [];
  234.         foreach ($receiverMapping as $name => $reference) {
  235.             $receiverNames[(string) $reference] = $name;
  236.         }
  237.         $buses = [];
  238.         foreach ($busIds as $busId) {
  239.             $buses[$busId] = new Reference($busId);
  240.         }
  241.         if ($hasRoutableMessageBus $container->hasDefinition('messenger.routable_message_bus')) {
  242.             $container->getDefinition('messenger.routable_message_bus')
  243.                 ->replaceArgument(0ServiceLocatorTagPass::register($container$buses));
  244.         }
  245.         if ($container->hasDefinition('console.command.messenger_consume_messages')) {
  246.             $consumeCommandDefinition $container->getDefinition('console.command.messenger_consume_messages');
  247.             if ($hasRoutableMessageBus) {
  248.                 $consumeCommandDefinition->replaceArgument(0, new Reference('messenger.routable_message_bus'));
  249.             }
  250.             $consumeCommandDefinition->replaceArgument(4array_values($receiverNames));
  251.             try {
  252.                 $consumeCommandDefinition->replaceArgument(6$busIds);
  253.             } catch (OutOfBoundsException) {
  254.                 // ignore to preserve compatibility with symfony/framework-bundle < 5.4
  255.             }
  256.         }
  257.         if ($container->hasDefinition('console.command.messenger_setup_transports')) {
  258.             $container->getDefinition('console.command.messenger_setup_transports')
  259.                 ->replaceArgument(1array_values($receiverNames));
  260.         }
  261.         $container->getDefinition('messenger.receiver_locator')->replaceArgument(0$receiverMapping);
  262.         $failureTransportsLocator ServiceLocatorTagPass::register($container$failureTransportsMap);
  263.         $failedCommandIds = [
  264.             'console.command.messenger_failed_messages_retry',
  265.             'console.command.messenger_failed_messages_show',
  266.             'console.command.messenger_failed_messages_remove',
  267.         ];
  268.         foreach ($failedCommandIds as $failedCommandId) {
  269.             if ($container->hasDefinition($failedCommandId)) {
  270.                 $definition $container->getDefinition($failedCommandId);
  271.                 $definition->replaceArgument(1$failureTransportsLocator);
  272.             }
  273.         }
  274.     }
  275.     private function registerBusToCollector(ContainerBuilder $containerstring $busId)
  276.     {
  277.         $container->setDefinition(
  278.             $tracedBusId 'debug.traced.'.$busId,
  279.             (new Definition(TraceableMessageBus::class, [new Reference($tracedBusId.'.inner')]))->setDecoratedService($busId)
  280.         );
  281.         $container->getDefinition('data_collector.messenger')->addMethodCall('registerBus', [$busId, new Reference($tracedBusId)]);
  282.     }
  283.     private function registerBusMiddleware(ContainerBuilder $containerstring $busId, array $middlewareCollection)
  284.     {
  285.         $middlewareReferences = [];
  286.         foreach ($middlewareCollection as $middlewareItem) {
  287.             $id $middlewareItem['id'];
  288.             $arguments $middlewareItem['arguments'] ?? [];
  289.             if (!$container->has($messengerMiddlewareId 'messenger.middleware.'.$id)) {
  290.                 $messengerMiddlewareId $id;
  291.             }
  292.             if (!$container->has($messengerMiddlewareId)) {
  293.                 throw new RuntimeException(sprintf('Invalid middleware: service "%s" not found.'$id));
  294.             }
  295.             if ($container->findDefinition($messengerMiddlewareId)->isAbstract()) {
  296.                 $childDefinition = new ChildDefinition($messengerMiddlewareId);
  297.                 $childDefinition->setArguments($arguments);
  298.                 if (isset($middlewareReferences[$messengerMiddlewareId $busId.'.middleware.'.$id])) {
  299.                     $messengerMiddlewareId .= '.'.ContainerBuilder::hash($arguments);
  300.                 }
  301.                 $container->setDefinition($messengerMiddlewareId$childDefinition);
  302.             } elseif ($arguments) {
  303.                 throw new RuntimeException(sprintf('Invalid middleware factory "%s": a middleware factory must be an abstract definition.'$id));
  304.             }
  305.             $middlewareReferences[$messengerMiddlewareId] = new Reference($messengerMiddlewareId);
  306.         }
  307.         $container->getDefinition($busId)->replaceArgument(0, new IteratorArgument(array_values($middlewareReferences)));
  308.     }
  309.     private function getServiceClass(ContainerBuilder $containerstring $serviceId): string
  310.     {
  311.         while (true) {
  312.             $definition $container->findDefinition($serviceId);
  313.             if (!$definition->getClass() && $definition instanceof ChildDefinition) {
  314.                 $serviceId $definition->getParent();
  315.                 continue;
  316.             }
  317.             return $definition->getClass();
  318.         }
  319.     }
  320. }