1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
declare(strict_types=1); |
4
|
|
|
|
5
|
|
|
namespace Spiral\Queue\Bootloader; |
6
|
|
|
|
7
|
|
|
use Psr\Container\ContainerInterface; |
8
|
|
|
use Spiral\Boot\AbstractKernel; |
9
|
|
|
use Spiral\Boot\Bootloader\Bootloader; |
10
|
|
|
use Spiral\Boot\EnvironmentInterface; |
11
|
|
|
use Spiral\Config\ConfiguratorInterface; |
12
|
|
|
use Spiral\Config\Patch\Append; |
13
|
|
|
use Spiral\Core\Container; |
14
|
|
|
use Spiral\Core\FactoryInterface; |
15
|
|
|
use Spiral\Queue\Config\QueueConfig; |
16
|
|
|
use Spiral\Queue\ContainerRegistry; |
17
|
|
|
use Spiral\Queue\Core\QueueInjector; |
18
|
|
|
use Spiral\Queue\Driver\NullDriver; |
19
|
|
|
use Spiral\Queue\Driver\SyncDriver; |
20
|
|
|
use Spiral\Queue\Failed\FailedJobHandlerInterface; |
21
|
|
|
use Spiral\Queue\Failed\LogFailedJobHandler; |
22
|
|
|
use Spiral\Queue\HandlerRegistryInterface; |
23
|
|
|
use Spiral\Queue\QueueConnectionProviderInterface; |
24
|
|
|
use Spiral\Queue\QueueInterface; |
25
|
|
|
use Spiral\Queue\QueueManager; |
26
|
|
|
use Spiral\Queue\QueueRegistry; |
27
|
|
|
use Spiral\Queue\SerializerInterface; |
28
|
|
|
use Spiral\Queue\SerializerRegistry; |
29
|
|
|
use Spiral\Queue\SerializerRegistryInterface; |
30
|
|
|
|
31
|
|
|
final class QueueBootloader extends Bootloader |
32
|
|
|
{ |
33
|
|
|
protected const SINGLETONS = [ |
34
|
|
|
HandlerRegistryInterface::class => QueueRegistry::class, |
35
|
|
|
FailedJobHandlerInterface::class => LogFailedJobHandler::class, |
36
|
|
|
QueueConnectionProviderInterface::class => QueueManager::class, |
37
|
|
|
SerializerRegistryInterface::class => SerializerRegistry::class, |
38
|
|
|
SerializerInterface::class => SerializerRegistryInterface::class, |
39
|
|
|
QueueManager::class => [self::class, 'initQueueManager'], |
40
|
|
|
QueueRegistry::class => [self::class, 'initRegistry'], |
41
|
|
|
SerializerRegistry::class => [self::class, 'initSerializerRegistry'], |
42
|
|
|
]; |
43
|
|
|
|
44
|
|
|
private ConfiguratorInterface $config; |
45
|
|
|
|
46
|
207 |
|
public function __construct(ConfiguratorInterface $config) |
47
|
|
|
{ |
48
|
207 |
|
$this->config = $config; |
49
|
|
|
} |
50
|
|
|
|
51
|
207 |
|
public function boot(Container $container, EnvironmentInterface $env, AbstractKernel $kernel): void |
52
|
|
|
{ |
53
|
207 |
|
$this->initQueueConfig($env); |
54
|
207 |
|
$this->registerQueue($container); |
55
|
|
|
|
56
|
207 |
|
$this->registerDriverAlias(SyncDriver::class, 'sync'); |
57
|
207 |
|
$container->bindInjector(QueueInterface::class, QueueInjector::class); |
58
|
|
|
|
59
|
207 |
|
$kernel->started(static function () use ($container): void { |
60
|
207 |
|
$registry = $container->get(HandlerRegistryInterface::class); |
61
|
207 |
|
$config = $container->get(QueueConfig::class); |
62
|
207 |
|
$serializersRegistry = $container->get(SerializerRegistryInterface::class); |
63
|
|
|
|
64
|
207 |
|
foreach ($config->getRegistryHandlers() as $jobType => $handler) { |
65
|
|
|
$registry->setHandler($jobType, $handler); |
66
|
|
|
} |
67
|
|
|
|
68
|
207 |
|
foreach ($config->getRegistrySerializers() as $jobType => $serializer) { |
69
|
|
|
if ($serializer instanceof Container\Autowire || \is_string($serializer)) { |
70
|
|
|
$serializer = $container->get($serializer); |
71
|
|
|
} |
72
|
|
|
$serializersRegistry->addSerializer($jobType, $serializer); |
73
|
|
|
} |
74
|
|
|
}); |
75
|
|
|
} |
76
|
|
|
|
77
|
207 |
|
public function registerDriverAlias(string $driverClass, string $alias): void |
78
|
|
|
{ |
79
|
207 |
|
$this->config->modify( |
80
|
|
|
QueueConfig::CONFIG, |
81
|
207 |
|
new Append('driverAliases', $alias, $driverClass) |
82
|
|
|
); |
83
|
|
|
} |
84
|
|
|
|
85
|
2 |
|
protected function initQueueManager(FactoryInterface $factory): QueueManager |
86
|
|
|
{ |
87
|
2 |
|
return $factory->make(QueueManager::class); |
88
|
|
|
} |
89
|
|
|
|
90
|
207 |
|
protected function initRegistry(ContainerInterface $container, ContainerRegistry $registry) |
91
|
|
|
{ |
92
|
207 |
|
return new QueueRegistry($container, $registry); |
93
|
|
|
} |
94
|
|
|
|
95
|
207 |
|
protected function initSerializerRegistry(Container $container, QueueConfig $config): SerializerRegistry |
96
|
|
|
{ |
97
|
207 |
|
$default = $config->getDefaultSerializer(); |
98
|
|
|
|
99
|
207 |
|
if ($default instanceof Container\Autowire || \is_string($default)) { |
100
|
|
|
$default = $container->get($default); |
101
|
|
|
} |
102
|
|
|
|
103
|
207 |
|
return new SerializerRegistry($default); |
104
|
|
|
} |
105
|
|
|
|
106
|
207 |
|
private function registerQueue(Container $container): void |
107
|
|
|
{ |
108
|
207 |
|
$container->bindSingleton( |
109
|
|
|
QueueInterface::class, |
110
|
207 |
|
static fn (QueueManager $manager): QueueInterface => $manager->getConnection() |
111
|
|
|
); |
112
|
|
|
} |
113
|
|
|
|
114
|
207 |
|
private function initQueueConfig(EnvironmentInterface $env): void |
115
|
|
|
{ |
116
|
207 |
|
$this->config->setDefaults( |
117
|
|
|
QueueConfig::CONFIG, |
118
|
|
|
[ |
119
|
207 |
|
'default' => $env->get('QUEUE_CONNECTION', 'sync'), |
120
|
|
|
'connections' => [ |
121
|
|
|
'sync' => [ |
122
|
|
|
'driver' => 'sync', |
123
|
|
|
], |
124
|
|
|
], |
125
|
|
|
'registry' => [ |
126
|
|
|
'handlers' => [], |
127
|
|
|
'serializers' => [], |
128
|
|
|
], |
129
|
|
|
'driverAliases' => [ |
130
|
|
|
'sync' => SyncDriver::class, |
131
|
|
|
'null' => NullDriver::class, |
132
|
|
|
], |
133
|
|
|
] |
134
|
|
|
); |
135
|
|
|
} |
136
|
|
|
} |
137
|
|
|
|