返回作为参数提供的函数时保留类型签名

Question

我有一个名为conditional 的函数，它接收一个布尔值和一个函数，如果布尔值为真则返回该函数，否则返回一个虚拟函数。

function conditional(x, f) {
  if(x)
    return f
  else
    return () => {}
}

所以，我可以执行如下代码：

let mySqrt = conditional(true, Math.sqrt)
let add = (x, y) => x + y
let myAdd = conditional(true, add)

但是，当我向条件函数添加类型时，提供给conditional 函数的函数的类型签名会丢失。这使得编写 mySqrt("a") 这样的代码成为可能，而不会出现 TypeScript 抱怨。

如何向条件函数添加类型以保留结果函数的类型？

Answer 1

你需要使用泛型类型参数来表示函数的类型：

function conditional<T extends (...args: any[]) => any>(x: boolean, f: T): T {
    if (x)
        return f;
    else
        return (() => { }) as any; // we need an assertion here to make the empty function compatible with T
}

// Usage
let mySqrt = conditional(true, Math.sqrt); //mySqrt will be of type (x: number) => number
let add = (x: number, y: number) => x + y;
let myAdd = conditional(true, add); // myAdd will be of type (x: number, y: number) => number

Answer 2

您可以结合使用泛型和方法重载签名。

function conditional<T extends Function>(x: true, f: T): T;
function conditional<T extends Function>(x: false, f: T): () => {};
function conditional<T extends Function>(x: boolean, f: T) {
  if(x)
    return f
  else
    return () => {}
}

let mySqrt = conditional(true, Math.sqrt)
mySqrt('a') // error

当提供重载方法签名时，编译器将隐藏实际的、更灵活的实现。