Merge pull request #105 from kmfk/fix-sns-confirmations

Fix for SNS subscription confirmations
uecode · Jun 30, 2016 · 2b7ee86 · 2b7ee86
2 parents ac8b279 + cebeee7
commit 2b7ee86
Showing 1 changed file with 18 additions and 6 deletions.
diff --git a/src/DependencyInjection/UecodeQPushExtension.php b/src/DependencyInjection/UecodeQPushExtension.php
@@ -94,10 +94,7 @@ public function load(array $configs, ContainerBuilder $container)
                 $class, [$queue, $values['options'], $client, new Reference($cache), new Reference('logger')]
             );
 
-            $name = sprintf('uecode_qpush.%s', $queue);
-
-            $container->setDefinition($name, $definition)
-                ->addTag('monolog.logger', ['channel' => 'qpush'])
+            $definition->addTag('monolog.logger', ['channel' => 'qpush'])
                 ->addTag(
                     'uecode_qpush.event_listener',
                     [
@@ -113,8 +110,23 @@ public function load(array $configs, ContainerBuilder $container)
                         'method' => "onMessageReceived",
                         'priority' => -255
                     ]
-                )
-            ;
+                );
+
+            if (!empty($values['options']['queue_name'])
+                && $config['providers'][$provider]['driver'] == 'aws'
+            ) {
+                $definition->addTag(
+                    'uecode_qpush.event_listener',
+                    [
+                        'event' => "{$values['options']['queue_name']}.on_notification",
+                        'method' => "onNotification",
+                        'priority' => 255
+                    ]
+                );
+            }
+
+            $name = sprintf('uecode_qpush.%s', $queue);
+            $container->setDefinition($name, $definition);
 
             $registry->addMethodCall('addProvider', [$queue, new Reference($name)]);
         }