AFFiNE-Mirror/packages/backend/server/src/fundamentals/utils/promise.ts

import { defer, retry } from 'rxjs';

export class RetryablePromise<T> extends Promise<T> {
  constructor(
    executor: (
      resolve: (value: T | PromiseLike<T>) => void,
      reject: (reason?: any) => void
    ) => void,
    retryTimes: number = 3,
    retryIntervalInMs: number = 300
  ) {
    super((resolve, reject) => {
      defer(() => new Promise<T>(executor))
        .pipe(
          retry({
            count: retryTimes,
            delay: retryIntervalInMs,
          })
        )
        .subscribe({
          next: v => {
            resolve(v);
          },
          error: e => {
            reject(e);
          },
        });
    });
  }
}

export function retryable<Ret = unknown>(
  asyncFn: () => Promise<Ret>,
  retryTimes = 3,
  retryIntervalInMs = 300
): Promise<Ret> {
  return new RetryablePromise<Ret>(
    (resolve, reject) => {
      asyncFn().then(resolve).catch(reject);
    },
    retryTimes,
    retryIntervalInMs
  );
}