( promise: Promise<T>, cancellation: CancellationToken, message?: string, )
| 24 | * is requested. Otherwise, throws a TaskCancelledError. |
| 25 | */ |
| 26 | export function timeoutPromise<T>( |
| 27 | promise: Promise<T>, |
| 28 | cancellation: CancellationToken, |
| 29 | message?: string, |
| 30 | ): Promise<T> { |
| 31 | if (cancellation.isCancellationRequested) { |
| 32 | return Promise.reject(new TaskCancelledError(message || 'Task cancelled')); |
| 33 | } |
| 34 | |
| 35 | const didTimeout = getDeferred<void>(); |
| 36 | const disposable = cancellation.onCancellationRequested(didTimeout.resolve); |
| 37 | |
| 38 | return Promise.race([ |
| 39 | didTimeout.promise.then(() => { |
| 40 | throw new TaskCancelledError(message || 'Task cancelled'); |
| 41 | }), |
| 42 | promise |
| 43 | .then(r => { |
| 44 | disposable.dispose(); |
| 45 | return r; |
| 46 | }) |
| 47 | .catch(err => { |
| 48 | disposable.dispose(); |
| 49 | throw err; |
| 50 | }), |
| 51 | ]); |
| 52 | } |
| 53 | |
| 54 | /** |
| 55 | * Like Promise.race, but cancels other promises after the first returns. |
no test coverage detected