fix #158: preserve context values

kamilsk · Oct 29, 2020 · a3f6ed5 · a3f6ed5
1 parent 7a5261c
commit a3f6ed5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/retry.go b/retry.go
@@ -30,7 +30,11 @@ func Do(
 	strategies ...func(strategy.Breaker, uint, error) bool,
 ) error {
 	var err, clean error
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, is := breaker.(context.Context)
+	if !is {
+		ctx = context.Background()
+	}
+	ctx, cancel := context.WithCancel(ctx)
 	for attempt, should := uint(0), true; should; attempt++ {
 		clean = unwrap(err)
 		for i, repeat := 0, len(strategies); should && i < repeat; i++ {