[Messenger] Add support for the DelayStamp in InMemoryTransport

fabpot · fabpot · commit 29511cac725b · 2023-03-18T10:18:04.000+01:00
diff --git a/src/Symfony/Component/Messenger/Tests/Transport/InMemory/InMemoryTransportTest.php b/src/Symfony/Component/Messenger/Tests/Transport/InMemory/InMemoryTransportTest.php
@@ -13,6 +13,7 @@
 
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Stamp\DelayStamp;
 use Symfony\Component\Messenger\Stamp\TransportMessageIdStamp;
 use Symfony\Component\Messenger\Tests\Fixtures\AnEnvelopeStamp;
 use Symfony\Component\Messenger\Tests\Fixtures\DummyMessage;
@@ -84,6 +85,15 @@ public function testQueue()
         $this->assertSame([], $this->transport->get());
     }
 
+    public function testQueueWithDelay()
+    {
+        $envelope1 = new Envelope(new \stdClass());
+        $envelope1 = $this->transport->send($envelope1);
+        $envelope2 = (new Envelope(new \stdClass()))->with(new DelayStamp(10_000));
+        $envelope2 = $this->transport->send($envelope2);
+        $this->assertSame([$envelope1], $this->transport->get());
+    }
+
     public function testQueueWithSerialization()
     {
         $envelope = new Envelope(new \stdClass());
diff --git a/src/Symfony/Component/Messenger/Transport/InMemory/InMemoryTransport.php b/src/Symfony/Component/Messenger/Transport/InMemory/InMemoryTransport.php
@@ -13,6 +13,7 @@
 
 use Symfony\Component\Messenger\Envelope;
 use Symfony\Component\Messenger\Exception\LogicException;
+use Symfony\Component\Messenger\Stamp\DelayStamp;
 use Symfony\Component\Messenger\Stamp\TransportMessageIdStamp;
 use Symfony\Component\Messenger\Transport\Serialization\SerializerInterface;
 use Symfony\Component\Messenger\Transport\TransportInterface;
@@ -47,6 +48,7 @@ class InMemoryTransport implements TransportInterface, ResetInterface
 
     private int $nextId = 1;
     private ?SerializerInterface $serializer;
+    private array $availableAt = [];
 
     public function __construct(SerializerInterface $serializer = null)
     {
@@ -55,7 +57,15 @@ public function __construct(SerializerInterface $serializer = null)
 
     public function get(): iterable
     {
-        return array_values($this->decode($this->queue));
+        $envelopes = [];
+        $now = new \DateTimeImmutable();
+        foreach ($this->decode($this->queue) as $id => $envelope) {
+            if (!isset($this->availableAt[$id]) || $now > $this->availableAt[$id]) {
+                $envelopes[] = $envelope;
+            }
+        }
+
+        return $envelopes;
     }
 
     public function ack(Envelope $envelope): void
@@ -66,7 +76,7 @@ public function ack(Envelope $envelope): void
             throw new LogicException('No TransportMessageIdStamp found on the Envelope.');
         }
 
-        unset($this->queue[$transportMessageIdStamp->getId()]);
+        unset($this->queue[$id = $transportMessageIdStamp->getId()], $this->availableAt[$id]);
     }
 
     public function reject(Envelope $envelope): void
@@ -77,7 +87,7 @@ public function reject(Envelope $envelope): void
             throw new LogicException('No TransportMessageIdStamp found on the Envelope.');
         }
 
-        unset($this->queue[$transportMessageIdStamp->getId()]);
+        unset($this->queue[$id = $transportMessageIdStamp->getId()], $this->availableAt[$id]);
     }
 
     public function send(Envelope $envelope): Envelope
@@ -88,6 +98,11 @@ public function send(Envelope $envelope): Envelope
         $this->sent[] = $encodedEnvelope;
         $this->queue[$id] = $encodedEnvelope;
 
+        /** @var DelayStamp|null $delayStamp */
+        if ($delayStamp = $envelope->last(DelayStamp::class)) {
+            $this->availableAt[$id] = (new \DateTimeImmutable())->modify(sprintf('+%d seconds', $delayStamp->getDelay() / 1000));
+        }
+
         return $envelope;
     }
 
