Chromium Code Reviews| Index: sdk/lib/_internal/pub/lib/src/utils.dart |
| diff --git a/sdk/lib/_internal/pub/lib/src/utils.dart b/sdk/lib/_internal/pub/lib/src/utils.dart |
| index 6a048cf1612a5184c6c0b0d309dc4cec16c561c9..da8f640e85dbed58d61939d658fcb017c0310d53 100644 |
| --- a/sdk/lib/_internal/pub/lib/src/utils.dart |
| +++ b/sdk/lib/_internal/pub/lib/src/utils.dart |
| @@ -168,6 +168,46 @@ void chainToCompleter(Future future, Completer completer) { |
| onError: (e) => completer.completeError(e)); |
| } |
| +/// Ensures that [stream] can emit at least one value successfully (or close |
| +/// without any values). |
| +/// |
| +/// For example, reading asynchronously from a non-existent file will return a |
| +/// stream that fails on the first chunk. In order to handle that more |
| +/// gracefully, you may want to check that the stream looks like it's working |
| +/// before you pipe the stream to something else. |
| +/// |
| +/// This lets you do that. It returns a [Future] that completes to a [Stream] |
| +/// emitting the same values and errors as [stream], but only if at least one |
| +/// value can be read successfully. If an error occurs before any values are |
| +/// emitted, the returned Future completes to that error. |
| +Future<Stream> validateStream(Stream stream) { |
| + var completer = new Completer<Stream>(); |
| + var controller = new StreamController(sync: true); |
| + |
| + StreamSubscription subscription; |
| + subscription = stream.listen((value) { |
| + // We got a value, so the stream is valid. |
| + if (!completer.isCompleted) completer.complete(controller.stream); |
| + controller.add(value); |
| + }, onError: (error) { |
| + // If the error came after values, it's OK. |
| + if (completer.isCompleted) controller.addError(error); |
|
nweiz
2013/08/12 21:35:08
I think you want to return here. Right now you'll
Bob Nystrom
2013/08/12 21:45:00
Oops, accidentally deleted that when I reordered t
|
| + |
| + // Otherwise, the error came first and the stream is invalid. |
| + completer.completeError(error); |
| + |
| + // We don't be returning the stream at all in this case, so unsubscribe |
| + // and swallow the error. |
| + subscription.cancel(); |
| + }, onDone: () { |
| + // It closed with no errors, so the stream is valid. |
| + if (!completer.isCompleted) completer.complete(controller.stream); |
| + controller.close(); |
| + }); |
| + |
| + return completer.future; |
| +} |
| + |
| // TODO(nweiz): remove this when issue 7964 is fixed. |
| /// Returns a [Future] that will complete to the first element of [stream]. |
| /// Unlike [Stream.first], this is safe to use with single-subscription streams. |