Merge pull request #191 from taki-mekhalfa/feat/add_queue_message_expiration_option_for_the_consumer

feat: add queue message expiration option for consumer

Author: wagslane

consumer_options.go

@@ -1,6 +1,8 @@
 package rabbitmq
 
 import (
+	"time"
+
 	amqp "github.com/rabbitmq/amqp091-go"
 	"github.com/wagslane/go-rabbitmq/internal/logger"
 )
@@ -329,3 +331,16 @@ func WithConsumerOptionsQueueQuorum(options *ConsumerOptions) {
 
 	options.QueueOptions.Args["x-queue-type"] = "quorum"
 }
+
+// WithConsumerOptionsQueueMessageExpiration sets the message expiration (TTL) for all messages in the queue.
+// This option defines how long a message can remain in the queue before it is discarded if not consumed.
+// The TTL is specified as a time.Duration and will be converted to milliseconds for RabbitMQ.
+// See https://www.rabbitmq.com/docs/ttl#per-queue-message-ttl
+func WithConsumerOptionsQueueMessageExpiration(ttl time.Duration) func(*ConsumerOptions) {
+	return func(options *ConsumerOptions) {
+		if options.QueueOptions.Args == nil {
+			options.QueueOptions.Args = Table{}
+		}
+		options.QueueOptions.Args["x-message-ttl"] = ttl.Milliseconds()
+	}
+}