You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
3.5 KiB
92 lines
3.5 KiB
import { __awaiter, __extends, __generator } from "tslib"; |
|
import { ApolloLink } from "../core/index.js"; |
|
import { Observable } from "../../utilities/index.js"; |
|
import { buildDelayFunction } from "./delayFunction.js"; |
|
import { buildRetryFunction } from "./retryFunction.js"; |
|
/** |
|
* Tracking and management of operations that may be (or currently are) retried. |
|
*/ |
|
var RetryableOperation = /** @class */ (function () { |
|
function RetryableOperation(observer, operation, forward, delayFor, retryIf) { |
|
var _this = this; |
|
this.observer = observer; |
|
this.operation = operation; |
|
this.forward = forward; |
|
this.delayFor = delayFor; |
|
this.retryIf = retryIf; |
|
this.retryCount = 0; |
|
this.currentSubscription = null; |
|
this.onError = function (error) { return __awaiter(_this, void 0, void 0, function () { |
|
var shouldRetry; |
|
return __generator(this, function (_a) { |
|
switch (_a.label) { |
|
case 0: |
|
this.retryCount += 1; |
|
return [4 /*yield*/, this.retryIf(this.retryCount, this.operation, error)]; |
|
case 1: |
|
shouldRetry = _a.sent(); |
|
if (shouldRetry) { |
|
this.scheduleRetry(this.delayFor(this.retryCount, this.operation, error)); |
|
return [2 /*return*/]; |
|
} |
|
this.observer.error(error); |
|
return [2 /*return*/]; |
|
} |
|
}); |
|
}); }; |
|
this.try(); |
|
} |
|
/** |
|
* Stop retrying for the operation, and cancel any in-progress requests. |
|
*/ |
|
RetryableOperation.prototype.cancel = function () { |
|
if (this.currentSubscription) { |
|
this.currentSubscription.unsubscribe(); |
|
} |
|
clearTimeout(this.timerId); |
|
this.timerId = undefined; |
|
this.currentSubscription = null; |
|
}; |
|
RetryableOperation.prototype.try = function () { |
|
this.currentSubscription = this.forward(this.operation).subscribe({ |
|
next: this.observer.next.bind(this.observer), |
|
error: this.onError, |
|
complete: this.observer.complete.bind(this.observer), |
|
}); |
|
}; |
|
RetryableOperation.prototype.scheduleRetry = function (delay) { |
|
var _this = this; |
|
if (this.timerId) { |
|
throw new Error("RetryLink BUG! Encountered overlapping retries"); |
|
} |
|
this.timerId = setTimeout(function () { |
|
_this.timerId = undefined; |
|
_this.try(); |
|
}, delay); |
|
}; |
|
return RetryableOperation; |
|
}()); |
|
var RetryLink = /** @class */ (function (_super) { |
|
__extends(RetryLink, _super); |
|
function RetryLink(options) { |
|
var _this = _super.call(this) || this; |
|
var _a = options || {}, attempts = _a.attempts, delay = _a.delay; |
|
_this.delayFor = |
|
typeof delay === "function" ? delay : buildDelayFunction(delay); |
|
_this.retryIf = |
|
typeof attempts === "function" ? attempts : buildRetryFunction(attempts); |
|
return _this; |
|
} |
|
RetryLink.prototype.request = function (operation, nextLink) { |
|
var _this = this; |
|
return new Observable(function (observer) { |
|
var retryable = new RetryableOperation(observer, operation, nextLink, _this.delayFor, _this.retryIf); |
|
return function () { |
|
retryable.cancel(); |
|
}; |
|
}); |
|
}; |
|
return RetryLink; |
|
}(ApolloLink)); |
|
export { RetryLink }; |
|
//# sourceMappingURL=retryLink.js.map
|