Passed
Pull Request — master (#753)
by Maxim
17:58
created

QueueBootloader::initQueueManager()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 3
Code Lines 1

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 2
CRAP Score 1

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 1
eloc 1
c 1
b 0
f 0
nc 1
nop 1
dl 0
loc 3
ccs 2
cts 2
cp 1
crap 1
rs 10
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