viem
Version:
45 lines • 1.6 kB
JavaScript
import { getAbortError, isAbortError, } from '../../errors/utils.js';
import { wait } from '../wait.js';
export function withRetry(fn, { delay: delay_ = 100, retryCount = 2, shouldRetry = () => true, signal, } = {}) {
return new Promise((resolve, reject) => {
const attemptRetry = async ({ count = 0 } = {}) => {
if (signal?.aborted) {
reject(getAbortError(signal));
return;
}
const retry = async ({ error }) => {
const delay = typeof delay_ === 'function' ? delay_({ count, error }) : delay_;
if (delay) {
try {
await wait(delay, { signal });
}
catch (err) {
reject(err);
return;
}
}
attemptRetry({ count: count + 1 });
};
try {
const data = await fn();
resolve(data);
}
catch (err) {
if (signal?.aborted) {
reject(getAbortError(signal));
return;
}
if (isAbortError(err)) {
reject(err);
return;
}
if (count < retryCount &&
(await shouldRetry({ count, error: err })))
return retry({ error: err });
reject(err);
}
};
attemptRetry();
});
}
//# sourceMappingURL=withRetry.js.map