generics - Typescript 中区分联合类型的通用匹配函数

Question

是否可以在可区分的联合类型上定义类似通用匹配函数的东西？ 假设我们有以下类型定义:

const Kinds = {
  A: 'A',
  B: 'B',
};
type Kind = typeof Kinds.A | typeof Kinds.B;
type Value = A | B;
interface A {
  kind: Kinds.A
}
interface B {
  kind: Kinds.B
}

使用 switch 语句可以定义匹配函数，例如:

interface Matcher<T> {
  Kinds.A: (value: A) => T
  Kinds.B: (value: B) => T
}
function match<T>(matcher: Matcher<T>) {
  return function(value: Value) {
    switch (value.kind) {
      case Kinds.A: return matcher[Kinds.A](value);
      case Kinds.B: return matcher[Kinds.B](value);
    }
  }
}

它完成了这项工作，但是定义这样的函数非常乏味，尤其是当一个函数有很多联合成员时。

是否有可能以某种方式简化这个定义，也许用 Mapped Types或来自最新 2.1 分支的其他现有方法。

我在玩“映射类型”，但我不确定是否真的有可能获得具体的Value，即使我知道Kind，例如某事喜欢:

type Matcher<T> = {[P in Kind]: (value: P) => T};
function match<T>(matcher: Matcher<T>) {
  return function(value: Value) {
    return matcher[value.kind](value);
  }
}

但是实际上可以将 P 转换为相应的 Value 类型。

Answer 1

这样做的关键是能够根据其种类获得工会成员。这可以用这样的类型来完成:

type UnionMemberByKind<K> = Extract<Union, { kind: K }>

Extract<T, U>返回 T 的成员匹配 U : 在这种情况下，它将返回具有指定 kind 的联合的单个成员.

使用该类型，您可以正确构建匹配器对象:

type Matcher<Res> = {
    [P in Union["kind"]]: (value: UnionMemberByKind<P>) => Res
}

然后基本上像以前一样定义你的匹配函数:

function match<T>(matcher: Matcher<T>) {
    return function(value: Union) {
        return matcher[value.kind](value as any);
    }
}

(as any 转换很不幸，但我找不到避免它的方法)