typescript - 限制参数函数的返回类型

Question

我有一个以函数作为参数的函数。参数函数返回几个枚举之一。如何使用参数函数的返回类型推断包装函数的类型。

我认为举个例子更容易解释:

const foo = { a: 1, b: '2' }

function h(k: (() => 'a' | 'b')) {
  return foo[k()]
}

const d = h(() => 'a') 
// expected: d should be of type number
//   actual: d is of type number | string

playground

Answer 1

您可以将返回的键类型定义为通用参数(它将被推断为单个键，而不是所有可能键的并集):

function h<K extends keyof typeof foo>(k: (() => K)) {
  return foo[k()]
}

const d = h(() => 'a') // now number

Playground