尝试在 Typescript 上编写自定义节流功能

Question

我有以下功能非常好

do {
    try {
        const { assets } = await myApi(arg1, arg2, arg3);     <<--  
        return assets; 
    } catch (err: any) {
        if (err.response.status == 429) {
            await sleep(300);
            console.log('Throttle happened, sleeping for 300ms');
        } else
            throw (err);
    }
}
while (true);

我想把这个方法通用化，用下面的方式调用

throttleHandler(myApi, ...args)

我进行了一些尝试，但打字稿抱怨它。我也不确定我是否做得正确。而且我不想使用any。我需要输入它。

type ThrottleHandler<TOut> = (...args: any) => TOut;

async function throttleHandler<T>(promiseFunction: ThrottleHandler<T>, ...args: any): Promise<T> {
  console.log("args", args, ...args)
  do {
    try {
      return await promiseFunction(...args);
    } catch (err: any) {

      if (err.response.status == 429) {
        await sleep(300);
        console.log('Throttle happened, sleeping for 300ms');
      } else
        throw (err);
    }
  }
  while (true);
}

Answer 1

type PromiseFn = (...args: any[]) => Promise<any>;
type UnwrappedReturnType<T extends PromiseFn> = T extends (...args: any) => Promise<infer R> ? R : any

async function throttleHandler<ApiFn extends PromiseFn>(
  promiseFunction: ApiFn,
  ...args: Parameters<ApiFn>
): Promise<UnwrappedReturnType<ApiFn>> {
  console.log("args", args, ...args)
  do {
    try {
      return await promiseFunction(...args);
    } catch (err: any) {

      if (err.response.status == 429) {
        await sleep(300);
        console.log('Throttle happened, sleeping for 300ms');
      } else
        throw (err);
    }
  }
  while (true);
}

const k = (arg1: string, arg2: number) => {
  console.log(arg1);
  return Promise.resolve(true);
}

const f = throttleHandler(k, 'param', 12);

以前的版本使用新的 TS 4.5 Awaited 类型，但这个在所有最新版本中都一样有效。

Playground Link