如何让返回值基于可辨识联合参数实现条件化？

问题原因 #

你遇到的错误本质是：TypeScript编译器无法在泛型函数内部，将switch分支里收窄后的具体返回值，和泛型条件类型Output<T>建立关联。因为泛型T是Input联合类型的任意子集，编译器没办法确认每个分支的返回值恰好对应当前T对应的Output<T>类型。

修复方案 #

方案1：使用函数重载（推荐，类型检查更严格） #

直接为每个Input变体定义输入输出的对应关系，编译器能精准匹配每个分支的类型：

type Input = { kind: "a" } | { kind: "b" } | { kind: "c" }
type OutputA = { a: string }
type OutputB = { b: string }
type OutputC = { c: string }

// 函数重载签名：定义每个输入对应的输出类型
function run(input: { kind: "a" }): OutputA;
function run(input: { kind: "b" }): OutputB;
function run(input: { kind: "c" }): OutputC;

// 实现签名：处理所有输入情况
function run(input: Input): OutputA | OutputB | OutputC {
  switch (input.kind) {
    case "a":
      return { a: "a" };
    case "b":
      return { b: "b" };
    case "c":
      return { c: "c" };
  }
}

调用时，TypeScript会根据传入的input自动推断出正确的返回类型，无需额外断言。

方案2：保留条件类型，配合类型断言 #

如果需要复用Output的类型映射关系，可以用映射类型重构Output，再通过类型断言告诉编译器分支返回值的正确性：

type Input = { kind: "a" } | { kind: "b" } | { kind: "c" }

// 用映射类型定义输入kind到输出类型的映射
type OutputMap = {
  a: { a: string },
  b: { b: string },
  c: { c: string }
}
// 基于Input的kind字段推导Output类型
type Output<T extends Input> = OutputMap[T["kind"]]

export function run<T extends Input>(input: T): Output<T> {
  switch (input.kind) {
    case "a":
      return { a: "a" } as Output<T>;
    case "b":
      return { b: "b" } as Output<T>;
    case "c":
      return { c: "c" } as Output<T>;
  }
}

这里的类型断言是安全的，因为我们已经通过switch精准收窄了input.kind，确保返回值和当前T对应的Output<T>一致。

关于条件类型和泛型的适用性 #

• 当需要复用输入输出的类型映射关系（比如多个函数都依赖Input和Output的对应规则），条件类型（或映射类型）+泛型是合适的选择，能保证类型定义的一致性。
• 如果只是单个函数的类型定义，函数重载更直观，不需要额外的类型推导逻辑，编译器的类型检查也更严格，无需手动断言。

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