Updated MaxAge to properly format the queue attribute when configurin…

…g a RabbitMQ stream
phatboyg · Jun 27, 2024 · e11c12a · e11c12a
1 parent 3e8b049
commit e11c12a
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/...ssTransit.RabbitMqTransport/RabbitMqTransport/Configuration/RabbitMqStreamConfigurator.cs b/...ssTransit.RabbitMqTransport/RabbitMqTransport/Configuration/RabbitMqStreamConfigurator.cs
@@ -9,6 +9,7 @@ public class RabbitMqStreamConfigurator :
     IRabbitMqStreamConfigurator
 {
     static readonly DateTimeTypeConverter _dateTimeConverter = new DateTimeTypeConverter();
+    static readonly TimeSpanTypeConverter _timeSpanConverter = new TimeSpanTypeConverter();
 
     readonly RabbitMqReceiveSettings _settings;
 
@@ -24,7 +25,20 @@ public long MaxLength
 
     public TimeSpan MaxAge
     {
-        set => _settings.QueueArguments["x-max-age"] = value;
+        set
+        {
+            string? text = null;
+            if (value.TotalDays >= 1)
+                text = $"{value.TotalDays:F0}D";
+            else if (value.TotalHours >= 1)
+                text = $"{value.TotalHours:F0}h";
+            else if (value.TotalMinutes >= 1)
+                text = $"{value.TotalMinutes:F0}m";
+            else if (value.TotalSeconds >= 1)
+                text = $"{value.TotalSeconds:F0}s";
+
+            _settings.QueueArguments["x-max-age"] = text;
+        }
     }
 
     public long MaxSegmentSize

diff --git a/tests/MassTransit.RabbitMqTransport.Tests/Stream_Specs.cs b/tests/MassTransit.RabbitMqTransport.Tests/Stream_Specs.cs
@@ -73,8 +73,10 @@ protected override void ConfigureConsumer(IReceiveEndpointConfigurator endpointC
 
             if (endpointConfigurator is IRabbitMqReceiveEndpointConfigurator rmq)
             {
-                rmq.Stream("consumer-tag", s =>
+                rmq.Stream("main-consumer", s =>
                 {
+                    s.MaxAge = TimeSpan.FromDays(14);
+
                     s.FromFirst();
                 });
             }