Listen for context cancellation during backoff period (#44)

awesome!
sethgrid · Jan 14, 2022 · c02ad50 · c02ad50
1 parent d2ad9ec
commit c02ad50
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pester.go b/pester.go
@@ -335,8 +335,13 @@ func (c *Client) pester(p params) (*http.Response, error) {
 					resp.Body.Close()
 				}
 
+				select {
 				// prevent a 0 from causing the tick to block, pass additional microsecond
-				<-time.After(c.Backoff(i) + 1*time.Microsecond)
+				case <-time.After(c.Backoff(i) + 1*time.Microsecond):
+				// allow context cancellation to cancel during backoff
+				case <-req.Context().Done():
+					return
+				}
 			}
 		}(n, request)