41 lines
834 B
JavaScript
41 lines
834 B
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
class TimeoutError extends Error {
|
||
|
|
constructor(message) {
|
||
|
|
super(message);
|
||
|
|
this.name = 'TimeoutError';
|
||
|
|
this.message = message;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = (promise, ms, fallback) => new Promise((resolve, reject) => {
|
||
|
|
if (!Number.isFinite(ms)) {
|
||
|
|
throw new TypeError('Expected `ms` to be a finite number');
|
||
|
|
}
|
||
|
|
|
||
|
|
const timer = setTimeout(() => {
|
||
|
|
if (typeof fallback === 'function') {
|
||
|
|
resolve(fallback());
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
const message = typeof fallback === 'string' ? fallback : `Promise timed out after ${ms} milliseconds`;
|
||
|
|
const err = fallback instanceof Error ? fallback : new TimeoutError(message);
|
||
|
|
|
||
|
|
reject(err);
|
||
|
|
}, ms);
|
||
|
|
|
||
|
|
promise.then(
|
||
|
|
val => {
|
||
|
|
clearTimeout(timer);
|
||
|
|
resolve(val);
|
||
|
|
},
|
||
|
|
err => {
|
||
|
|
clearTimeout(timer);
|
||
|
|
reject(err);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
});
|
||
|
|
|
||
|
|
module.exports.TimeoutError = TimeoutError;
|