|
|
@@ -1,5 +1,112 @@
|
|
|
-// @ts-expect-error -- missing types
|
|
|
-import createFetchRetry from '@vercel/fetch-retry';
|
|
|
+import retry from 'async-retry';
|
|
|
+
|
|
|
+// retry settings
|
|
|
+const MIN_TIMEOUT = 10;
|
|
|
+const MAX_RETRIES = 5;
|
|
|
+const MAX_RETRY_AFTER = 20;
|
|
|
+const FACTOR = 6;
|
|
|
+
|
|
|
+function isClientError(err: any): err is NodeJS.ErrnoException {
|
|
|
+ if (!err) return false;
|
|
|
+ return (
|
|
|
+ err.code === 'ERR_UNESCAPED_CHARACTERS' ||
|
|
|
+ err.message === 'Request path contains unescaped characters'
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+interface FetchRetryOpt {
|
|
|
+ minTimeout?: number,
|
|
|
+ retries?: number,
|
|
|
+ factor?: number,
|
|
|
+ maxRetryAfter?: number,
|
|
|
+ retry?: number,
|
|
|
+ onRetry?: (err: Error) => void
|
|
|
+}
|
|
|
+
|
|
|
+function createFetchRetry($fetch: typeof fetch): typeof fetch {
|
|
|
+ const fetchRetry = async (url: string | URL, opts: RequestInit & { retry?: FetchRetryOpt } = {}) => {
|
|
|
+ const retryOpts = Object.assign(
|
|
|
+ {
|
|
|
+ // timeouts will be [10, 60, 360, 2160, 12960]
|
|
|
+ // (before randomization is added)
|
|
|
+ minTimeout: MIN_TIMEOUT,
|
|
|
+ retries: MAX_RETRIES,
|
|
|
+ factor: FACTOR,
|
|
|
+ maxRetryAfter: MAX_RETRY_AFTER,
|
|
|
+ },
|
|
|
+ opts.retry
|
|
|
+ );
|
|
|
+
|
|
|
+ try {
|
|
|
+ return await retry(async (bail) => {
|
|
|
+ try {
|
|
|
+ // this will be retried
|
|
|
+ const res = await $fetch(url, opts);
|
|
|
+
|
|
|
+ if ((res.status >= 500 && res.status < 600) || res.status === 429) {
|
|
|
+ // NOTE: doesn't support http-date format
|
|
|
+ const retryAfterHeader = res.headers.get('retry-after');
|
|
|
+ if (retryAfterHeader) {
|
|
|
+ const retryAfter = parseInt(retryAfterHeader, 10);
|
|
|
+ if (retryAfter) {
|
|
|
+ if (retryAfter > retryOpts.maxRetryAfter) {
|
|
|
+ return res;
|
|
|
+ } else {
|
|
|
+ await new Promise((r) => setTimeout(r, retryAfter * 1e3));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ throw new ResponseError(res);
|
|
|
+ } else {
|
|
|
+ return res;
|
|
|
+ }
|
|
|
+ } catch (err: unknown) {
|
|
|
+ if (err instanceof Error) {
|
|
|
+ if (err.name === 'AbortError') {
|
|
|
+ return bail(err);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (isClientError(err)) {
|
|
|
+ return bail(err);
|
|
|
+ }
|
|
|
+ throw err;
|
|
|
+ }
|
|
|
+ }, retryOpts);
|
|
|
+ } catch (err) {
|
|
|
+ if (err instanceof ResponseError) {
|
|
|
+ return err.res;
|
|
|
+ }
|
|
|
+ throw err;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for (const k of Object.keys($fetch)) {
|
|
|
+ const key = k as keyof typeof $fetch;
|
|
|
+ fetchRetry[key] = $fetch[key];
|
|
|
+ }
|
|
|
+
|
|
|
+ return fetchRetry as typeof fetch;
|
|
|
+}
|
|
|
+
|
|
|
+export class ResponseError extends Error {
|
|
|
+ readonly res: Response;
|
|
|
+ readonly code: number;
|
|
|
+ readonly statusCode: number;
|
|
|
+ readonly url: string;
|
|
|
+
|
|
|
+ constructor(res: Response) {
|
|
|
+ super(res.statusText);
|
|
|
+
|
|
|
+ if (Error.captureStackTrace) {
|
|
|
+ Error.captureStackTrace(this, ResponseError);
|
|
|
+ }
|
|
|
+
|
|
|
+ this.name = this.constructor.name;
|
|
|
+ this.res = res;
|
|
|
+ this.code = this.statusCode = res.status;
|
|
|
+ this.url = res.url;
|
|
|
+ }
|
|
|
+}
|
|
|
|
|
|
export const defaultRequestInit: RequestInit = {
|
|
|
headers: {
|
|
|
@@ -7,4 +114,4 @@ export const defaultRequestInit: RequestInit = {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-export const fetchWithRetry: typeof fetch = createFetchRetry(fetch);
|
|
|
+export const fetchWithRetry = createFetchRetry(fetch);
|
SukkaW