symfony · VincentLanglet · Mar 21, 2024 · Feb 7, 2025
diff --git a/src/Symfony/Bundle/FrameworkBundle/DependencyInjection/FrameworkExtension.php b/src/Symfony/Bundle/FrameworkBundle/DependencyInjection/FrameworkExtension.php
@@ -2300,6 +2300,7 @@ private function registerMessengerConfiguration(array $config, ContainerBuilder

        $defaultMiddleware = [
            'before' => [
+                ['id' => 'add_self_stampable_stamps_middleware'],
                ['id' => 'add_bus_name_stamp_middleware'],
                ['id' => 'reject_redelivered_message_middleware'],
                ['id' => 'dispatch_after_current_bus'],
@@ -2325,7 +2326,7 @@ private function registerMessengerConfiguration(array $config, ContainerBuilder
                $defaultMiddleware['after'][1]['arguments'] = [$bus['default_middleware']['allow_no_handlers']];

                // argument to add_bus_name_stamp_middleware
-                $defaultMiddleware['before'][0]['arguments'] = [$busId];
+                $defaultMiddleware['before'][1]['arguments'] = [$busId];

                $middleware = array_merge($defaultMiddleware['before'], $middleware, $defaultMiddleware['after']);
            }

diff --git a/src/Symfony/Bundle/FrameworkBundle/Resources/config/messenger.php b/src/Symfony/Bundle/FrameworkBundle/Resources/config/messenger.php
@@ -25,6 +25,7 @@
 use Symfony\Component\Messenger\EventListener\StopWorkerOnRestartSignalListener;
 use Symfony\Component\Messenger\Handler\RedispatchMessageHandler;
 use Symfony\Component\Messenger\Middleware\AddBusNameStampMiddleware;
+use Symfony\Component\Messenger\Middleware\AddSelfStampableStampsMiddleware;
 use Symfony\Component\Messenger\Middleware\DeduplicateMiddleware;
 use Symfony\Component\Messenger\Middleware\DispatchAfterCurrentBusMiddleware;
 use Symfony\Component\Messenger\Middleware\FailedMessageProcessingMiddleware;
@@ -92,6 +93,8 @@
                service('lock.factory'),
            ])

+        ->set('messenger.middleware.add_self_stampable_stamps_middleware', AddSelfStampableStampsMiddleware::class)
+
        ->set('messenger.middleware.add_bus_name_stamp_middleware', AddBusNameStampMiddleware::class)
            ->abstract()


diff --git a/src/Symfony/Bundle/FrameworkBundle/Tests/DependencyInjection/FrameworkExtensionTestCase.php b/src/Symfony/Bundle/FrameworkBundle/Tests/DependencyInjection/FrameworkExtensionTestCase.php
@@ -1064,6 +1064,7 @@ public function testMessengerWithMultipleBusesWithoutDeduplicateMiddleware()
        $this->assertTrue($container->has('messenger.bus.commands'));
        $this->assertSame([], $container->getDefinition('messenger.bus.commands')->getArgument(0));
        $this->assertEquals([
+            ['id' => 'add_self_stampable_stamps_middleware'],
            ['id' => 'add_bus_name_stamp_middleware', 'arguments' => ['messenger.bus.commands']],
            ['id' => 'reject_redelivered_message_middleware'],
            ['id' => 'dispatch_after_current_bus'],
@@ -1104,6 +1105,7 @@ public function testMessengerWithMultipleBusesWithDeduplicateMiddleware()
        $this->assertTrue($container->has('messenger.bus.commands'));
        $this->assertSame([], $container->getDefinition('messenger.bus.commands')->getArgument(0));
        $this->assertEquals([
+            ['id' => 'add_self_stampable_stamps_middleware'],
            ['id' => 'add_bus_name_stamp_middleware', 'arguments' => ['messenger.bus.commands']],
            ['id' => 'reject_redelivered_message_middleware'],
            ['id' => 'dispatch_after_current_bus'],

diff --git a/src/Symfony/Component/Messenger/CHANGELOG.md b/src/Symfony/Component/Messenger/CHANGELOG.md
@@ -7,6 +7,7 @@ CHANGELOG
 * Add `CloseableTransportInterface` to allow closing the transport
 * Add `SentForRetryStamp` that identifies whether a failed message was sent for retry
 * Add `Symfony\Component\Messenger\Middleware\DeduplicateMiddleware` and `Symfony\Component\Messenger\Stamp\DeduplicateStamp`
+ * Add `Symfony\Component\Messenger\Middleware\AddSelfStampableStampsMiddleware` and `Symfony\Component\Messenger\Message\SelfStampableInterface`

 7.2
 ---

diff --git a/src/Symfony/Component/Messenger/Message/SelfStampableInterface.php b/src/Symfony/Component/Messenger/Message/SelfStampableInterface.php
@@ -0,0 +1,25 @@
+<?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\Message;
+
+use Symfony\Component\Messenger\Stamp\StampInterface;
+
+interface SelfStampableInterface
+{
+    /**
+     * List of stamps which will be automatically added to the envelope,
+     * if there is no other stamp of the same class already set.
+     *
+     * @return array<StampInterface>
+     */
+    public function getDefaultStamps(): array;
+}
diff --git a/src/Symfony/Component/Messenger/Middleware/AddSelfStampableStampsMiddleware.php b/src/Symfony/Component/Messenger/Middleware/AddSelfStampableStampsMiddleware.php
@@ -0,0 +1,35 @@
+<?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\Middleware;
+
+use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Message\SelfStampableInterface;
+
+/**
+ * Automatically add stamps from the SelfStampableInterface.
+ */
+class AddSelfStampableStampsMiddleware implements MiddlewareInterface
+{
+    public function handle(Envelope $envelope, StackInterface $stack): Envelope
+    {
+        $message = $envelope->getMessage();
+        if ($message instanceof SelfStampableInterface) {
+            foreach ($message->getDefaultStamps() as $stamp) {
+                if (null === $envelope->last($stamp::class)) {
+                    $envelope = $envelope->with($stamp);
+                }
+            }
+        }
+
+        return $stack->next()->handle($envelope, $stack);
+    }
+}
diff --git a/src/Symfony/Component/Messenger/Tests/Fixtures/SelfStampableDummyMessage.php b/src/Symfony/Component/Messenger/Tests/Fixtures/SelfStampableDummyMessage.php
@@ -0,0 +1,23 @@
+<?php
+
+namespace Symfony\Component\Messenger\Tests\Fixtures;
+
+use Symfony\Component\Messenger\Message\SelfStampableInterface;
+use Symfony\Component\Messenger\Stamp\DelayStamp;
+
+class SelfStampableDummyMessage implements DummyMessageInterface, SelfStampableInterface
+{
+    public function __construct(private string $message)
+    {
+    }
+
+    public function getMessage(): string
+    {
+        return $this->message;
+    }
+
+    public function getDefaultStamps(): array
+    {
+        return [new DelayStamp(1)];
+    }
+}
diff --git a/src/Symfony/Component/Messenger/Tests/Middleware/AddSelfStampableStampsMiddlewareTest.php b/src/Symfony/Component/Messenger/Tests/Middleware/AddSelfStampableStampsMiddlewareTest.php
@@ -0,0 +1,49 @@
+<?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\Tests\Middleware;
+
+use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Middleware\AddSelfStampableStampsMiddleware;
+use Symfony\Component\Messenger\Stamp\DelayStamp;
+use Symfony\Component\Messenger\Test\Middleware\MiddlewareTestCase;
+use Symfony\Component\Messenger\Tests\Fixtures\SelfStampableDummyMessage;
+
+final class AddSelfStampableStampsMiddlewareTest extends MiddlewareTestCase
+{
+    public function testSelfStampableStampsMiddleware()
+    {
+        $message = new SelfStampableDummyMessage('');
+        $envelope = new Envelope($message);
+
+        $decorator = new AddSelfStampableStampsMiddleware();
+
+        $envelope = $decorator->handle($envelope, $this->getStackMock(true));
+
+        $delayStamp = $envelope->last(DelayStamp::class);
+        $this->assertNotNull($delayStamp);
+        $this->assertSame(1, $delayStamp->getDelay());
+    }
+
+    public function testSelfStampableStampsMiddlewareIfStampExists()
+    {
+        $message = new SelfStampableDummyMessage('');
+        $envelope = new Envelope($message, [new DelayStamp(5)]);
+
+        $decorator = new AddSelfStampableStampsMiddleware();
+
+        $envelope = $decorator->handle($envelope, $this->getStackMock(true));
+
+        $delayStamp = $envelope->last(DelayStamp::class);
+        $this->assertNotNull($delayStamp);
+        $this->assertSame(5, $delayStamp->getDelay());
+    }
+}