@tanstack/query-core
Version:
The framework agnostic core that powers TanStack Query
145 lines (144 loc) • 4.06 kB
JavaScript
import "./chunk-WPSKCR32.js";
// src/retryer.ts
import { focusManager } from "./focusManager.js";
import { onlineManager } from "./onlineManager.js";
import { isServer, sleep } from "./utils.js";
function defaultRetryDelay(failureCount) {
return Math.min(1e3 * 2 ** failureCount, 3e4);
}
function canFetch(networkMode) {
return (networkMode ?? "online") === "online" ? onlineManager.isOnline() : true;
}
var CancelledError = class {
constructor(options) {
this.revert = options == null ? void 0 : options.revert;
this.silent = options == null ? void 0 : options.silent;
}
};
function isCancelledError(value) {
return value instanceof CancelledError;
}
function createRetryer(config) {
let isRetryCancelled = false;
let failureCount = 0;
let isResolved = false;
let continueFn;
let promiseResolve;
let promiseReject;
const promise = new Promise((outerResolve, outerReject) => {
promiseResolve = outerResolve;
promiseReject = outerReject;
});
const cancel = (cancelOptions) => {
var _a;
if (!isResolved) {
reject(new CancelledError(cancelOptions));
(_a = config.abort) == null ? void 0 : _a.call(config);
}
};
const cancelRetry = () => {
isRetryCancelled = true;
};
const continueRetry = () => {
isRetryCancelled = false;
};
const shouldPause = () => !focusManager.isFocused() || config.networkMode !== "always" && !onlineManager.isOnline();
const resolve = (value) => {
var _a;
if (!isResolved) {
isResolved = true;
(_a = config.onSuccess) == null ? void 0 : _a.call(config, value);
continueFn == null ? void 0 : continueFn();
promiseResolve(value);
}
};
const reject = (value) => {
var _a;
if (!isResolved) {
isResolved = true;
(_a = config.onError) == null ? void 0 : _a.call(config, value);
continueFn == null ? void 0 : continueFn();
promiseReject(value);
}
};
const pause = () => {
return new Promise((continueResolve) => {
var _a;
continueFn = (value) => {
const canContinue = isResolved || !shouldPause();
if (canContinue) {
continueResolve(value);
}
return canContinue;
};
(_a = config.onPause) == null ? void 0 : _a.call(config);
}).then(() => {
var _a;
continueFn = void 0;
if (!isResolved) {
(_a = config.onContinue) == null ? void 0 : _a.call(config);
}
});
};
const run = () => {
if (isResolved) {
return;
}
let promiseOrValue;
try {
promiseOrValue = config.fn();
} catch (error) {
promiseOrValue = Promise.reject(error);
}
Promise.resolve(promiseOrValue).then(resolve).catch((error) => {
var _a;
if (isResolved) {
return;
}
const retry = config.retry ?? (isServer ? 0 : 3);
const retryDelay = config.retryDelay ?? defaultRetryDelay;
const delay = typeof retryDelay === "function" ? retryDelay(failureCount, error) : retryDelay;
const shouldRetry = retry === true || typeof retry === "number" && failureCount < retry || typeof retry === "function" && retry(failureCount, error);
if (isRetryCancelled || !shouldRetry) {
reject(error);
return;
}
failureCount++;
(_a = config.onFail) == null ? void 0 : _a.call(config, failureCount, error);
sleep(delay).then(() => {
if (shouldPause()) {
return pause();
}
return;
}).then(() => {
if (isRetryCancelled) {
reject(error);
} else {
run();
}
});
});
};
if (canFetch(config.networkMode)) {
run();
} else {
pause().then(run);
}
return {
promise,
cancel,
continue: () => {
const didContinue = continueFn == null ? void 0 : continueFn();
return didContinue ? promise : Promise.resolve();
},
cancelRetry,
continueRetry
};
}
export {
CancelledError,
canFetch,
createRetryer,
isCancelledError
};
//# sourceMappingURL=retryer.js.map