Requeue with a custom visibility timeout

Author: deguif

pkg/sqs/SqsConsumer.php

@@ -138,7 +138,7 @@ public function reject(Message $message, bool $requeue = false): void
                 '@region' => $this->queue->getRegion(),
                 'QueueUrl' => $this->context->getQueueUrl($this->queue),
                 'ReceiptHandle' => $message->getReceiptHandle(),
-                'VisibilityTimeout' => 0,
+                'VisibilityTimeout' => $message->getRequeueVisibilityTimeout(),
             ]);
         } else {
             $this->context->getSqsClient()->deleteMessage([

pkg/sqs/SqsMessage.php

@@ -48,13 +48,19 @@ class SqsMessage implements Message
      */
     private $receiptHandle;
 
+    /**
+     * @var int
+     */
+    private $requeueVisibilityTimeout;
+
     public function __construct(string $body = '', array $properties = [], array $headers = [])
     {
         $this->body = $body;
         $this->properties = $properties;
         $this->headers = $headers;
         $this->redelivered = false;
         $this->delaySeconds = 0;
+        $this->requeueVisibilityTimeout = 0;
     }
 
     public function setBody(string $body): void
@@ -229,4 +235,19 @@ public function getReceiptHandle(): ?string
     {
         return $this->receiptHandle;
     }
+
+    /**
+     * The number of seconds before the message can be visible again when requeuing. Valid values: 0 to 43200. Maximum: 12 hours.
+     *
+     * Set requeue visibility timeout
+     */
+    public function setRequeueVisibilityTimeout(int $seconds): void
+    {
+        $this->requeueVisibilityTimeout = $seconds;
+    }
+
+    public function getRequeueVisibilityTimeout(): int
+    {
+        return $this->requeueVisibilityTimeout;
+    }
 }