1 | <?php |
||
2 | |||
3 | namespace OldSound\RabbitMqBundle\DependencyInjection; |
||
4 | |||
5 | use OldSound\RabbitMqBundle\RabbitMq\ConsumerInterface; |
||
6 | use OldSound\RabbitMqBundle\RabbitMq\ProducerInterface; |
||
7 | use Symfony\Component\Config\Definition\ConfigurationInterface; |
||
8 | use Symfony\Component\Config\Definition\Exception\InvalidConfigurationException; |
||
9 | use Symfony\Component\Config\FileLocator; |
||
10 | use Symfony\Component\DependencyInjection\ContainerBuilder; |
||
11 | use Symfony\Component\DependencyInjection\ContainerInterface; |
||
12 | use Symfony\Component\DependencyInjection\Definition; |
||
13 | use Symfony\Component\DependencyInjection\Extension\Extension; |
||
14 | use Symfony\Component\DependencyInjection\Loader\XmlFileLoader; |
||
15 | use Symfony\Component\DependencyInjection\Reference; |
||
16 | |||
17 | /** |
||
18 | * OldSoundRabbitMqExtension. |
||
19 | * |
||
20 | * @author Alvaro Videla |
||
21 | * @author Marc Weistroff <[email protected]> |
||
22 | */ |
||
23 | class OldSoundRabbitMqExtension extends Extension |
||
24 | { |
||
25 | /** |
||
26 | * @var ContainerBuilder |
||
27 | */ |
||
28 | private $container; |
||
29 | |||
30 | /** |
||
31 | * @var Boolean Whether the data collector is enabled |
||
32 | */ |
||
33 | private $collectorEnabled; |
||
34 | |||
35 | private $channelIds = []; |
||
36 | |||
37 | private $config = []; |
||
38 | |||
39 | 34 | public function load(array $configs, ContainerBuilder $container): void |
|
40 | { |
||
41 | 34 | $this->container = $container; |
|
42 | |||
43 | 34 | $loader = new XmlFileLoader($this->container, new FileLocator([__DIR__ . '/../Resources/config'])); |
|
44 | 34 | $loader->load('rabbitmq.xml'); |
|
45 | |||
46 | 34 | $configuration = $this->getConfiguration($configs, $container); |
|
47 | 34 | $this->config = $this->processConfiguration($configuration, $configs); |
|
48 | |||
49 | 34 | $this->collectorEnabled = $this->config['enable_collector']; |
|
50 | |||
51 | 34 | $this->loadConnections(); |
|
52 | 34 | $this->loadBindings(); |
|
53 | 34 | $this->loadProducers(); |
|
54 | 34 | $this->loadConsumers(); |
|
55 | 34 | $this->loadMultipleConsumers(); |
|
56 | 34 | $this->loadDynamicConsumers(); |
|
57 | 34 | $this->loadBatchConsumers(); |
|
58 | 34 | $this->loadAnonConsumers(); |
|
59 | 34 | $this->loadRpcClients(); |
|
60 | 34 | $this->loadRpcServers(); |
|
61 | |||
62 | 34 | if ($this->collectorEnabled && $this->channelIds) { |
|
0 ignored issues
–
show
|
|||
63 | 31 | $channels = []; |
|
64 | 31 | foreach (array_unique($this->channelIds) as $id) { |
|
65 | 31 | $channels[] = new Reference($id); |
|
66 | } |
||
67 | |||
68 | 31 | $definition = $container->getDefinition('old_sound_rabbit_mq.data_collector'); |
|
69 | 31 | $definition->replaceArgument(0, $channels); |
|
70 | } else { |
||
71 | 3 | $this->container->removeDefinition('old_sound_rabbit_mq.data_collector'); |
|
72 | } |
||
73 | 34 | } |
|
74 | |||
75 | 34 | public function getConfiguration(array $config, ContainerBuilder $container): ?ConfigurationInterface |
|
76 | { |
||
77 | 34 | return new Configuration($this->getAlias()); |
|
78 | } |
||
79 | |||
80 | 34 | protected function loadConnections() |
|
81 | { |
||
82 | 34 | foreach ($this->config['connections'] as $key => $connection) { |
|
83 | 34 | $connectionSuffix = $connection['use_socket'] ? 'socket_connection.class' : 'connection.class'; |
|
84 | 34 | $classParam = |
|
85 | 34 | $connection['lazy'] |
|
86 | 25 | ? '%old_sound_rabbit_mq.lazy.'.$connectionSuffix.'%' |
|
87 | 34 | : '%old_sound_rabbit_mq.'.$connectionSuffix.'%'; |
|
88 | |||
89 | 34 | $definition = new Definition('%old_sound_rabbit_mq.connection_factory.class%', [ |
|
90 | 34 | $classParam, $connection, |
|
91 | ]); |
||
92 | 34 | if (isset($connection['connection_parameters_provider'])) { |
|
93 | $definition->addArgument(new Reference($connection['connection_parameters_provider'])); |
||
94 | unset($connection['connection_parameters_provider']); |
||
95 | } |
||
96 | 34 | $definition->setPublic(false); |
|
97 | 34 | $factoryName = sprintf('old_sound_rabbit_mq.connection_factory.%s', $key); |
|
98 | 34 | $this->container->setDefinition($factoryName, $definition); |
|
99 | |||
100 | 34 | $definition = new Definition($classParam); |
|
101 | 34 | if (method_exists($definition, 'setFactory')) { |
|
102 | // to be inlined in services.xml when dependency on Symfony DependencyInjection is bumped to 2.6 |
||
103 | 34 | $definition->setFactory([new Reference($factoryName), 'createConnection']); |
|
104 | } else { |
||
105 | // to be removed when dependency on Symfony DependencyInjection is bumped to 2.6 |
||
106 | $definition->setFactoryService($factoryName); |
||
107 | $definition->setFactoryMethod('createConnection'); |
||
108 | } |
||
109 | 34 | $definition->addTag('old_sound_rabbit_mq.connection'); |
|
110 | 34 | $definition->setPublic(true); |
|
111 | |||
112 | 34 | $this->container->setDefinition(sprintf('old_sound_rabbit_mq.connection.%s', $key), $definition); |
|
113 | } |
||
114 | 34 | } |
|
115 | |||
116 | 34 | protected function loadBindings() |
|
117 | { |
||
118 | 34 | if ($this->config['sandbox']) { |
|
119 | return; |
||
120 | } |
||
121 | 34 | foreach ($this->config['bindings'] as $binding) { |
|
122 | 25 | ksort($binding); |
|
123 | 25 | $definition = new Definition($binding['class']); |
|
124 | 25 | $definition->addTag('old_sound_rabbit_mq.binding'); |
|
125 | 25 | $definition->addMethodCall('setArguments', [$binding['arguments']]); |
|
126 | 25 | $definition->addMethodCall('setDestination', [$binding['destination']]); |
|
127 | 25 | $definition->addMethodCall('setDestinationIsExchange', [$binding['destination_is_exchange']]); |
|
128 | 25 | $definition->addMethodCall('setExchange', [$binding['exchange']]); |
|
129 | 25 | $definition->addMethodCall('isNowait', [$binding['nowait']]); |
|
130 | 25 | $definition->addMethodCall('setRoutingKey', [$binding['routing_key']]); |
|
131 | 25 | $this->injectConnection($definition, $binding['connection']); |
|
132 | 25 | $key = md5(json_encode($binding)); |
|
133 | 25 | if ($this->collectorEnabled) { |
|
134 | // in the context of a binding, I don't thing logged channels are needed? |
||
135 | 25 | $this->injectLoggedChannel($definition, $key, $binding['connection']); |
|
136 | } |
||
137 | 25 | $this->container->setDefinition(sprintf('old_sound_rabbit_mq.binding.%s', $key), $definition); |
|
138 | } |
||
139 | 34 | } |
|
140 | |||
141 | 34 | protected function loadProducers() |
|
142 | { |
||
143 | 34 | if ($this->config['sandbox'] == false) { |
|
144 | 34 | foreach ($this->config['producers'] as $key => $producer) { |
|
145 | 30 | $definition = new Definition($producer['class']); |
|
146 | 30 | $definition->setPublic(true); |
|
147 | 30 | $definition->addTag('old_sound_rabbit_mq.base_amqp'); |
|
148 | 30 | $definition->addTag('old_sound_rabbit_mq.producer'); |
|
149 | //this producer doesn't define an exchange -> using AMQP Default |
||
150 | 30 | if (!isset($producer['exchange_options'])) { |
|
151 | 1 | $producer['exchange_options'] = $this->getDefaultExchangeOptions(); |
|
152 | } |
||
153 | 30 | $definition->addMethodCall('setExchangeOptions', [$this->normalizeArgumentKeys($producer['exchange_options'])]); |
|
154 | //this producer doesn't define a queue -> using AMQP Default |
||
155 | 30 | if (!isset($producer['queue_options'])) { |
|
156 | 30 | $producer['queue_options'] = $this->getDefaultQueueOptions(); |
|
157 | } |
||
158 | 30 | $definition->addMethodCall('setQueueOptions', [$producer['queue_options']]); |
|
159 | 30 | $this->injectConnection($definition, $producer['connection']); |
|
160 | 30 | if ($this->collectorEnabled) { |
|
161 | 28 | $this->injectLoggedChannel($definition, $key, $producer['connection']); |
|
162 | } |
||
163 | 30 | if (!$producer['auto_setup_fabric']) { |
|
164 | $definition->addMethodCall('disableAutoSetupFabric'); |
||
165 | } |
||
166 | |||
167 | 30 | if ($producer['enable_logger']) { |
|
168 | $this->injectLogger($definition); |
||
169 | } |
||
170 | |||
171 | 30 | $producerServiceName = sprintf('old_sound_rabbit_mq.%s_producer', $key); |
|
172 | |||
173 | 30 | $this->container->setDefinition($producerServiceName, $definition); |
|
174 | 30 | if (null !== $producer['service_alias']) { |
|
175 | 25 | $this->container->setAlias($producer['service_alias'], $producerServiceName); |
|
176 | } |
||
177 | |||
178 | // register alias for argument auto wiring |
||
179 | 30 | if (method_exists($this->container, 'registerAliasForArgument')) { |
|
180 | 30 | $argName = !str_ends_with(strtolower($key), 'producer') ? sprintf('%sProducer', $key) : $key; |
|
181 | 30 | $this->container |
|
182 | 30 | ->registerAliasForArgument($producerServiceName, ProducerInterface::class, $argName) |
|
183 | 30 | ->setPublic(false); |
|
184 | |||
185 | 30 | $this->container |
|
186 | 30 | ->registerAliasForArgument($producerServiceName, $producer['class'], $argName) |
|
187 | 30 | ->setPublic(false); |
|
188 | } |
||
189 | |||
190 | 30 | $definition->addMethodCall('setDefaultRoutingKey', [$producer['default_routing_key']]); |
|
191 | 30 | $definition->addMethodCall('setContentType', [$producer['default_content_type']]); |
|
192 | 30 | $definition->addMethodCall('setDeliveryMode', [$producer['default_delivery_mode']]); |
|
193 | } |
||
194 | } else { |
||
195 | foreach ($this->config['producers'] as $key => $producer) { |
||
196 | $definition = new Definition('%old_sound_rabbit_mq.fallback.class%'); |
||
197 | $producerServiceName = sprintf('old_sound_rabbit_mq.%s_producer', $key); |
||
198 | $this->container->setDefinition($producerServiceName, $definition); |
||
199 | |||
200 | // register alias for argumen auto wiring |
||
201 | if (method_exists($this->container, 'registerAliasForArgument')) { |
||
202 | $argName = !str_ends_with(strtolower($key), 'producer') ? sprintf('%sProducer', $key) : $key; |
||
203 | $this->container |
||
204 | ->registerAliasForArgument($producerServiceName, ProducerInterface::class, $argName) |
||
205 | ->setPublic(false); |
||
206 | } |
||
207 | } |
||
208 | } |
||
209 | 34 | } |
|
210 | |||
211 | 34 | protected function loadConsumers() |
|
212 | { |
||
213 | 34 | foreach ($this->config['consumers'] as $key => $consumer) { |
|
214 | 29 | $definition = new Definition('%old_sound_rabbit_mq.consumer.class%'); |
|
215 | 29 | $definition->setPublic(true); |
|
216 | 29 | $definition->addTag('old_sound_rabbit_mq.base_amqp'); |
|
217 | 29 | $definition->addTag('old_sound_rabbit_mq.consumer'); |
|
218 | //this consumer doesn't define an exchange -> using AMQP Default |
||
219 | 29 | if (!isset($consumer['exchange_options'])) { |
|
220 | $consumer['exchange_options'] = $this->getDefaultExchangeOptions(); |
||
221 | } |
||
222 | 29 | $definition->addMethodCall('setExchangeOptions', [$this->normalizeArgumentKeys($consumer['exchange_options'])]); |
|
223 | //this consumer doesn't define a queue -> using AMQP Default |
||
224 | 29 | if (!isset($consumer['queue_options'])) { |
|
225 | $consumer['queue_options'] = $this->getDefaultQueueOptions(); |
||
226 | } |
||
227 | 29 | $definition->addMethodCall('setQueueOptions', [$this->normalizeArgumentKeys($consumer['queue_options'])]); |
|
228 | 29 | $definition->addMethodCall('setCallback', [[new Reference($consumer['callback']), 'execute']]); |
|
229 | |||
230 | 29 | if (array_key_exists('qos_options', $consumer)) { |
|
231 | 25 | $definition->addMethodCall('setQosOptions', [ |
|
232 | 25 | $consumer['qos_options']['prefetch_size'], |
|
233 | 25 | $consumer['qos_options']['prefetch_count'], |
|
234 | 25 | $consumer['qos_options']['global'], |
|
235 | ]); |
||
236 | } |
||
237 | |||
238 | 29 | if (isset($consumer['idle_timeout'])) { |
|
239 | $definition->addMethodCall('setIdleTimeout', [$consumer['idle_timeout']]); |
||
240 | } |
||
241 | 29 | if (isset($consumer['idle_timeout_exit_code'])) { |
|
242 | $definition->addMethodCall('setIdleTimeoutExitCode', [$consumer['idle_timeout_exit_code']]); |
||
243 | } |
||
244 | 29 | if (isset($consumer['timeout_wait'])) { |
|
245 | 25 | $definition->addMethodCall('setTimeoutWait', [$consumer['timeout_wait']]); |
|
246 | } |
||
247 | 29 | if (isset($consumer['graceful_max_execution'])) { |
|
248 | $definition->addMethodCall( |
||
249 | 'setGracefulMaxExecutionDateTimeFromSecondsInTheFuture', |
||
250 | [$consumer['graceful_max_execution']['timeout']] |
||
251 | ); |
||
252 | $definition->addMethodCall( |
||
253 | 'setGracefulMaxExecutionTimeoutExitCode', |
||
254 | [$consumer['graceful_max_execution']['exit_code']] |
||
255 | ); |
||
256 | } |
||
257 | 29 | if (!$consumer['auto_setup_fabric']) { |
|
258 | $definition->addMethodCall('disableAutoSetupFabric'); |
||
259 | } |
||
260 | if (isset($consumer['options'])) { |
||
261 | 29 | $definition->addMethodCall( |
|
262 | 29 | 'setConsumerOptions', |
|
263 | 28 | [$this->normalizeArgumentKeys($consumer['options'])] |
|
264 | ); |
||
265 | } |
||
266 | 29 | ||
267 | 1 | $this->injectConnection($definition, $consumer['connection']); |
|
268 | if ($this->collectorEnabled) { |
||
269 | $this->injectLoggedChannel($definition, $key, $consumer['connection']); |
||
270 | 29 | } |
|
271 | 29 | ||
272 | 29 | if ($consumer['enable_logger']) { |
|
273 | $this->injectLogger($definition); |
||
274 | } |
||
275 | 29 | ||
276 | 29 | $name = sprintf('old_sound_rabbit_mq.%s_consumer', $key); |
|
277 | 29 | $this->container->setDefinition($name, $definition); |
|
278 | 29 | $this->addDequeuerAwareCall($consumer['callback'], $name); |
|
279 | 29 | ||
280 | // register alias for argument auto wiring |
||
281 | 29 | if (method_exists($this->container, 'registerAliasForArgument')) { |
|
282 | 29 | $argName = !str_ends_with(strtolower($key), 'consumer') ? sprintf('%sConsumer', $key) : $key; |
|
283 | 29 | $this->container |
|
284 | ->registerAliasForArgument($name, ConsumerInterface::class, $argName) |
||
285 | ->setPublic(false); |
||
286 | 34 | ||
287 | $this->container |
||
288 | 34 | ->registerAliasForArgument($name, '%old_sound_rabbit_mq.consumer.class%', $argName) |
|
289 | ->setPublic(false); |
||
290 | 34 | } |
|
291 | 25 | } |
|
292 | 25 | } |
|
293 | |||
294 | 25 | protected function loadMultipleConsumers() |
|
295 | { |
||
296 | foreach ($this->config['multiple_consumers'] as $key => $consumer) { |
||
297 | $queues = []; |
||
298 | $callbacks = []; |
||
299 | |||
300 | if (empty($consumer['queues']) && empty($consumer['queues_provider'])) { |
||
301 | 25 | throw new InvalidConfigurationException( |
|
302 | 25 | "Error on loading $key multiple consumer. " . |
|
303 | 25 | "Either 'queues' or 'queues_provider' parameters should be defined." |
|
304 | 25 | ); |
|
305 | } |
||
306 | |||
307 | 25 | foreach ($consumer['queues'] as $queueName => $queueOptions) { |
|
308 | $queues[$queueOptions['name']] = $queueOptions; |
||
309 | 25 | $queues[$queueOptions['name']]['callback'] = [new Reference($queueOptions['callback']), 'execute']; |
|
310 | 25 | $callbacks[] = $queueOptions['callback']; |
|
311 | 25 | } |
|
312 | 25 | ||
313 | 25 | $definition = new Definition('%old_sound_rabbit_mq.multi_consumer.class%'); |
|
314 | $definition |
||
315 | 25 | ->setPublic(true) |
|
316 | 25 | ->addTag('old_sound_rabbit_mq.base_amqp') |
|
317 | 25 | ->addTag('old_sound_rabbit_mq.multi_consumer') |
|
318 | 25 | ->addMethodCall('setExchangeOptions', [$this->normalizeArgumentKeys($consumer['exchange_options'])]) |
|
319 | ->addMethodCall('setQueues', [$this->normalizeArgumentKeys($queues)]); |
||
320 | |||
321 | if ($consumer['queues_provider']) { |
||
322 | 25 | $definition->addMethodCall( |
|
323 | 'setQueuesProvider', |
||
324 | [new Reference($consumer['queues_provider'])] |
||
325 | ); |
||
326 | } |
||
327 | |||
328 | if (array_key_exists('qos_options', $consumer)) { |
||
329 | $definition->addMethodCall('setQosOptions', [ |
||
330 | 25 | $consumer['qos_options']['prefetch_size'], |
|
331 | $consumer['qos_options']['prefetch_count'], |
||
332 | $consumer['qos_options']['global'], |
||
333 | 25 | ]); |
|
334 | } |
||
335 | |||
336 | 25 | if (isset($consumer['idle_timeout'])) { |
|
337 | 25 | $definition->addMethodCall('setIdleTimeout', [$consumer['idle_timeout']]); |
|
338 | } |
||
339 | 25 | if (isset($consumer['idle_timeout_exit_code'])) { |
|
340 | $definition->addMethodCall('setIdleTimeoutExitCode', [$consumer['idle_timeout_exit_code']]); |
||
341 | } |
||
342 | if (isset($consumer['timeout_wait'])) { |
||
343 | $definition->addMethodCall('setTimeoutWait', [$consumer['timeout_wait']]); |
||
344 | } |
||
345 | if (isset($consumer['graceful_max_execution'])) { |
||
346 | $definition->addMethodCall( |
||
347 | 'setGracefulMaxExecutionDateTimeFromSecondsInTheFuture', |
||
348 | [$consumer['graceful_max_execution']['timeout']] |
||
349 | 25 | ); |
|
350 | $definition->addMethodCall( |
||
351 | 'setGracefulMaxExecutionTimeoutExitCode', |
||
352 | [$consumer['graceful_max_execution']['exit_code']] |
||
353 | 25 | ); |
|
354 | 25 | } |
|
355 | 25 | if (!$consumer['auto_setup_fabric']) { |
|
356 | $definition->addMethodCall('disableAutoSetupFabric'); |
||
357 | } |
||
358 | 25 | if (isset($consumer['options'])) { |
|
359 | $definition->addMethodCall( |
||
360 | 'setConsumerOptions', |
||
361 | [$this->normalizeArgumentKeys($consumer['options'])] |
||
362 | 25 | ); |
|
363 | 25 | } |
|
364 | 25 | ||
365 | 25 | $this->injectConnection($definition, $consumer['connection']); |
|
366 | if ($this->collectorEnabled) { |
||
367 | 25 | $this->injectLoggedChannel($definition, $key, $consumer['connection']); |
|
368 | 25 | } |
|
369 | |||
370 | if ($consumer['enable_logger']) { |
||
371 | 34 | $this->injectLogger($definition); |
|
372 | } |
||
373 | 34 | ||
374 | $name = sprintf('old_sound_rabbit_mq.%s_multiple', $key); |
||
375 | 34 | $this->container->setDefinition($name, $definition); |
|
376 | 25 | if ($consumer['queues_provider']) { |
|
377 | $this->addDequeuerAwareCall($consumer['queues_provider'], $name); |
||
378 | } |
||
379 | foreach ($callbacks as $callback) { |
||
380 | $this->addDequeuerAwareCall($callback, $name); |
||
381 | } |
||
382 | } |
||
383 | 25 | } |
|
384 | |||
385 | 25 | protected function loadDynamicConsumers() |
|
386 | 25 | { |
|
387 | 25 | foreach ($this->config['dynamic_consumers'] as $key => $consumer) { |
|
388 | 25 | if (empty($consumer['queue_options_provider'])) { |
|
389 | 25 | throw new InvalidConfigurationException( |
|
390 | 25 | "Error on loading $key dynamic consumer. " . |
|
391 | "'queue_provider' parameter should be defined." |
||
392 | 25 | ); |
|
393 | } |
||
394 | |||
395 | $definition = new Definition('%old_sound_rabbit_mq.dynamic_consumer.class%'); |
||
396 | $definition |
||
397 | ->setPublic(true) |
||
398 | ->addTag('old_sound_rabbit_mq.base_amqp') |
||
399 | ->addTag('old_sound_rabbit_mq.consumer') |
||
400 | 25 | ->addTag('old_sound_rabbit_mq.dynamic_consumer') |
|
401 | 25 | ->addMethodCall('setExchangeOptions', [$this->normalizeArgumentKeys($consumer['exchange_options'])]) |
|
402 | 25 | ->addMethodCall('setCallback', [[new Reference($consumer['callback']), 'execute']]); |
|
403 | |||
404 | if (array_key_exists('qos_options', $consumer)) { |
||
405 | 25 | $definition->addMethodCall('setQosOptions', [ |
|
406 | $consumer['qos_options']['prefetch_size'], |
||
407 | $consumer['qos_options']['prefetch_count'], |
||
408 | 25 | $consumer['qos_options']['global'], |
|
409 | ]); |
||
410 | } |
||
411 | 25 | ||
412 | $definition->addMethodCall( |
||
413 | 'setQueueOptionsProvider', |
||
414 | 25 | [new Reference($consumer['queue_options_provider'])] |
|
415 | ); |
||
416 | |||
417 | if (isset($consumer['idle_timeout'])) { |
||
418 | $definition->addMethodCall('setIdleTimeout', [$consumer['idle_timeout']]); |
||
419 | } |
||
420 | if (isset($consumer['idle_timeout_exit_code'])) { |
||
421 | $definition->addMethodCall('setIdleTimeoutExitCode', [$consumer['idle_timeout_exit_code']]); |
||
422 | } |
||
423 | if (isset($consumer['timeout_wait'])) { |
||
424 | 25 | $definition->addMethodCall('setTimeoutWait', [$consumer['timeout_wait']]); |
|
425 | } |
||
426 | if (isset($consumer['graceful_max_execution'])) { |
||
427 | $definition->addMethodCall( |
||
428 | 25 | 'setGracefulMaxExecutionDateTimeFromSecondsInTheFuture', |
|
429 | 25 | [$consumer['graceful_max_execution']['timeout']] |
|
430 | 25 | ); |
|
431 | $definition->addMethodCall( |
||
432 | 'setGracefulMaxExecutionTimeoutExitCode', |
||
433 | 25 | [$consumer['graceful_max_execution']['exit_code']] |
|
434 | ); |
||
435 | } |
||
436 | if (!$consumer['auto_setup_fabric']) { |
||
437 | 25 | $definition->addMethodCall('disableAutoSetupFabric'); |
|
438 | 25 | } |
|
439 | 25 | if (isset($consumer['options'])) { |
|
440 | 25 | $definition->addMethodCall( |
|
441 | 'setConsumerOptions', |
||
442 | 34 | [$this->normalizeArgumentKeys($consumer['options'])] |
|
443 | ); |
||
444 | 34 | } |
|
445 | |||
446 | 34 | $this->injectConnection($definition, $consumer['connection']); |
|
447 | if ($this->collectorEnabled) { |
||
448 | $this->injectLoggedChannel($definition, $key, $consumer['connection']); |
||
449 | } |
||
450 | |||
451 | if ($consumer['enable_logger']) { |
||
452 | $this->injectLogger($definition); |
||
453 | } |
||
454 | |||
455 | $name = sprintf('old_sound_rabbit_mq.%s_dynamic', $key); |
||
456 | $this->container->setDefinition($name, $definition); |
||
457 | $this->addDequeuerAwareCall($consumer['callback'], $name); |
||
458 | $this->addDequeuerAwareCall($consumer['queue_options_provider'], $name); |
||
459 | } |
||
460 | } |
||
461 | |||
462 | protected function loadBatchConsumers() |
||
463 | { |
||
464 | foreach ($this->config['batch_consumers'] as $key => $consumer) { |
||
465 | $definition = new Definition('%old_sound_rabbit_mq.batch_consumer.class%'); |
||
466 | |||
467 | if (!isset($consumer['exchange_options'])) { |
||
468 | $consumer['exchange_options'] = $this->getDefaultExchangeOptions(); |
||
469 | } |
||
470 | |||
471 | $definition |
||
472 | ->setPublic(true) |
||
473 | ->addTag('old_sound_rabbit_mq.base_amqp') |
||
474 | ->addTag('old_sound_rabbit_mq.batch_consumer') |
||
475 | ->addMethodCall('setTimeoutWait', [$consumer['timeout_wait']]) |
||
476 | ->addMethodCall('setPrefetchCount', [$consumer['qos_options']['prefetch_count']]) |
||
477 | ->addMethodCall('setCallback', [[new Reference($consumer['callback']), 'batchExecute']]) |
||
478 | ->addMethodCall('setExchangeOptions', [$this->normalizeArgumentKeys($consumer['exchange_options'])]) |
||
479 | ->addMethodCall('setQueueOptions', [$this->normalizeArgumentKeys($consumer['queue_options'])]) |
||
480 | ->addMethodCall('setQosOptions', [ |
||
481 | $consumer['qos_options']['prefetch_size'], |
||
482 | $consumer['qos_options']['prefetch_count'], |
||
483 | $consumer['qos_options']['global'], |
||
484 | ]) |
||
485 | ; |
||
486 | |||
487 | if (isset($consumer['idle_timeout_exit_code'])) { |
||
488 | $definition->addMethodCall('setIdleTimeoutExitCode', [$consumer['idle_timeout_exit_code']]); |
||
489 | } |
||
490 | |||
491 | if (isset($consumer['idle_timeout'])) { |
||
492 | $definition->addMethodCall('setIdleTimeout', [$consumer['idle_timeout']]); |
||
493 | } |
||
494 | |||
495 | if (isset($consumer['graceful_max_execution'])) { |
||
496 | $definition->addMethodCall( |
||
497 | 'setGracefulMaxExecutionDateTimeFromSecondsInTheFuture', |
||
498 | [$consumer['graceful_max_execution']['timeout']] |
||
499 | ); |
||
500 | } |
||
501 | |||
502 | if (!$consumer['auto_setup_fabric']) { |
||
503 | 34 | $definition->addMethodCall('disableAutoSetupFabric'); |
|
504 | } |
||
505 | 34 | ||
506 | if (isset($consumer['options'])) { |
||
507 | 34 | $definition->addMethodCall( |
|
508 | 25 | 'setConsumerOptions', |
|
509 | [$this->normalizeArgumentKeys($consumer['options'])] |
||
510 | 25 | ); |
|
511 | 25 | } |
|
512 | 25 | ||
513 | 25 | if ($consumer['keep_alive']) { |
|
514 | 25 | $definition->addMethodCall('keepAlive'); |
|
515 | 25 | } |
|
516 | 25 | ||
517 | 25 | $this->injectConnection($definition, $consumer['connection']); |
|
518 | if ($this->collectorEnabled) { |
||
519 | $this->injectLoggedChannel($definition, $key, $consumer['connection']); |
||
520 | 25 | } |
|
521 | 25 | ||
522 | 25 | if ($consumer['enable_logger']) { |
|
523 | $this->injectLogger($definition); |
||
524 | 34 | } |
|
525 | |||
526 | $this->container->setDefinition(sprintf('old_sound_rabbit_mq.%s_batch', $key), $definition); |
||
527 | } |
||
528 | } |
||
529 | |||
530 | protected function loadAnonConsumers() |
||
531 | { |
||
532 | foreach ($this->config['anon_consumers'] as $key => $anon) { |
||
533 | $definition = new Definition('%old_sound_rabbit_mq.anon_consumer.class%'); |
||
534 | 30 | $definition |
|
535 | ->setPublic(true) |
||
536 | 30 | ->addTag('old_sound_rabbit_mq.base_amqp') |
|
537 | 1 | ->addTag('old_sound_rabbit_mq.anon_consumer') |
|
538 | ->addMethodCall('setExchangeOptions', [$this->normalizeArgumentKeys($anon['exchange_options'])]) |
||
539 | 1 | ->addMethodCall('setCallback', [[new Reference($anon['callback']), 'execute']]); |
|
540 | |||
541 | if (isset($anon['options'])) { |
||
542 | $definition->addMethodCall( |
||
543 | 1 | 'setConsumerOptions', |
|
544 | 1 | [$this->normalizeArgumentKeys($anon['options'])] |
|
545 | 1 | ); |
|
546 | } |
||
547 | |||
548 | 1 | $this->injectConnection($definition, $anon['connection']); |
|
549 | if ($this->collectorEnabled) { |
||
550 | 1 | $this->injectLoggedChannel($definition, $key, $anon['connection']); |
|
551 | } |
||
552 | 30 | ||
553 | $name = sprintf('old_sound_rabbit_mq.%s_anon', $key); |
||
554 | $this->container->setDefinition($name, $definition); |
||
555 | $this->addDequeuerAwareCall($anon['callback'], $name); |
||
556 | } |
||
557 | } |
||
558 | |||
559 | /** |
||
560 | * Symfony 2 converts '-' to '_' when defined in the configuration. This leads to problems when using x-ha-policy |
||
561 | * parameter. So we revert the change for right configurations. |
||
562 | * |
||
563 | * @param array $config |
||
564 | * |
||
565 | * @return array |
||
566 | */ |
||
567 | private function normalizeArgumentKeys(array $config): array |
||
568 | { |
||
569 | if (isset($config['arguments'])) { |
||
570 | $arguments = $config['arguments']; |
||
571 | // support for old configuration |
||
572 | if (is_string($arguments)) { |
||
573 | $arguments = $this->argumentsStringAsArray($arguments); |
||
574 | } |
||
575 | |||
576 | $newArguments = []; |
||
577 | foreach ($arguments as $key => $value) { |
||
578 | if (strstr($key, '_')) { |
||
579 | 34 | $key = str_replace('_', '-', $key); |
|
580 | } |
||
581 | 34 | $newArguments[$key] = $value; |
|
582 | 28 | } |
|
583 | 28 | $config['arguments'] = $newArguments; |
|
584 | } |
||
585 | 28 | return $config; |
|
586 | 28 | } |
|
587 | 28 | ||
588 | 28 | /** |
|
589 | 28 | * Support for arguments provided as string. Support for old configuration files. |
|
590 | * |
||
591 | 28 | * @deprecated |
|
592 | 28 | * @param string $arguments |
|
593 | * @return array |
||
594 | 28 | */ |
|
595 | 28 | private function argumentsStringAsArray($arguments): array |
|
596 | { |
||
597 | 28 | $argumentsArray = []; |
|
598 | |||
599 | 28 | $argumentPairs = explode(',', $arguments); |
|
600 | foreach ($argumentPairs as $argument) { |
||
601 | 34 | $argumentPair = explode(':', $argument); |
|
602 | $type = 'S'; |
||
603 | 34 | if (isset($argumentPair[2])) { |
|
604 | $type = $argumentPair[2]; |
||
605 | 34 | } |
|
606 | 25 | $argumentsArray[$argumentPair[0]] = [$type, $argumentPair[1]]; |
|
607 | } |
||
608 | 25 | ||
609 | 25 | return $argumentsArray; |
|
610 | 25 | } |
|
611 | 25 | ||
612 | 25 | protected function loadRpcClients() |
|
613 | 25 | { |
|
614 | 25 | foreach ($this->config['rpc_clients'] as $key => $client) { |
|
615 | 25 | $definition = new Definition('%old_sound_rabbit_mq.rpc_client.class%'); |
|
616 | $definition->setLazy($client['lazy']); |
||
617 | 25 | $definition |
|
618 | ->addTag('old_sound_rabbit_mq.rpc_client') |
||
619 | ->addMethodCall('initClient', [$client['expect_serialized_response']]); |
||
620 | $this->injectConnection($definition, $client['connection']); |
||
621 | if ($this->collectorEnabled) { |
||
622 | $this->injectLoggedChannel($definition, $key, $client['connection']); |
||
623 | } |
||
624 | 25 | if (array_key_exists('unserializer', $client)) { |
|
625 | 25 | $definition->addMethodCall('setUnserializer', [$client['unserializer']]); |
|
626 | } |
||
627 | 25 | if (array_key_exists('direct_reply_to', $client)) { |
|
628 | 25 | $definition->addMethodCall('setDirectReplyTo', [$client['direct_reply_to']]); |
|
629 | } |
||
630 | 25 | $definition->setPublic(true); |
|
631 | 25 | ||
632 | $this->container->setDefinition(sprintf('old_sound_rabbit_mq.%s_rpc', $key), $definition); |
||
633 | 25 | } |
|
634 | } |
||
635 | 34 | ||
636 | protected function loadRpcServers() |
||
637 | 31 | { |
|
638 | foreach ($this->config['rpc_servers'] as $key => $server) { |
||
639 | 31 | $definition = new Definition('%old_sound_rabbit_mq.rpc_server.class%'); |
|
640 | 31 | $definition |
|
641 | ->setPublic(true) |
||
642 | 31 | ->addTag('old_sound_rabbit_mq.base_amqp') |
|
643 | 31 | ->addTag('old_sound_rabbit_mq.rpc_server') |
|
644 | 31 | ->addMethodCall('initServer', [$key]) |
|
645 | ->addMethodCall('setCallback', [[new Reference($server['callback']), 'execute']]); |
||
646 | 31 | $this->injectConnection($definition, $server['connection']); |
|
647 | if ($this->collectorEnabled) { |
||
648 | 31 | $this->injectLoggedChannel($definition, $key, $server['connection']); |
|
649 | 31 | } |
|
650 | 31 | if (array_key_exists('qos_options', $server)) { |
|
651 | $definition->addMethodCall('setQosOptions', [ |
||
652 | 33 | $server['qos_options']['prefetch_size'], |
|
653 | $server['qos_options']['prefetch_count'], |
||
654 | 33 | $server['qos_options']['global'], |
|
655 | 33 | ]); |
|
656 | } |
||
657 | 34 | if (array_key_exists('exchange_options', $server)) { |
|
658 | $definition->addMethodCall('setExchangeOptions', [$server['exchange_options']]); |
||
659 | 34 | } |
|
660 | if (array_key_exists('queue_options', $server)) { |
||
661 | $definition->addMethodCall('setQueueOptions', [$server['queue_options']]); |
||
662 | } |
||
663 | if (array_key_exists('serializer', $server)) { |
||
664 | $definition->addMethodCall('setSerializer', [$server['serializer']]); |
||
665 | } |
||
666 | $this->container->setDefinition(sprintf('old_sound_rabbit_mq.%s_server', $key), $definition); |
||
667 | } |
||
668 | } |
||
669 | 29 | ||
670 | protected function injectLoggedChannel(Definition $definition, $name, $connectionName) |
||
671 | 29 | { |
|
672 | 29 | $id = sprintf('old_sound_rabbit_mq.channel.%s', $name); |
|
673 | $channel = new Definition('%old_sound_rabbit_mq.logged.channel.class%'); |
||
674 | $channel |
||
675 | ->setPublic(false) |
||
676 | ->addTag('old_sound_rabbit_mq.logged_channel'); |
||
677 | $this->injectConnection($channel, $connectionName); |
||
678 | |||
679 | $this->container->setDefinition($id, $channel); |
||
680 | |||
681 | $this->channelIds[] = $id; |
||
682 | 1 | $definition->addArgument(new Reference($id)); |
|
683 | } |
||
684 | 1 | ||
685 | 1 | protected function injectConnection(Definition $definition, $connectionName) |
|
686 | { |
||
687 | 1 | $definition->addArgument(new Reference(sprintf('old_sound_rabbit_mq.connection.%s', $connectionName))); |
|
688 | 1 | } |
|
689 | |||
690 | public function getAlias(): string |
||
691 | { |
||
692 | return 'old_sound_rabbit_mq'; |
||
693 | } |
||
694 | |||
695 | 1 | /** |
|
696 | * Add proper dequeuer aware call |
||
697 | * |
||
698 | 1 | * @param string $callback |
|
699 | * @param string $name |
||
700 | * @throws \ReflectionException |
||
701 | */ |
||
702 | protected function addDequeuerAwareCall($callback, $name) |
||
703 | { |
||
704 | if (!$this->container->has($callback)) { |
||
705 | return; |
||
706 | } |
||
707 | |||
708 | $callbackDefinition = $this->container->findDefinition($callback); |
||
709 | $refClass = new \ReflectionClass($callbackDefinition->getClass()); |
||
710 | 30 | if ($refClass->implementsInterface('OldSound\RabbitMqBundle\RabbitMq\DequeuerAwareInterface')) { |
|
711 | $callbackDefinition->addMethodCall('setDequeuer', [new Reference($name)]); |
||
712 | } |
||
713 | 30 | } |
|
714 | |||
715 | private function injectLogger(Definition $definition) |
||
716 | { |
||
717 | $definition->addTag('monolog.logger', [ |
||
718 | 'channel' => 'phpamqplib', |
||
719 | ]); |
||
720 | $definition->addMethodCall('setLogger', [new Reference('logger', ContainerInterface::IGNORE_ON_INVALID_REFERENCE)]); |
||
721 | } |
||
722 | |||
723 | /** |
||
724 | * Get default AMQP exchange options |
||
725 | * |
||
726 | * @return array |
||
727 | */ |
||
728 | protected function getDefaultExchangeOptions(): array |
||
729 | { |
||
730 | return [ |
||
731 | 'name' => '', |
||
732 | 'type' => 'direct', |
||
733 | 'passive' => true, |
||
734 | 'declare' => false, |
||
735 | ]; |
||
736 | } |
||
737 | |||
738 | /** |
||
739 | * Get default AMQP queue options |
||
740 | * |
||
741 | * @return array |
||
742 | */ |
||
743 | protected function getDefaultQueueOptions(): array |
||
744 | { |
||
745 | return [ |
||
746 | 'name' => '', |
||
747 | 'declare' => false, |
||
748 | ]; |
||
749 | } |
||
750 | } |
||
751 |
This check marks implicit conversions of arrays to boolean values in a comparison. While in PHP an empty array is considered to be equal (but not identical) to false, this is not always apparent.
Consider making the comparison explicit by using
empty(..)
or! empty(...)
instead.