Abort from backoff sleep period in redis subscription.

blocknative · Aug 8, 2023 · 48b9796 · 48b9796
1 parent 7b5d4c0
commit 48b9796
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/stream/transport/redis/redis.go b/stream/transport/redis/redis.go
@@ -68,7 +68,12 @@ func (s Subscription) Next(ctx context.Context) (transport.Message, error) {
 				WithField("attempt", b.Attempt()).
 				WithField("backoff", b.ForAttempt(b.Attempt())).
 				Warn("failed to get subscription message from redis")
-			time.Sleep(b.Duration())
+			select {
+			case <-time.After(b.Duration()):
+			case <-ctx.Done():
+				return transport.Message{}, ctx.Err()
+			}
+
 			continue
 		}