如何缩小联合类型约束下的泛型参数类型？

问题原因 #

你当前的泛型写法中，T是string | SomeClass的子类型，编译器在分支判断时无法将泛型参数T精确窄化为string或SomeClass——它只能知道x是T & string或T & SomeClass，但无法确定T本身就是单一的string或SomeClass，因此返回的string[]或SomeClass[]无法匹配T[]的类型约束。

解决方案 #

方案1：使用函数重载（推荐） #

函数重载能直接明确每种输入类型对应的输出类型，完全规避泛型窄化的问题：

// 定义重载签名，明确输入输出的对应关系
function f(x: string): string[];
function f(x: SomeClass): SomeClass[];

// 实现签名，处理具体逻辑
function f(x: string | SomeClass): string[] | SomeClass[] {
  if (typeof x === "string") {
    const ret = fImpl(parseSomeClass(x));
    return ret.map(r => r.toString());
  } else {
    return fImpl(x);
  }
}

调用该函数时，TypeScript会根据传入的参数类型自动匹配对应的重载签名，返回正确的类型。

方案2：泛型结合条件类型与类型断言 #

如果坚持使用泛型，可以通过条件类型定义返回值，并配合类型断言让编译器通过检查：

function f<T extends string | SomeClass>(x: T): T extends string ? string[] : SomeClass[] {
  if (typeof x === "string") {
    const ret = fImpl(parseSomeClass(x));
    // 用类型断言告诉编译器当前分支的返回类型匹配条件类型
    return ret.map(r => r.toString()) as ReturnType<typeof f<T>>;
  } else {
    return fImpl(x) as ReturnType<typeof f<T>>;
  }
}

这种方式保留了泛型的写法，但需要手动添加断言，可读性略低于重载方案。

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