TypeScript中如何根据对象属性设置情况定义函数条件返回类型

现有实现的局限性 #

当前写法T extends Options ? string : number属于粗粒度类型判断，仅校验传入参数能否赋值给完整的Options接口，不会解析参数内部的具体属性取值，无法实现细分场景的返回类型匹配。

可行实现方案 #

方案1：泛型捕获精确入参结构 + 条件返回类型 #

通过泛型自动捕获用户传入 options 的字面量结构，搭配结构化条件类型匹配对应返回值，代码示例：

// 定义不同场景对应的返回类型规则
type AbcReturn<T> =
  // 优先匹配传入 flagA 为 true 的场景
  T extends { flagA: true } ? TypeA :
  // 其次匹配传入 flagB 为 true 的场景
  T extends { flagB: true } ? TypeB :
  // 其余场景的默认返回类型
  DefaultType;

// 重写函数定义
function abc<T extends Options | undefined>(name: string, options?: T): AbcReturn<T> {
  // 函数内部逻辑保持原有判断不变
  if (options?.flagA) return {} as TypeA;
  if (options?.flagB) return {} as TypeB;
  return {} as DefaultType;
}

实际调用时的类型推导效果：

• 传入{ flagA: true }，返回类型自动推导为TypeA
• 传入{ flagB: true }，返回类型自动推导为TypeB
• 不传参数、传入空对象{}，返回类型自动推导为DefaultType
• 同时传入{ flagA: true, flagB: true }时，按照条件分支的优先级匹配到TypeA，调整条件书写顺序即可调整优先级。

方案2：函数重载分支匹配 #

如果希望IDE在调用时给出更清晰的入参提示，可以使用函数重载的方式定义不同分支，类型匹配精度和前者一致，代码示例：

// 依次定义不同入参对应的重载签名
function abc(name: string, options: { flagA: true }): TypeA;
function abc(name: string, options: { flagB: true }): TypeB;
function abc(name: string, options?: Options): DefaultType;

// 通用实现逻辑（不会对调用方暴露内部类型）
function abc(name: string, options?: Options) {
  if (options?.flagA) return {} as TypeA;
  if (options?.flagB) return {} as TypeB;
  return {} as DefaultType;
}

两种方案的选型差异：

• 泛型+条件类型的写法适合入参分支较多的场景，代码冗余度低
• 函数重载的写法IDE提示更友好，会逐一展示不同入参对应的返回类型，不会出现复杂分布式条件类型难以理解的问题。

常见踩坑提示 #

• 条件匹配不要仅判断属性是否存在，建议对齐函数内部的判断逻辑，明确匹配属性取值为true，避免误匹配flagA: false这类不符合判断逻辑的入参
• 如果需要兼容flag为false的场景，可以在条件判断中引入 infer 提取属性的具体取值，将取值结果纳入返回类型规则
• 函数内部实现的类型不需要做过度精细化的标注，TS 只会对外部调用方暴露你定义的签名规则，只要逻辑和签名保持一致即可。

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