TypeScript if判断后泛型返回值类型推断异常问题

问题根因 #

TypeScript 控制流分析不会对泛型类型参数执行收窄操作：即使通过if (p === EModule.T1)把参数p的类型收窄到EModule.T1，泛型参数T仍然维持T extends EModule的原始约束，不会被自动推导为字面量类型EModule.T1。
此时分支内返回值需要匹配IMapT[T]类型，由于T可能是EModule下的任意成员，编译器会要求返回值同时满足所有可能的返回类型约束，也就是T1 & T2，和实际返回的对象结构不匹配，就会抛出类型错误。

修复方案 #

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

通过重载签名为不同入参定义精确的返回类型，绕开泛型收窄限制，类型安全度最高：

enum EModule {
  T1 = 0,
  T2 = 1,
}

interface T1 {
  a: number
}
interface T2 {
  b: string
}

interface IMapT {
  [EModule.T1]: T1
  [EModule.T2]: T2
}

// 对外暴露的重载签名
function check(p: EModule.T1): T1 | null
function check(p: EModule.T2): T2 | null
// 内部实现签名，类型可适当宽松
function check(p: EModule): IMapT[EModule] | null {
  if (p === EModule.T1) {
    return { a: 10 }
  }
  if (p === EModule.T2) {
    return { b: '' }
  }
  return null
}

const x = check(EModule.T1) // 自动推断为 T1 | null，类型完全符合预期

方案2：类型断言（快速修复） #

如果不想额外写重载签名，可以在分支返回时通过类型断言手动指定类型，跳过编译器的泛型匹配检查：

function check<T extends EModule> (p: T): IMapT[T] | null {
  if (p === EModule.T1) {
    return { a: 10 } as IMapT[T]
  }
  if (p === EModule.T2) {
    return { b: '' } as IMapT[T]
  }
  return null
}

注意：类型断言会绕过编译器类型校验，使用时需要自行保证分支逻辑和返回值结构完全匹配，避免后续代码迭代出现类型不匹配的隐患。

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