abort-controller-x/lib/proactiveRetry.js

Abortable async function helpers

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.proactiveRetry = void 0;
const AbortError_1 = require("./AbortError");
const delay_1 = require("./delay");
const execute_1 = require("./execute");
/**
 * Proactively retry a function with exponential backoff.
 *
 * Also known as hedging.
 *
 * The function will be called multiple times in parallel until it succeeds, in
 * which case all the other calls will be aborted.
 */
function proactiveRetry(signal, fn, options = {}) {
    const { baseMs = 1000, onError, maxAttempts = Infinity } = options;
    return (0, execute_1.execute)(signal, (resolve, reject) => {
        const innerAbortController = new AbortController();
        let attemptsExhausted = false;
        const promises = new Map();
        function handleFulfilled(value) {
            innerAbortController.abort();
            promises.clear();
            resolve(value);
        }
        function handleRejected(err, attempt) {
            promises.delete(attempt);
            if (attemptsExhausted && promises.size === 0) {
                reject(err);
                return;
            }
            if ((0, AbortError_1.isAbortError)(err)) {
                return;
            }
            if (onError) {
                try {
                    onError(err, attempt);
                }
                catch (err) {
                    innerAbortController.abort();
                    promises.clear();
                    reject(err);
                }
            }
        }
        async function makeAttempts(signal) {
            for (let attempt = 0;; attempt++) {
                const promise = fn(signal, attempt);
                promises.set(attempt, promise);
                promise.then(handleFulfilled, err => handleRejected(err, attempt));
                if (attempt + 1 >= maxAttempts) {
                    break;
                }
                // https://aws.amazon.com/ru/blogs/architecture/exponential-backoff-and-jitter/
                const backoff = Math.pow(2, attempt) * baseMs;
                const delayMs = Math.round((backoff * (1 + Math.random())) / 2);
                await (0, delay_1.delay)(signal, delayMs);
            }
            attemptsExhausted = true;
        }
        makeAttempts(innerAbortController.signal).catch(AbortError_1.catchAbortError);
        return () => {
            innerAbortController.abort();
        };
    });
}
exports.proactiveRetry = proactiveRetry;
//# sourceMappingURL=proactiveRetry.js.map