如何为返回参数传入函数的conditional函数保留类型签名

这个问题在TypeScript里很常见——当你直接用宽泛的Function类型标注时，会彻底丢失原函数的具体参数和返回值类型信息。要解决这个问题，我们需要用泛型来精准捕获传入函数的类型签名。

问题根源 #

如果给conditional函数加了这样的类型注解：

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

TypeScript只会把返回值识别成泛泛的Function类型，完全追踪不到Math.sqrt或者add函数的具体参数、返回值类型，自然没法提供类型提示和检查了。

解决方案：用泛型锁定类型信息 #

我们可以用泛型T来捕获传入函数的精确类型，让TypeScript完整记住原函数的参数和返回值结构：

// 泛型T约束为任意函数类型
function conditional<T extends (...args: any[]) => any>(x: boolean, f: T): T | ((...args: any[]) => void) {
  if (x) {
    return f;
  } else {
    // 返回可接受任意参数的空函数，和原JS行为完全一致
    return (...args: any[]) => {};
  }
}

调用示例与类型验证 #

现在调用conditional时，原函数的类型签名会被完整保留：

// mySqrt的类型被推断为(x: number) => number，和Math.sqrt完全一致
let mySqrt = conditional(true, Math.sqrt);
console.log(mySqrt(4)); // 类型检查通过，返回2
// mySqrt("4"); // 类型报错，因为Math.sqrt只接受number类型参数

// add函数的类型是(x: number, y: number) => number
let add = (x: number, y: number) => x + y;
// myAdd的类型和add完全匹配
let myAdd = conditional(true, add);
console.log(myAdd(1, 2)); // 类型检查通过，返回3

// emptyAdd的类型是(...args: any[]) => void
let emptyAdd = conditional(false, add);
emptyAdd(1, 2); // 不会报错，符合原JS中空函数忽略参数的行为

可选优化：严格限制空函数参数 #

如果你希望空函数只能无参调用（更严格的类型检查），可以把返回的空函数类型改成() => void：

function conditional<T extends (...args: any[]) => any>(x: boolean, f: T): T | (() => void) {
  if (x) return f;
  else return () => {};
}

这时调用emptyAdd(1,2)就会触发类型错误，适合需要严格参数校验的场景。

内容的提问来源于stack exchange，提问作者Mohideen Imran Khan