respect ctx in strategy sleeps

repeater_test.go

@@ -22,10 +22,11 @@ func TestRepeatFixed(t *testing.T) {
 		return e
 	}
 
-	err := NewDefault(10, time.Millisecond).Do(context.Background(), fun)
+	st := time.Now()
+	err := NewDefault(10, time.Millisecond*10).Do(context.Background(), fun)
 	assert.Nil(t, err, "should be ok")
 	assert.Equal(t, 5, called, "called 5 times")
-
+	assert.True(t, time.Since(st) >= 40*time.Millisecond)
 	called = 0
 	err = NewDefault(4, time.Millisecond).Do(context.Background(), fun)
 	assert.NotNil(t, err, "should be err")
@@ -76,9 +77,11 @@ func TestRepeatFixedCanceled(t *testing.T) {
 		return errors.New("some error")
 	}
 
+	st := time.Now()
 	err := NewDefault(10, time.Millisecond*50).Do(ctx, fun)
 	assert.EqualError(t, err, "some error")
 	assert.Equal(t, 2, called)
+	assert.True(t, time.Since(st) >= time.Millisecond*60 && time.Since(st) < time.Millisecond*70)
 }
 
 func TestRepeatFixedCriticalError(t *testing.T) {

strategy/backoff.go

@@ -47,7 +47,7 @@ func (b *Backoff) Start(ctx context.Context) (ch chan struct{}) {
 				if b.jitter {
 					delay = rnd.Float64()*(float64(2*minDelay)) + (delay - float64(minDelay))
 				}
-				time.Sleep(time.Duration(delay))
+				sleep(ctx, time.Duration(delay))
 			}
 		}
 	}()

strategy/fixed.go

@@ -33,7 +33,7 @@ func (s *FixedDelay) Start(ctx context.Context) (ch chan struct{}) {
 				return
 			default:
 				ch <- struct{}{}
-				time.Sleep(s.delay)
+				sleep(ctx, s.delay)
 			}
 		}
 	}()

strategy/strategy.go

@@ -2,7 +2,10 @@
 // Strategy result is a channel acting like time.Timer ot time.Tick
 package strategy
 
-import "context"
+import (
+	"context"
+	"time"
+)
 
 // Interface for repeater strategy. Returns channel with ticks
 type Interface interface {
@@ -26,3 +29,12 @@ func (s *Once) Start(ctx context.Context) (ch chan struct{}) {
 	}()
 	return ch
 }
+
+func sleep(ctx context.Context, duration time.Duration) {
+	select {
+	case <-time.After(duration):
+		return
+	case <-ctx.Done():
+		return
+	}
+}