Typescript 在Switch陈述中输入保护类

发布于04月28日

我已经很多年没有使用Typescript了，不记得也不知道如何正确地在Switch声明中输入保护许多类.

class A {}
class B {}
class C {}

type OneOfThem = A | B | C;

function test(foo: OneOfThem): string {
    switch(/* something using foo */) {
        /* A */:
            return "A";
        /* B */:
            return "B";
        /* C */:
            return "C";

        /* should not need to use "default" as all cases are handled */
    }
}

我发现并try 了几个选项，例如:

Using foo.constructor
Using instanceof inside the case statements
将额外成员添加到要在Switch陈述中使用的三个类中

但它们都不起作用(Function lacks ending return statement and return type does not include 'undefined').

我的记忆是否起了作用，而这在课堂上是不可能的？

推荐答案

将额外成员添加到要在Switch陈述中使用的三个类中

对于您的情况，这确实可能是最简单的解决方案:它使用Discriminated Union，通过向每个类添加discriminant个成员，具有literal种类型(即不仅仅是string或number)，以便您(和TypScript)可以区分foo来自哪个类:

class A {
    readonly kind = "a"
    //       ^? "a" string literal, not just `string`
}
class B {
    kind: "b" = "b"
    // ^? "b" string literal, as explicitly typed
}
class C {
    kind = "c" as const
    // ^? "c" string literal, thanks to the const assertion
}

然后简单地根据这kind个区分属性进行区分:

function test(foo: OneOfThem): string { // Okay
    switch (foo.kind) {
        case "a":
            foo
            //^? A inferred by TS, based on `foo.kind === "a"` type guard
            return "A";
        case "b":
            foo
            //^? B
            return "B";
        case "c":
            foo
            //^? C
            return "C";
        /* should not need to use "default" as all cases are handled */
    }
}

Playground Link