106 lines
2.5 KiB
JavaScript
106 lines
2.5 KiB
JavaScript
export class TimeoutError extends Error {
|
|
constructor(message) {
|
|
super(message);
|
|
this.name = 'TimeoutError';
|
|
}
|
|
}
|
|
|
|
/**
|
|
An error to be thrown when the request is aborted by AbortController.
|
|
DOMException is thrown instead of this Error when DOMException is available.
|
|
*/
|
|
export class AbortError extends Error {
|
|
constructor(message) {
|
|
super();
|
|
this.name = 'AbortError';
|
|
this.message = message;
|
|
}
|
|
}
|
|
|
|
/**
|
|
TODO: Remove AbortError and just throw DOMException when targeting Node 18.
|
|
*/
|
|
const getDOMException = errorMessage => globalThis.DOMException === undefined ?
|
|
new AbortError(errorMessage) :
|
|
new DOMException(errorMessage);
|
|
|
|
/**
|
|
TODO: Remove below function and just 'reject(signal.reason)' when targeting Node 18.
|
|
*/
|
|
const getAbortedReason = signal => {
|
|
const reason = signal.reason === undefined ?
|
|
getDOMException('This operation was aborted.') :
|
|
signal.reason;
|
|
|
|
return reason instanceof Error ? reason : getDOMException(reason);
|
|
};
|
|
|
|
export default function pTimeout(promise, milliseconds, fallback, options) {
|
|
let timer;
|
|
|
|
const cancelablePromise = new Promise((resolve, reject) => {
|
|
if (typeof milliseconds !== 'number' || Math.sign(milliseconds) !== 1) {
|
|
throw new TypeError(`Expected \`milliseconds\` to be a positive number, got \`${milliseconds}\``);
|
|
}
|
|
|
|
if (milliseconds === Number.POSITIVE_INFINITY) {
|
|
resolve(promise);
|
|
return;
|
|
}
|
|
|
|
options = {
|
|
customTimers: {setTimeout, clearTimeout},
|
|
...options
|
|
};
|
|
|
|
if (options.signal) {
|
|
const {signal} = options;
|
|
if (signal.aborted) {
|
|
reject(getAbortedReason(signal));
|
|
}
|
|
|
|
signal.addEventListener('abort', () => {
|
|
reject(getAbortedReason(signal));
|
|
});
|
|
}
|
|
|
|
timer = options.customTimers.setTimeout.call(undefined, () => {
|
|
if (typeof fallback === 'function') {
|
|
try {
|
|
resolve(fallback());
|
|
} catch (error) {
|
|
reject(error);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
const message = typeof fallback === 'string' ? fallback : `Promise timed out after ${milliseconds} milliseconds`;
|
|
const timeoutError = fallback instanceof Error ? fallback : new TimeoutError(message);
|
|
|
|
if (typeof promise.cancel === 'function') {
|
|
promise.cancel();
|
|
}
|
|
|
|
reject(timeoutError);
|
|
}, milliseconds);
|
|
|
|
(async () => {
|
|
try {
|
|
resolve(await promise);
|
|
} catch (error) {
|
|
reject(error);
|
|
} finally {
|
|
options.customTimers.clearTimeout.call(undefined, timer);
|
|
}
|
|
})();
|
|
});
|
|
|
|
cancelablePromise.clear = () => {
|
|
clearTimeout(timer);
|
|
timer = undefined;
|
|
};
|
|
|
|
return cancelablePromise;
|
|
}
|