TypeScript 4.4元组联合类型推断失效问题及解决方案咨询

问题复现 #

下述代码在 TypeScript 4.3.5 版本中可正常完成类型推断，升级到 4.4.2 版本后出现推断异常：

// 4.3.5版本运行正常，4.4.2版本推断异常
type ArgsTupleUnion<T> =
  | [obj: T, x: (arg: T) => void]
  | [y: (arg: number) => void];

function f<T>(...args: ArgsTupleUnion<T>) {}

f({ k: 15 }, (arg) => { // 4.3.5中arg类型为 {k: number}，4.4.2中arg为any
  console.log(arg.k);
});

f((a: number) => {});

type ArgsTuple<T> =
  | [obj: T, x: (arg: T) => void]
  // 移除第二个联合分支后推断恢复正常

function f2<T>(...args: ArgsTuple<T>) {}

f2({ k: 15 }, (arg) => { // 4.3.5和4.4.2版本中，arg都可正确推断为 {k: number}
  console.log(arg.k);
});

异常表现 #

调用f函数传入对象和回调时，4.3.5版本可正确推断回调参数arg的类型为{ k: number }，4.4.2版本中arg被推断为any；仅当移除元组的第二个联合分支后，推断恢复正常。

根因说明 #

该问题是 TypeScript 4.4 版本对联合类型元组的参数推断逻辑调整引入的回归问题，此前已有多位开发者反馈同类场景的类型推断失效问题。

变通方案 #

方案1：使用函数重载替代联合元组参数（最推荐） #

将原本写在参数类型中的联合元组拆分为多个独立的函数重载声明，可完全保留预期的类型推断效果，兼容所有 TS 版本：

// 重载声明：匹配传入对象+回调的场景
function f<T>(obj: T, callback: (arg: T) => void): void;
// 重载声明：匹配仅传入数字回调的场景
function f(callback: (arg: number) => void): void;
// 实现签名兼容所有重载逻辑
function f(...args: unknown[]) {
  // 内部实现逻辑和原函数保持一致即可
}

f({ k: 15 }, (arg) => {
  // arg 可正确推断为 { k: number }
  console.log(arg.k);
});

f((a: number) => {
  // a 可正确推断为 number
});

方案2：辅助泛型推断 #

如果不想调整函数结构，也可以通过给泛型添加默认值，或者调用时显式指定泛型的方式修复推断：

// 给泛型添加默认值辅助推断
type ArgsTupleUnion<T = unknown> =
  | [obj: T, x: (arg: T) => void]
  | [y: (arg: number) => void];

function f<T>(...args: ArgsTupleUnion<T>) {}

// 或者调用时显式指定泛型类型
f<{k: number}>({ k: 15 }, (arg) => {
  console.log(arg.k);
});

内容的提问来源于stack exchange，提问作者JH-