@ace-fetch/graphql
Version:
Fetch Provider.
92 lines (91 loc) • 3.77 kB
JavaScript
;
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);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.createRetryPlugin = exports.registRetry = exports.RetryCountSymbol = void 0;
var core_1 = require("@ace-util/core");
var env_1 = require("../env");
exports.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[exports.RetryCountSymbol] = config[exports.RetryCountSymbol] || 0;
// Check if we've maxed out the total number of retries
if (config[exports.RetryCountSymbol] < curOptions_1.maxCount) {
// Increase the retry count
config[exports.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[exports.RetryCountSymbol]) - 1) * 1000 : 1;
(0, core_1.warn)(!env_1.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 () {
(0, core_1.warn)(!env_1.debug, "retry ".concat(config[exports.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
*/
function registRetry(request, options) {
var curOptions = __assign(__assign({}, defaultOptions), options);
var retryRequest = function (config) {
var requestPromise = request(config);
if (!(0, core_1.isPromise)(requestPromise)) {
return requestPromise;
}
return requestPromise.catch(function (error) {
return retryHandler(error, config, curOptions, retryRequest);
});
};
return retryRequest;
}
exports.registRetry = registRetry;
/**
* 注册重试插件
* 只在regist graphqls上运行 (and 自定义条件下)
* @param options 插件配置
*/
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;
}, {});
};
};
exports.createRetryPlugin = createRetryPlugin;