Messenger: add support for HandledStamp (#2420)

* Messenger: add support for HandledStamp

* messenger test handledstamp

Author: dunglas

src/Bridge/Symfony/Messenger/DataPersister.php

@@ -18,6 +18,7 @@
 use ApiPlatform\Core\Util\ClassInfoTrait;
 use Symfony\Component\Messenger\Envelope;
 use Symfony\Component\Messenger\MessageBusInterface;
+use Symfony\Component\Messenger\Stamp\HandledStamp;
 
 /**
  * Dispatches the given resource using the message bus of Symfony Messenger.
@@ -52,9 +53,12 @@ public function supports($data): bool
      */
     public function persist($data)
     {
-        $this->messageBus->dispatch($data);
+        $envelope = $this->messageBus->dispatch($data);
+        if (null === $stamp = $envelope->last(HandledStamp::class)) {
+            return $data;
+        }
 
-        return $data;
+        return $stamp->getResult();
     }
 
     /**

tests/Bridge/Symfony/Messenger/DataPersisterTest.php

@@ -22,6 +22,7 @@
 use Prophecy\Argument;
 use Symfony\Component\Messenger\Envelope;
 use Symfony\Component\Messenger\MessageBusInterface;
+use Symfony\Component\Messenger\Stamp\HandledStamp;
 
 /**
  * @author Kévin Dunglas <[email protected]>
@@ -60,4 +61,15 @@ public function testRemove()
         $dataPersister = new DataPersister($this->prophesize(ResourceMetadataFactoryInterface::class)->reveal(), $messageBus->reveal());
         $dataPersister->remove($dummy);
     }
+
+    public function testHandle()
+    {
+        $dummy = new Dummy();
+
+        $messageBus = $this->prophesize(MessageBusInterface::class);
+        $messageBus->dispatch($dummy)->willReturn(new Envelope(new \stdClass(), new HandledStamp('result', 'DummyHandler::__invoke')))->shouldBeCalled();
+
+        $dataPersister = new DataPersister($this->prophesize(ResourceMetadataFactoryInterface::class)->reveal(), $messageBus->reveal());
+        $this->assertSame('result', $dataPersister->persist($dummy));
+    }
 }