Typescript 包装函数时保留泛型

发布于04月02日

我试图定义一个可以包装任意函数的函数，同时保留参数类型和返回类型.我让它在函数没有泛型的情况下工作，但我很难让它与泛型一起工作.下面是一个简化的例子:

function wrap<F extends (...args: any[]) => any>(test: F) {
  return (...args: Parameters<typeof test>): ReturnType<typeof test> => {
    return test(...args);
  };
}

function simpleTest(a: number): number {
  return a;
}

// works
// type: (a: number) => number
const wrappedSimpleTest = wrap(simpleTest);

function genericTest<T>(a: T): T {
  return a;
}

// doesn't work
// type: (a: unknown) => unknown
// desired type: <T>(a: T) => T
const wrappedGenericTest = wrap(genericTest);

function genericTest2<T, U>(a: T, b: U): T|U {
  return Math.random() < 0.5 ? a : b;
}

// doesn't work
// type: (a: unknown, b: unknown) => unknown
// desired type: <T, U>(a: T, b: U) => T|U
const wrappedGenericTest2 = wrap(genericTest2);

推荐答案

您在这里寻找的特定行为可以通过使用microsoft/TypeScript#30215中实现的TypeScript的support for higher order type inference from generic functions来实现.一般来说，语言不能在类型系统中以这种方式抽象泛型；按照microsoft/TypeScript#1213中的要求，这需要higher kinded types个字符，而TypeScript没有.相反，它使用一些具有特定限制和要求的启发式方法，试图以这种方式推断通用函数类型.

有一种方法可以让它工作:

function wrap<A extends any[], R>(test: (...args: A) => R) {
  return (...args: A): R => {
    return test(...args);
  };
}
    
const wrappedGenericTest = wrap(genericTest);
// const wrappedGenericTest: <T>(a: T) => T ????

const wrappedGenericTest2 = wrap(genericTest2);
// const wrappedGenericTest2: <T, U>(a: T, b: U) => T | U ????

我用一对与参数和返回类型对应的类型参数A和R替换了F类型参数.这里使用F的问题是，the Parameters<T> utility type和the ReturnType<T> utility type是conditional types，编译器无法跟踪泛型.

通过直接使用A和R，编译器可以跟踪高阶类型操作，并获得所需的通用函数类型.

Playground link to code