Fix Class AmqpReceivedStamp does not exist

Warxcell · Warxcell · commit e158c561a489 · 2021-06-19T19:09:07.000+03:00
diff --git a/src/Symfony/Bundle/FrameworkBundle/Resources/config/messenger.php b/src/Symfony/Bundle/FrameworkBundle/Resources/config/messenger.php
@@ -13,6 +13,7 @@
 
 use Symfony\Component\DependencyInjection\ServiceLocator;
 use Symfony\Component\Messenger\Bridge\AmazonSqs\Transport\AmazonSqsTransportFactory;
+use Symfony\Component\Messenger\Bridge\Amqp\Middleware\RejectRedeliveredMessageMiddleware;
 use Symfony\Component\Messenger\Bridge\Amqp\Transport\AmqpTransportFactory;
 use Symfony\Component\Messenger\Bridge\Beanstalkd\Transport\BeanstalkdTransportFactory;
 use Symfony\Component\Messenger\Bridge\Redis\Transport\RedisTransportFactory;
@@ -26,7 +27,6 @@
 use Symfony\Component\Messenger\Middleware\DispatchAfterCurrentBusMiddleware;
 use Symfony\Component\Messenger\Middleware\FailedMessageProcessingMiddleware;
 use Symfony\Component\Messenger\Middleware\HandleMessageMiddleware;
-use Symfony\Component\Messenger\Middleware\RejectRedeliveredMessageMiddleware;
 use Symfony\Component\Messenger\Middleware\SendMessageMiddleware;
 use Symfony\Component\Messenger\Middleware\TraceableMiddleware;
 use Symfony\Component\Messenger\Middleware\ValidationMiddleware;
@@ -42,8 +42,8 @@
 use Symfony\Component\Messenger\Transport\TransportFactory;
 
 return static function (ContainerConfigurator $container) {
-    $container->services()
-        ->alias(SerializerInterface::class, 'messenger.default_serializer')
+    $services = $container->services();
+    $services->alias(SerializerInterface::class, 'messenger.default_serializer')
 
         // Asynchronous
         ->set('messenger.senders_locator', SendersLocator::class)
@@ -89,11 +89,13 @@
         ->set('messenger.middleware.validation', ValidationMiddleware::class)
             ->args([
                 service('validator'),
-            ])
+            ]);
 
-        ->set('messenger.middleware.reject_redelivered_message_middleware', RejectRedeliveredMessageMiddleware::class)
+    if (class_exists(RejectRedeliveredMessageMiddleware::class)) {
+        $services->set('messenger.middleware.reject_redelivered_message_middleware', RejectRedeliveredMessageMiddleware::class);
+    }
 
-        ->set('messenger.middleware.failed_message_processing_middleware', FailedMessageProcessingMiddleware::class)
+    $services->set('messenger.middleware.failed_message_processing_middleware', FailedMessageProcessingMiddleware::class)
 
         ->set('messenger.middleware.traceable', TraceableMiddleware::class)
             ->abstract()
