TypeScript联合类型场景下函数重载无法正常工作问题求助

我明白你遇到的问题——明明给联合类型的每个分支都定义了对应的函数重载，但TypeScript似乎没能正确关联输入类型和返回类型。咱们来一步步搞定这个问题。

问题根源 #

你的重载语法本身是正确的，但TypeScript在处理以联合类型作为输入的函数实现时，需要更明确的类型关联逻辑。另外，当你用Z类型的变量调用f时，默认返回类型会是两个重载返回值的联合；如果想要根据输入的具体子类型（A或B）得到精确的返回类型，我们可以用更优雅的方式优化。

方法一：泛型+条件类型强化重载推断 #

给函数添加泛型约束，结合条件类型来精准关联输入和返回类型，同时用as const让返回值的类型更明确：

type A = { type: "a", x: number }
type B = { type: "b", x: number, y: number }
type Z = A | B

// 泛型重载，捕获输入的具体子类型
function f<T extends Z>(z: T): T extends A ? { x: number, y: undefined } : { x: number, y: number }
function f(z: Z) {
  if (z.type === "a") {
    return {x: z.x, y: undefined} as const
  } else {
    return {x: z.x, y: z.y} as const
  }
}

// 测试验证：
const aInput: A = { type: "a", x: 5 }
const resultA = f(aInput) // resultA类型为 { x: number, y: undefined }

const bInput: B = { type: "b", x: 10, y: 20 }
const resultB = f(bInput) // resultB类型为 { x: number, y: number }

const zInput: Z = Math.random() > 0.5 ? aInput : bInput
const resultZ = f(zInput) // resultZ为联合类型，符合预期

这里的条件类型会根据泛型T捕获的具体输入类型，自动推导对应的返回类型，as const则让函数返回的对象字面量类型更精确，帮助TypeScript完成类型匹配。

方法二：用条件类型替代重载（更简洁） #

如果不需要保留重载的写法，直接用泛型+条件类型定义函数，代码更简洁，类型逻辑也更清晰：

type A = { type: "a", x: number }
type B = { type: "b", x: number, y: number }
type Z = A | B

// 提前定义返回类型的条件逻辑
type FReturn<T extends Z> = T extends A 
  ? { x: number, y: undefined } 
  : { x: number, y: number }

function f<T extends Z>(z: T): FReturn<T> {
  if (z.type === "a") {
    return {x: z.x, y: undefined} as FReturn<T>
  } else {
    return {x: z.x, y: z.y} as FReturn<T>
  }
}

这种方式避免了重复编写重载签名，同时让输入和返回的类型关系一目了然。

补充说明：为什么原重载看似"不生效" #

当你用Z类型的变量调用f时，TypeScript只能推断返回类型是两个重载返回值的联合（因为Z本身是联合类型），这其实是符合语言设计逻辑的。但如果你传入的是明确的A或B类型，原重载其实是可以正确工作的——如果你觉得返回类型不够精确，那上面的泛型+条件类型方案就能完美解决这个问题。

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