TypeScript泛型推断异常：为何A未被推断为联合类型？如何修复？

问题背景 #

以下代码中，define函数的泛型参数A被编译器推断为string，而非预期的string | number，导致handler函数返回[1,2]时出现类型不兼容错误：

const define = <A>({ handler }: { handler: (some: boolean) => A[] }) => ({ handler });
const a = define({
  handler: (some: boolean) => {
    if (some) {
      return ["foo", "bar"];
    } else {
      return [1, 2];
    }
  },
});

原因分析 #

TypeScript的泛型推断逻辑中，处理多分支返回值的函数时，默认会优先选取第一个出现的具体类型作为泛型参数的推断结果，而非自动合并所有分支类型为联合类型。这里handler函数第一个返回分支是string[]，编译器直接将A推断为string，忽略了后续返回number[]的分支，导致类型校验失败。

修复方案 #

1. 手动指定泛型参数 #

直接为define显式传入联合类型string | number，强制编译器使用该类型作为泛型参数：

const a = define<string | number>({
  handler: (some: boolean) => {
    if (some) {
      return ["foo", "bar"];
    } else {
      return [1, 2];
    }
  },
});

2. 调整泛型定义目标 #

将泛型参数直接对应到返回的数组类型而非数组元素类型，让编译器自动推断数组的联合类型：

const define = <T extends unknown[]>({ handler }: { handler: (some: boolean) => T }) => ({ handler });
const a = define({
  handler: (some: boolean) => {
    if (some) {
      return ["foo", "bar"];
    } else {
      return [1, 2];
    }
  },
});

3. 使用as const断言保留精确类型（可选） #

如果需要更精确的字面量类型，可以在返回数组时添加as const断言，注意这会将数组转为只读元组：

const a = define({
  handler: (some: boolean) => {
    if (some) {
      return ["foo", "bar"] as const;
    } else {
      return [1, 2] as const;
    }
  },
});

内容的提问来源于stack exchange，提问作者florian norbert bepunkt