diff --git a/src/Symfony/Component/Messenger/Bridge/Amqp/Middleware/RejectRedeliveredMessageMiddleware.php b/src/Symfony/Component/Messenger/Bridge/Amqp/Middleware/RejectRedeliveredMessageMiddleware.php
@@ -0,0 +1,52 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Messenger\Bridge\Amqp\Middleware;
+
+use Symfony\Component\Messenger\Bridge\Amqp\Transport\AmqpReceivedStamp;
+use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Exception\RejectRedeliveredMessageException;
+use Symfony\Component\Messenger\Middleware\MiddlewareInterface;
+use Symfony\Component\Messenger\Middleware\StackInterface;
+use Symfony\Component\Messenger\Transport\AmqpExt\AmqpReceivedStamp as LegacyAmqpReceivedStamp;
+
+/**
+ * Middleware that throws a RejectRedeliveredMessageException when a message is detected that has been redelivered by AMQP.
+ *
+ * The middleware runs before the HandleMessageMiddleware and prevents redelivered messages from being handled directly.
+ * The thrown exception is caught by the worker and will trigger the retry logic according to the retry strategy.
+ *
+ * AMQP redelivers messages when they do not get acknowledged or rejected. This can happen when the connection times out
+ * or an exception is thrown before acknowledging or rejecting. When such errors happen again while handling the
+ * redelivered message, the message would get redelivered again and again. The purpose of this middleware is to prevent
+ * infinite redelivery loops and to unblock the queue by republishing the redelivered messages as retries with a retry
+ * limit and potential delay.
+ *
+ * @author Tobias Schultze <http://tobion.de>
+ */
+class RejectRedeliveredMessageMiddleware implements MiddlewareInterface
+{
+    public function handle(Envelope $envelope, StackInterface $stack): Envelope
+    {
+        $amqpReceivedStamp = $envelope->last(AmqpReceivedStamp::class);
+        if ($amqpReceivedStamp instanceof AmqpReceivedStamp && $amqpReceivedStamp->getAmqpEnvelope()->isRedelivery()) {
+            throw new RejectRedeliveredMessageException('Redelivered message from AMQP detected that will be rejected and trigger the retry logic.');
+        }
+
+        // Legacy code to support symfony/messenger < 5.1
+        $amqpReceivedStamp = $envelope->last(LegacyAmqpReceivedStamp::class);
+        if ($amqpReceivedStamp instanceof LegacyAmqpReceivedStamp && $amqpReceivedStamp->getAmqpEnvelope()->isRedelivery()) {
+            throw new RejectRedeliveredMessageException('Redelivered message from AMQP detected that will be rejected and trigger the retry logic.');
+        }
+
+        return $stack->next()->handle($envelope, $stack);
+    }
+}
diff --git a/src/Symfony/Component/Messenger/Middleware/RejectRedeliveredMessageMiddleware.php b/src/Symfony/Component/Messenger/Middleware/RejectRedeliveredMessageMiddleware.php
@@ -11,40 +11,9 @@
 
 namespace Symfony\Component\Messenger\Middleware;
 
-use Symfony\Component\Messenger\Bridge\Amqp\Transport\AmqpReceivedStamp;
-use Symfony\Component\Messenger\Envelope;
-use Symfony\Component\Messenger\Exception\RejectRedeliveredMessageException;
-use Symfony\Component\Messenger\Transport\AmqpExt\AmqpReceivedStamp as LegacyAmqpReceivedStamp;
-
 /**
- * Middleware that throws a RejectRedeliveredMessageException when a message is detected that has been redelivered by AMQP.
- *
- * The middleware runs before the HandleMessageMiddleware and prevents redelivered messages from being handled directly.
- * The thrown exception is caught by the worker and will trigger the retry logic according to the retry strategy.
- *
- * AMQP redelivers messages when they do not get acknowledged or rejected. This can happen when the connection times out
- * or an exception is thrown before acknowledging or rejecting. When such errors happen again while handling the
- * redelivered message, the message would get redelivered again and again. The purpose of this middleware is to prevent
- * infinite redelivery loops and to unblock the queue by republishing the redelivered messages as retries with a retry
- * limit and potential delay.
- *
- * @author Tobias Schultze <http://tobion.de>
+ * @deprecated since Symfony 5.2, to be removed in 6.0. Use Symfony\Component\Messenger\Bridge\Amqp\Middleware\RejectRedeliveredMessageMiddleware instead.
  */
-class RejectRedeliveredMessageMiddleware implements MiddlewareInterface
+class RejectRedeliveredMessageMiddleware extends \Symfony\Component\Messenger\Bridge\Amqp\Middleware\RejectRedeliveredMessageMiddleware
 {
-    public function handle(Envelope $envelope, StackInterface $stack): Envelope
-    {
-        $amqpReceivedStamp = $envelope->last(AmqpReceivedStamp::class);
-        if ($amqpReceivedStamp instanceof AmqpReceivedStamp && $amqpReceivedStamp->getAmqpEnvelope()->isRedelivery()) {
-            throw new RejectRedeliveredMessageException('Redelivered message from AMQP detected that will be rejected and trigger the retry logic.');
-        }
-
-        // Legacy code to support symfony/messenger < 5.1
-        $amqpReceivedStamp = $envelope->last(LegacyAmqpReceivedStamp::class);
-        if ($amqpReceivedStamp instanceof LegacyAmqpReceivedStamp && $amqpReceivedStamp->getAmqpEnvelope()->isRedelivery()) {
-            throw new RejectRedeliveredMessageException('Redelivered message from AMQP detected that will be rejected and trigger the retry logic.');
-        }
-
-        return $stack->next()->handle($envelope, $stack);
-    }
 }
