Pass BaseResponse to when()
This makes it clear that callers shouldn't try to read the response
body in the callback.
diff --git a/lib/http_retry.dart b/lib/http_retry.dart
index 03cebd8..b50d680 100644
--- a/lib/http_retry.dart
+++ b/lib/http_retry.dart
@@ -17,7 +17,7 @@
final int _retries;
/// The callback that determines whether a request should be retried.
- final bool Function(StreamedResponse) _when;
+ final bool Function(BaseResponse) _when;
/// The callback that determines how long to wait before retrying a request.
final Duration Function(int) _delay;
@@ -37,7 +37,7 @@
/// given (zero-based) retry.
RetryClient(this._inner,
{int retries,
- bool when(StreamedResponse response),
+ bool when(BaseResponse response),
Duration delay(int retryCount)})
: _retries = retries ?? 3,
_when = when ?? ((response) => response.statusCode == 503),
@@ -54,13 +54,15 @@
/// in order. It will wait for `delays[0]` after the initial request,
/// `delays[1]` after the first retry, and so on.
RetryClient.withDelays(Client inner, Iterable<Duration> delays,
- {bool when(StreamedResponse response)})
+ {bool when(BaseResponse response)})
: this._withDelays(inner, delays.toList(), when: when);
RetryClient._withDelays(Client inner, List<Duration> delays,
- {bool when(StreamedResponse response)})
+ {bool when(BaseResponse response)})
: this(inner,
- retries: delays.length, delay: (retryCount) => delays[retryCount]);
+ retries: delays.length,
+ delay: (retryCount) => delays[retryCount],
+ when: when);
Future<StreamedResponse> send(BaseRequest request) async {
var splitter = new StreamSplitter(request.finalize());
