Adding support for delayed messages on SqsBatchedDispatcher

Im5tu · Apr 8, 2021 · e296897 · e296897
1 parent 9dd67fc
commit e296897
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/OpenMessage.AWS.SQS/SqsBatchedDispatcher.cs b/src/OpenMessage.AWS.SQS/SqsBatchedDispatcher.cs
@@ -56,10 +56,13 @@ public override async Task DispatchAsync(Message<T> message, CancellationToken c
             {
                 Id = Guid.NewGuid().ToString("N"),
                 MessageAttributes = GetMessageProperties(message),
-                DelaySeconds = DelaySeconds(message),
                 MessageBody = json
             };
 
+            var delay = DelaySeconds(message);
+            if (delay.HasValue)
+                request.DelaySeconds = delay.Value;
+
             var msg = new SendSqsMessageCommand
             {
                 Message = request,
@@ -81,14 +84,14 @@ public override async Task DispatchAsync(Message<T> message, CancellationToken c
             }
         }
 
-        private static int DelaySeconds(Message<T> message)
+        private static int? DelaySeconds(Message<T> message)
         {
             if (message is ISupportSendDelay delay && delay.SendDelay > TimeSpan.Zero)
             {
                 return Math.Min(MaximumSqsDelaySeconds, (int) delay.SendDelay.TotalSeconds);
             }
 
-            return 0;
+            return null;
         }
 
         private Dictionary<string, MessageAttributeValue> GetMessageProperties(Message<T> message)