| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file | |
| 2 // for details. All rights reserved. Use of this source code is governed by a | |
| 3 // BSD-style license that can be found in the LICENSE file. | |
| 4 | |
| 5 library substitute_future; | |
| 6 | |
| 7 import 'dart:async'; | |
| 8 | |
| 9 /// A wrapper for [Future] that allows other [Future]s to be substituted in as | |
| 10 /// the wrapped [Future]. This is used for injecting timeout errors into | |
| 11 /// long-running [Future]s. | |
| 12 class SubstituteFuture<T> implements Future<T> { | |
| 13 /// The wrapped [Future]. | |
| 14 Future<T> _inner; | |
| 15 | |
| 16 /// The completer that corresponds to [this]'s result. | |
| 17 final Completer<T> _completer = new Completer<T>(); | |
| 18 | |
| 19 /// Whether or not [this] has been completed yet. | |
| 20 bool _complete = false; | |
| 21 | |
| 22 SubstituteFuture(Future wrapped) { | |
| 23 substitute(wrapped); | |
| 24 } | |
| 25 | |
| 26 Stream<T> asStream() => _completer.future.asStream(); | |
| 27 Future catchError(onError(AsyncError asyncError), {bool test(error)}) => | |
| 28 _completer.future.catchError(onError, test: test); | |
| 29 Future then(onValue(T value), {onError(AsyncError asyncError)}) => | |
| 30 _completer.future.then(onValue, onError: onError); | |
| 31 Future<T> whenComplete(action()) => _completer.future.whenComplete(action); | |
| 32 | |
| 33 /// Substitutes [newFuture] for the currently wrapped [Future], which is | |
| 34 /// returned. | |
| 35 Future<T> substitute(Future<T> newFuture) { | |
| 36 if (_complete) { | |
| 37 throw new StateError("You may not call substitute on a SubstituteFuture " | |
| 38 "that's already complete."); | |
| 39 } | |
| 40 | |
| 41 var oldFuture = _inner; | |
| 42 _inner = newFuture; | |
| 43 _inner.then((value) { | |
| 44 if (_inner != newFuture) return; | |
| 45 _completer.complete(value); | |
| 46 _complete = true; | |
| 47 }).catchError((error) { | |
| 48 if (_inner != newFuture) return; | |
| 49 _completer.completeError(error); | |
| 50 _complete = true; | |
| 51 }); | |
| 52 return oldFuture; | |
| 53 } | |
| 54 } | |
| OLD | NEW |