Merge 668c153 into 5db745c

sonata-project · Jan 21, 2020 · e9905c1 · e9905c1
2 parents 5db745c + 668c153
commit e9905c1
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 4 deletions.
diff --git a/docs/reference/advanced_configuration.rst b/docs/reference/advanced_configuration.rst
@@ -80,6 +80,10 @@ Full configuration options:
                         pass:                 guest
                         vhost:                guest
                         console_url:          'http://localhost:55672/api'
+
+                        # By default the RabbitMQ configuration uses a DelayStrategyAware connection factory.
+                        # The default connection is assigned a RabbitMqDlxDelayStrategy object.
+                        # To specify your own custom delay strategy override the 'sonata.notification.backend.rabbitmq.delay_strategy' service
                         factory_class:        \Enqueue\AmqpLib\AmqpConnectionFactory
             consumers:
 
@@ -141,4 +145,3 @@ run `composer require enqueue/amqp-ext:^0.8` and change `factory_class` option i
 .. _`enqueue/amqp-lib`: https://github.com/php-enqueue/enqueue-dev/blob/master/docs/transport/amqp_lib.md
 .. _`enqueue/amqp-ext`: https://github.com/php-enqueue/enqueue-dev/blob/master/docs/transport/amqp_ext.md
 .. _`enqueue/amqp-bunny`: https://github.com/php-enqueue/enqueue-dev/blob/master/docs/transport/amqp_bunny.md
-
diff --git a/src/Backend/AMQPBackendDispatcher.php b/src/Backend/AMQPBackendDispatcher.php
@@ -13,8 +13,8 @@
 
 namespace Sonata\NotificationBundle\Backend;
 
+use Enqueue\AmqpTools\DelayStrategy;
 use Enqueue\AmqpTools\DelayStrategyAware;
-use Enqueue\AmqpTools\RabbitMqDlxDelayStrategy;
 use Guzzle\Http\Client as GuzzleClient;
 use Interop\Amqp\AmqpConnectionFactory;
 use Interop\Amqp\AmqpContext;
@@ -62,6 +62,11 @@ class AMQPBackendDispatcher extends QueueBackendDispatcher
      */
     private $context;
 
+    /**
+     * @var DelayStrategy
+     */
+    private $delayStrategy = null;
+
     /**
      * @param string $defaultQueue
      */
@@ -129,8 +134,8 @@ final public function getContext()
                 'vhost' => $this->settings['vhost'],
             ]);
 
-            if ($factory instanceof DelayStrategyAware) {
-                $factory->setDelayStrategy(new RabbitMqDlxDelayStrategy());
+            if ($factory instanceof DelayStrategyAware && $this->delayStrategy instanceof DelayStrategy) {
+                $factory->setDelayStrategy($this->delayStrategy);
             }
 
             $this->context = $factory->createContext();
@@ -269,6 +274,11 @@ public function initialize()
     {
     }
 
+    public function setDelayStrategy(DelayStrategy $delayStrategy): void
+    {
+        $this->delayStrategy = $delayStrategy;
+    }
+
     /**
      * Calls the rabbitmq management api /api/<vhost>/queues endpoint to list the available queues.
      *

diff --git a/src/DependencyInjection/SonataNotificationExtension.php b/src/DependencyInjection/SonataNotificationExtension.php
@@ -383,6 +383,10 @@ protected function configureRabbitmq(ContainerBuilder $container, array $config)
             throw new \RuntimeException('You need to specify a valid default queue for the rabbitmq backend!');
         }
 
+        if (class_exists('\Enqueue\AmqpTools\RabbitMqDlxDelayStrategy')) {
+            $container->setDefinition('sonata.notification.backend.rabbitmq.delay_strategy', new Definition(\Enqueue\AmqpTools\RabbitMqDlxDelayStrategy::class));
+        }
+
         $container->getDefinition('sonata.notification.backend.rabbitmq')
             ->replaceArgument(0, $connection)
             ->replaceArgument(1, $queues)

diff --git a/src/Resources/config/backend.xml b/src/Resources/config/backend.xml
@@ -20,6 +20,9 @@
             <argument/>
             <argument/>
             <argument/>
+            <call method="setDelayStrategy">
+                <argument type="service" id="sonata.notification.backend.rabbitmq.delay_strategy" on-invalid="ignore"/>
+            </call>
         </service>
     </services>
 </container>