@ace-fetch/graphql/esm/plugins/retry.js

Fetch Provider.

var __assign = (this && this.__assign) || function () {
    __assign = Object.assign || function(t) {
        for (var s, i = 1, n = arguments.length; i < n; i++) {
            s = arguments[i];
            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
                t[p] = s[p];
        }
        return t;
    };
    return __assign.apply(this, arguments);
};
import { isPromise, warn } from '@ace-util/core';
import { debug } from '../env';
export var RetryCountSymbol = '__RetryCount__';
var defaultOptions = {
    maxCount: 3,
    delay: true,
    /**
     * The request was made but no response was received
     * `error.request` is an instance of XMLHttpRequest in the browser and an instance of
     * http.ClientRequest in node.js
     */
    validateError: function (error) { return error.message === 'Network Error'; },
};
function retryHandler(error, config, options, retryRequest) {
    if (!!(config === null || config === void 0 ? void 0 : config.retry)) {
        var curOptions_1 = typeof config.retry === 'boolean' ? options : __assign(__assign({}, options), config.retry);
        if (!curOptions_1.validateError(error))
            return Promise.reject(error);
        // Set the variable for keeping track of the retry count
        config[RetryCountSymbol] = config[RetryCountSymbol] || 0;
        // Check if we've maxed out the total number of retries
        if (config[RetryCountSymbol] < curOptions_1.maxCount) {
            // Increase the retry count
            config[RetryCountSymbol] += 1;
            // Create new promise to handle exponential backoff.
            // formula(2 ^ c - 1 / 2) * 1000(for mS to seconds)
            var backoff = new Promise(function (resolve) {
                var backOffDelay = curOptions_1.delay ? (1 / 2) * (Math.pow(2, config[RetryCountSymbol]) - 1) * 1000 : 1;
                warn(!debug, "retry delay ".concat(backOffDelay, "ms"));
                setTimeout(function () {
                    resolve(null);
                }, backOffDelay);
            });
            // Return the promise in which recalls axios to retry the request
            return backoff.then(function () {
                warn(!debug, "retry ".concat(config[RetryCountSymbol], " time(s)"));
                return retryRequest(config);
            });
        }
    }
    return Promise.reject(error);
}
/**
 * regist retry plugin on current promise request
 * @param request request promise
 * @param options catch error options
 */
export function registRetry(request, options) {
    var curOptions = __assign(__assign({}, defaultOptions), options);
    var retryRequest = function (config) {
        var requestPromise = request(config);
        if (!isPromise(requestPromise)) {
            return requestPromise;
        }
        return requestPromise.catch(function (error) {
            return retryHandler(error, config, curOptions, retryRequest);
        });
    };
    return retryRequest;
}
/**
 * 注册重试插件
 * 只在regist graphqls上运行 (and 自定义条件下)
 * @param options 插件配置
 */
export var createRetryPlugin = function (options) {
    if (options === void 0) { options = {}; }
    return function (_a) {
        var registGraphqls = _a.registGraphqls;
        return Object.keys(registGraphqls).reduce(function (prev, key) {
            prev[key] = registRetry(registGraphqls[key], options);
            return prev;
        }, {});
    };
};