TypeScript中包装重载函数：如何保留精准返回类型推断

你的问题核心在于包装函数没有保留原函数的泛型关联逻辑，导致TypeScript无法追踪具体的枚举成员类型，只能返回联合类型。

解决方案：给包装函数添加泛型约束 #

不需要重复声明类型，只需要让包装函数也使用泛型参数，继承原函数的枚举约束即可：

enum VariantEnum { a, b }

type ReturnA = string
type ReturnB = number

type VariantReturnMap<V> = 
  V extends VariantEnum.a ? ReturnA :
  V extends VariantEnum.b ? ReturnB : 
  undefined

declare function sampleFunction<V extends VariantEnum>(variant: V): VariantReturnMap<V>

// 修改后的包装函数，引入泛型参数V
function wrapperFunction<V extends VariantEnum>(variant: V) {
  // 其他逻辑
  return sampleFunction(variant) // 返回类型会自动匹配V对应的Return类型
}

// 测试验证
const resA = wrapperFunction(VariantEnum.a) // 类型为ReturnA（string）
const resB = wrapperFunction(VariantEnum.b) // 类型为ReturnB（number）

原理说明 #

• 原函数sampleFunction通过泛型参数V和条件类型VariantReturnMap<V>，建立了具体枚举成员和返回类型的一一对应关系。
• 你原来的wrapperFunction参数直接用VariantEnum（本质是VariantEnum.a | VariantEnum.b联合类型），调用sampleFunction时，TypeScript会对联合类型进行分布式解析，最终返回ReturnA | ReturnB。
• 修改后的包装函数通过泛型参数V，让TypeScript能够追踪传入的单个枚举成员类型，而非联合类型，从而触发条件类型的对应分支，得到精准的返回类型。

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