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

时间:2019-02-05 22:10:24

标签: typescript if-statement types branch

我有这样的东西:

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 ...。没有一个是正确的。

1 个答案:

答案 0 :(得分: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');