/
SendersLocator.php
77 lines (64 loc) · 2.5 KB
/
SendersLocator.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
<?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\Transport\Sender;
use Psr\Container\ContainerInterface;
use Symfony\Component\Messenger\Envelope;
use Symfony\Component\Messenger\Exception\RuntimeException;
use Symfony\Component\Messenger\Handler\HandlersLocator;
use Symfony\Component\Messenger\Stamp\TransportNamesStamp;
/**
* Maps a message to a list of senders.
*
* @author Fabien Potencier <fabien@symfony.com>
*/
class SendersLocator implements SendersLocatorInterface
{
private array $sendersMap;
private ContainerInterface $sendersLocator;
/**
* @param array<string, list<string>> $sendersMap An array, keyed by "type", set to an array of sender aliases
* @param ContainerInterface $sendersLocator Locator of senders, keyed by sender alias
*/
public function __construct(array $sendersMap, ContainerInterface $sendersLocator)
{
$this->sendersMap = $sendersMap;
$this->sendersLocator = $sendersLocator;
}
public function getSenders(Envelope $envelope): iterable
{
if ($envelope->all(TransportNamesStamp::class)) {
foreach ($envelope->last(TransportNamesStamp::class)->getTransportNames() as $senderAlias) {
yield from $this->getSenderFromAlias($senderAlias);
}
return;
}
$seen = [];
foreach (HandlersLocator::listTypes($envelope) as $type) {
if (str_ends_with($type, '*') && $seen) {
// the '*' acts as a fallback, if other senders already matched
// with previous types, skip the senders bound to the fallback
continue;
}
foreach ($this->sendersMap[$type] ?? [] as $senderAlias) {
if (!\in_array($senderAlias, $seen, true)) {
$seen[] = $senderAlias;
yield from $this->getSenderFromAlias($senderAlias);
}
}
}
}
private function getSenderFromAlias(string $senderAlias): iterable
{
if (!$this->sendersLocator->has($senderAlias)) {
throw new RuntimeException(sprintf('Invalid senders configuration: sender "%s" is not in the senders locator.', $senderAlias));
}
yield $senderAlias => $this->sendersLocator->get($senderAlias);
}
}