|
| 1 | +'use strict' |
| 2 | + |
| 3 | +/** |
| 4 | + * @type {<T = void, R = T>(time: number, options?: { promise?: Promise<T>; fallback?: () => R | Promise<R>; signal?: AbortSignal; rejection?: boolean; }) => Promise<T | Awaited<R>>} |
| 5 | + */ |
| 6 | +async function timeout (time, options = {}) { |
| 7 | + let timer |
| 8 | + /** @type {() => void} */ |
| 9 | + let abort |
| 10 | + const { promise, fallback, signal, rejection } = options |
| 11 | + if (signal?.aborted) { |
| 12 | + throw new DOMException('The operation was aborted.', 'AbortError') |
| 13 | + } |
| 14 | + const delay = new Promise((resolve, reject) => { |
| 15 | + abort = () => { |
| 16 | + clearTimeout(timer) |
| 17 | + reject( |
| 18 | + signal?.reason ?? |
| 19 | + new DOMException('The operation was aborted.', 'AbortError') |
| 20 | + ) |
| 21 | + } |
| 22 | + signal?.addEventListener('abort', abort) |
| 23 | + timer = setTimeout(() => { |
| 24 | + if (typeof fallback === 'function') { |
| 25 | + try { |
| 26 | + resolve(fallback()) |
| 27 | + } catch (err) { |
| 28 | + reject(err) |
| 29 | + } |
| 30 | + } else { |
| 31 | + if (rejection === false) { |
| 32 | + resolve(undefined) |
| 33 | + } else { |
| 34 | + reject(new DOMException('The operation was aborted.', 'AbortError')) |
| 35 | + } |
| 36 | + } |
| 37 | + }, time) |
| 38 | + }) |
| 39 | + try { |
| 40 | + return promise ? await Promise.race([promise, delay]) : await delay |
| 41 | + } finally { |
| 42 | + clearTimeout(timer) |
| 43 | + signal?.removeEventListener('abort', abort) |
| 44 | + } |
| 45 | +} |
| 46 | + |
| 47 | +/** |
| 48 | + * @type {<T, R = T>(promise: Promise<T>, time: number, fallback?: () => R | Promise<R>) => Promise<Awaited<R> | T>} |
| 49 | + */ |
| 50 | +function pTimeout (promise, time, fallback) { |
| 51 | + return timeout(time, { promise, fallback }) |
| 52 | +} |
| 53 | + |
| 54 | +/** |
| 55 | + * @param {number} time |
| 56 | + */ |
| 57 | +function delay (time) { |
| 58 | + return timeout(time, { rejection: false }) |
| 59 | +} |
| 60 | + |
| 61 | +module.exports = { |
| 62 | + timeout, |
| 63 | + delay, |
| 64 | + pTimeout |
| 65 | +} |
0 commit comments