38 lines
974 B
TypeScript
38 lines
974 B
TypeScript
import * as fetchRetryFactory from "fetch-retry";
|
|
import type { RequestInitRetryParams } from "fetch-retry";
|
|
import type { Logger } from "src/tracing/logger";
|
|
|
|
function getUrlFromInput(input: RequestInfo | URL): string {
|
|
if (input instanceof URL) {
|
|
return input.href;
|
|
}
|
|
if (typeof input === "string") {
|
|
return input;
|
|
}
|
|
return input.url;
|
|
}
|
|
|
|
export function retriedFetchFactory(
|
|
logger: Logger,
|
|
fetch: typeof globalThis.fetch = globalThis.fetch
|
|
) {
|
|
return async (
|
|
input: RequestInfo | URL,
|
|
init: RequestInitRetryParams<typeof fetch> = {}
|
|
): Promise<Response> => {
|
|
return fetchRetryFactory.default(fetch)(input, {
|
|
retryOn: function (attempt, error, response) {
|
|
if (error !== null || !response || response.status >= 500) {
|
|
logger.warn(
|
|
`Retrying fetch for ${getUrlFromInput(input)}, attempt ${attempt}`
|
|
);
|
|
|
|
return true;
|
|
}
|
|
return false;
|
|
},
|
|
retryDelay: (attempt) => Math.pow(1.5, attempt) * 500,
|
|
...init
|
|
});
|
|
};
|
|
}
|