Fix Redis fallback ack_deadline_ms handling by updating message IDs at receive time

Author: Soroushsrd

omniqueue/src/backends/redis/fallback.rs

@@ -51,29 +51,32 @@ async fn receive_with_timeout<R: RedisConnection>(
     consumer: &RedisConsumer<R>,
     timeout: Duration,
 ) -> Result<Option<Delivery>> {
-    let payload: Option<Vec<u8>> = consumer
-        .redis
-        .get()
-        .await
-        .map_err(QueueError::generic)?
-        .brpoplpush(
-            &consumer.queue_key,
-            &consumer.processing_queue_key,
-            // The documentation at https://redis.io/docs/latest/commands/brpoplpush/ does not
-            // state what unit the timeout is, but `BLPOP` and `BLMPOP` have similar timeout
-            // parameters that are documented as being seconds.
-            timeout.as_secs_f64(),
-        )
+    let mut conn = consumer.redis.get().await.map_err(QueueError::generic)?;
+
+    let payload: Option<Vec<u8>> = conn
+        .brpop(&consumer.queue_key, timeout.as_secs_f64())
         .await
         .map_err(QueueError::generic)?;
 
     match payload {
-        Some(old_payload) => Some(internal_to_delivery(
-            internal_from_list(&old_payload)?.into(),
-            consumer,
-            old_payload,
-        ))
-        .transpose(),
+        Some(old_payload) => {
+            // Creating a new payload with a new timestamp
+            // this is done to avoid the message being re-enqueued
+            // too early!
+            let new_payload = internal_to_list_payload(internal_from_list(&old_payload)?.into());
+
+            let _: () = conn
+                .lpush(&consumer.processing_queue_key, &new_payload)
+                .await
+                .map_err(QueueError::generic)?;
+
+            Some(internal_to_delivery(
+                internal_from_list(&new_payload)?.into(),
+                consumer,
+                new_payload,
+            ))
+            .transpose()
+        }
         None => Ok(None),
     }
 }