打字稿：如何根据类型进行分支

Question

我有这样的东西：

interface A {
  a1: string;
  a2: number;
  a3: boolean;
}

interface B {
  b1: number;
  b2: boolean;
  b3: string;
}

function foo<K1 extends keyof A, K2 extends keyof B>(input: K1 | K2) {
  if (input keyof A ) { // <--- THIS IS WRONG!
    console.log('got A type');
  } else {
    console.log('got B type');
  }
}

foo('a1');
foo('b2');

如何更新 if 语句以使其根据类型正确分支？

我试过 keyof、typeof、instanceof .... none 其中是正确的。

Answer 1

接口在运行时不存在，它们完全是编译时构造。所以没有办法在表达式中使用类型，因为当运行代码时类型不会存在。

我们能做的最好的事情是创建一个包含接口所有键的对象，编译器保证包含接口的所有键并且只包含接口的键

然后我们可以在自定义类型保护中使用这个对象来帮助编译器缩小键的类型。

一般的解决方案如下所示：

interface A {
    a1: string;
    a2: number;
    a3?: boolean;
}

interface B {
    b1: number;
    b2: boolean;
    b3: string;
}

// Factory function for key type-guards
function interfaceKeys<T>(keys: Record<keyof T, 0>) {
    return function (o: PropertyKey): o is keyof T {
        return o in keys;
    }
}
// The objects here are compiler enforced to have all the keys and nothing but the keys of each interface
const isAkey = interfaceKeys<A>({ a1: 0, a2: 0, a3: 0 })
const isBkey = interfaceKeys<B>({ b1: 0, b2: 0, b3: 0 })


function foo<K1 extends keyof A, K2 extends keyof B>(input: K1 | K2) {
    if (isAkey(input)) { // custom type guard usage
        console.log('got A type');
        input // is K1
    } else {
        console.log('got B type');
        input // is K2
    }
}

foo('a1');
foo('b2');

打字稿：如何根据类型进行分支

Typescript: How to branch based on type

branch

types

if-statement

typescript