TypeScript根据接口成员值条件定义方法返回类型及泛型推断

核心问题 #

假设接口的某个成员（single）的值是静态已知的（本场景中为布尔字面量类型），是否可以基于该成员的值，确定同接口下另一个成员方法的返回类型？
初始待优化的接口定义如下：

interface I {
    single: boolean;
    // 需求：当single === true时返回类型为number，否则为Set<number>
    returnSingle: () => number | Set<number>
}

补充提问1：给出的泛型实现是否正确？ #

提问者给出的实现代码：

type SingleOrSet<C extends boolean> = C extends true ? number : Set<number>;

interface I<T extends boolean> {
    single: T;
    // 当single === true时返回类型为number，否则为Set<number>
    returnSingle: () => SingleOrSet<T>
}

const p: I<true> = {single: true, returnSingle: () => {return 1}};

结论：该实现完全符合类型安全要求。
写法通过泛型参数T绑定single字段的字面量类型，再通过条件类型SingleOrSet完成类型映射，示例中p.returnSingle()的返回值会被TS正确推导为number，不会残留冗余的联合类型判断。

补充提问2：是否可以自动推断泛型参数T，无需手动传入类型参数？ #

可以，两种常用无侵入方案：

• 方案1：泛型工厂函数包装，利用TS对函数入参的泛型自动推断能力

  function createI<T extends boolean>(instance: I<T>): I<T> {
      return instance
  }
  
  // 无需手动传入I<true>，T会根据single的值自动推导为true字面量类型
  const p = createI({
      single: true,
      returnSingle: () => 1
  })
  // p.returnSingle() 自动推导为number类型
  
  const q = createI({
      single: false,
      returnSingle: () => new Set<number>()
  })
  // q.returnSingle() 自动推导为Set<number>类型
  

• 方案2：TS 4.9及以上版本使用satisfies运算符，无需额外包装函数

  const p = {
      single: true,
      returnSingle: () => 1
  } satisfies I<boolean>
  // 保留对象字面量的精确字面量类型，同时校验是否符合I接口的结构要求，p.returnSingle()自动推导为number
  

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