Skip to content

StimulCross/p-retry

 
 

Repository files navigation

This is a fork of the popular p-retry library with support for both ESM and CommonJS module systems.

Note

Difference from the original library:
This fork includes a modified implementation of the AbortError class, as well as improved integration with AbortController.

  • AbortError now fully supports the standard cause property. You can pass an error cause via the second constructor argument:

    throw new AbortError('Operation aborted', { cause: originalError });
  • The pRetry function throws an instance of AbortError directly when aborted, instead of throwing the original error. If you need access to the original error, you can retrieve it via the cause property.

  • When AbortController.abort(reason) is used, the thrown AbortError will have:

    • cause set to signal.reason, if available,

    • signal set to the corresponding AbortSignal, allowing consumers to inspect it:

      const controller = new AbortController();
      controller.abort(new Error('Database unavailable'));
      
      try {
      	await pRetry(fn, { signal: controller.signal });
      } catch (e) {
      	if (e instanceof AbortError) {
      		console.error('Aborted because:', e.cause); // → Error('Database unavailable')
      		console.log(e.signal.aborted); // → true
      	}
      }
  • The AbortError class also exposes two static helper methods for convenience:

    AbortError.fromSignal(signal, 'Aborted by signal'); // → includes signal and cause
    AbortError.fromError(originalError, 'Aborted by error'); // → wraps an error as cause

p-retry

Retry a promise-returning or async function

It does exponential backoff and supports custom retry strategies for failed operations.

Install

npm install @stimulcross/p-retry
yarn add @stimulcross/p-retry
pnpm add @stimulcross/p-retry

Usage

import { AbortError, pRetry } from '@stimulcross/p-retry';

async function run() {
	try {
		// your logic that may throw
		throw new ValidationError();
	} catch (err) {
		if (err instanceof ValidationError) {
			// Abort all retries and preserve the original cause
			throw new AbortError('Aborting due to validation error.', { cause: err });
		}

		// Other errors will be retried by the library.
		throw err;
	}
}

try {
	await pRetry(run, { retries: 5 });
} catch (error) {
	if (error instanceof AbortError) {
		console.error('Aborted with message:', error.message);
		console.error('Original cause:', error.cause);
	}
}

console.log(await pRetry(run, { retries: 5 }));

API

pRetry(input, options?)

Returns a Promise that is fulfilled when calling input returns a fulfilled promise. If calling input returns a rejected promise, input is called again until the max retries are reached, it then rejects with the last rejection reason.

Does not retry on most TypeErrors, with the exception of network errors. This is done on a best case basis as different browsers have different messages to indicate this. See whatwg/fetch#526 (comment)

input

Type: Function

Receives the number of attempts as the first argument and is expected to return a Promise or any value.

options

Type: object

onFailedAttempt(context)

Type: Function

Callback invoked on each retry. Receives a context object containing the error and retry state information.

import { pRetry } from '@stimulcross/p-retry';

const run = async () => {
	const response = await fetch('https://sindresorhus.com/unicorn');

	if (!response.ok) {
		throw new Error(response.statusText);
	}

	return response.json();
};

const result = await pRetry(run, {
	onFailedAttempt: ({ error, attemptNumber, retriesLeft }) => {
		console.log(`Attempt ${attemptNumber} failed. There are ${retriesLeft} retries left.`);
		// 1st request => Attempt 1 failed. There are 5 retries left.
		// 2nd request => Attempt 2 failed. There are 4 retries left.
		// …
	},
	retries: 5,
});

console.log(result);

The onFailedAttempt function can return a promise. For example, to add a delay:

import { pRetry } from '@stimulcross/p-retry';
import delay from 'delay';

const run = async () => {  };

const result = await pRetry(run, {
	onFailedAttempt: async () => {
		console.log('Waiting for 1 second before retrying');
		await delay(1000);
	}
});

If the onFailedAttempt function throws, all retries will be aborted and the original promise will reject with the thrown error.

shouldRetry(context)

Type: Function

Decide if a retry should occur based on the context. Returning true triggers a retry, false aborts with the error.

It is not called for TypeError (except network errors) and AbortError.

import { pRetry } from '@stimulcross/p-retry';

const run = async () => {  };

const result = await pRetry(run, {
	shouldRetry: ({error, attemptNumber, retriesLeft}) => !(error instanceof CustomError);
});

In the example above, the operation will be retried unless the error is an instance of CustomError.

retries

Type: number
Default: 10

The maximum amount of times to retry the operation.

factor

Type: number
Default: 2

The exponential factor to use.

minTimeout

Type: number
Default: 1000

The number of milliseconds before starting the first retry.

maxTimeout

Type: number
Default: Infinity

The maximum number of milliseconds between two retries.

randomize

Type: boolean
Default: false

Randomizes the timeouts by multiplying with a factor between 1 and 2.

maxRetryTime

Type: number
Default: Infinity

The maximum time (in milliseconds) that the retried operation is allowed to run.

signal

Type: AbortSignal

You can abort retrying using AbortController.

import { pRetry } from '@stimulcross/p-retry';

const run = async () => {  };
const controller = new AbortController();

cancelButton.addEventListener('click', () => {
	controller.abort(new Error('User clicked cancel button'));
});

try {
	await pRetry(run, {signal: controller.signal});
} catch (error) {
	console.log(error.message);
	//=> 'User clicked cancel button'
}
unref

Type: boolean
Default: false

Prevents retry timeouts from keeping the process alive.

Only affects platforms with a .unref() method on timeouts, such as Node.js.

makeRetriable(function, options?)

Wrap a function so that each call is automatically retried on failure.

import { makeRetriable } from '@stimulcross/p-retry';

const fetchWithRetry = makeRetriable(fetch, { retries: 5 });

const response = await fetchWithRetry('https://sindresorhus.com/unicorn');

AbortError(message, {cause})

Abort retrying and reject the promise.

message

Type: string

An error message.

errorOptions

Type: object

Options with cause property.

Tip

You can pass arguments to the function being retried by wrapping it in an inline arrow function:

import { pRetry } from '@stimulcross/p-retry';

const run = async emoji => {
	// …
};

// Without arguments
await pRetry(run, { retries: 5 });

// With arguments
await pRetry(() => run('🦄'), { retries: 5 });

FAQ

How do I mock timers when testing with this package?

The package uses setTimeout and clearTimeout from the global scope, so you can use the Node.js test timer mocking or a package like sinon.

How do I stop retries when the process receives SIGINT (Ctrl+C)?

Use an AbortController to signal cancellation on SIGINT, and pass its signal to pRetry:

import { pRetry } from '@stimulcross/p-retry';

const controller = new AbortController();

process.once('SIGINT', () => {
	controller.abort(new Error('SIGINT received'));
});

try {
	await pRetry(run, { signal: controller.signal });
} catch (error) {
	console.log('Retry stopped due to:', error.message);
}

The package does not handle process signals itself to avoid global side effects.

Related

About

Retry a promise-returning or async function

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • TypeScript 93.8%
  • JavaScript 6.2%