TypeScript泛型函数如何实现返回类型从使用处自动推断

你最初的写法报错核心原因是泛型参数位置错误：

• 把泛型U定义在函数签名的泛型位：type Thing<T> = <U>(value: T) => U，这个写法的语义是函数调用方可以任意指定U的类型，要求函数必须能返回任意调用方需要的U类型，这显然不可能实现，编译器会认为返回值类型和U不匹配，抛出'U' could be instantiated with an arbitrary type which could be unrelated to 'string'错误。
• 你实际需要的语义是：函数接收固定类型T的入参，返回值类型由函数实现决定，无需提前声明U，由TypeScript根据实际传入的函数自动推断返回类型。

不要把返回值泛型定义在函数内部泛型位，只需要在类型定义中约束入参类型，返回值类型交给TypeScript的上下文类型推断自动推导即可。

基础场景实现 #

// 定义Thing类型：只约束入参为T类型，返回值类型由实际赋值的函数自动推断
type Thing<T> = (value: T) => unknown

// 赋值时TS会自动推断返回值类型，无需手动声明
const shouldMakeStrings: Thing<string> = (a: string) => a + 'foo' // 自动推断类型为 (a: string) => string
const shouldMakeObjects: Thing<string> = (a: string) => ({ foo: a }) // 自动推断类型为 (a: string) => { foo: string }

如果需要更宽松的类型兼容，也可以把unknown替换为any，不影响自动推断效果。

完整业务场景实现 #

针对回调注册/Modifier场景，配合泛型捕获回调返回类型，可以实现完全的自动类型推断：

type A = { foo: string }
type B = { bar: string }
type C = number[]

type Modifiable = {
  'typeA': A,
  'typeB': B,
  'typeC': C
}

// 定义Modifier类型，仅约束入参类型
type Modifier<T> = (value: T) => unknown
type ModifierMap = {
  [P in keyof Modifiable]?: Modifier<Modifiable[P]>
}

const modifiers: ModifierMap = {}

// 注册函数用泛型同时捕获类型key和回调返回值类型，无需手动声明返回类型
const setModifier = <V extends keyof Modifiable, R>(
  type: V, 
  callback: (value: Modifiable[V]) => R
) => {
  modifiers[type] = callback
}

// 注册回调时TS会自动推断入参和返回值类型
setModifier('typeA', (input: A) => input.foo + 'a') // 返回值自动推断为string
setModifier('typeA', a => a.foo) // 入参a自动通过上下文推断为A类型，返回值为string
setModifier('typeA', (input: A) => ({ ...input, upperFoo: input.foo.toUpperCase() })) // 返回值自动推断为A & { upperFoo: string }

// 使用时自动匹配对应返回类型
const makeModifiedA = (modifiers?: ModifierMap) => {
  const a = { foo: "hello world" }
  return modifiers?.typeA ? modifiers.typeA(a) : a
}

makeModifiedA(modifiers) // 类型自动推断为A | string | { foo: string, upperFoo: string }，与实际运行结果一致

如果需要更精确的返回类型，可以在调用对应处理函数时通过泛型传入当前注册的回调返回类型，绝大多数业务场景下上述实现已经可以满足自动推断需求。

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