Go中的switch语句：没有你想象中那么简单 – 使用它的6种方式

发表于： 2023年12月2日 2023年12月2日
分类： Golang
标签： go, Golang, switch

Go以其简洁而著称，但并不是每个人都熟悉这种语言中switch语句的多样性。首先，如果你对Go的switch语句还不熟悉，它可能与其他语言相比有些不同。

下面是一个简单的示例来展示它是什么样子的：

func main() {
    var i int = 1

    switch i {
        case 1:
        fmt.Println("i is 1")
        case 2:
        fmt.Println("i is 2")
        default:
        fmt.Println("i is not 1 or 2")
    }
}

Go的switch语句有一个很酷的特性，即在找到匹配项后就会停止执行，不需要在每个case的末尾加上break语句。

在Go的switch语句中有两个部分：分号前的部分是初始化器，分号后的部分是要检查的值。

可以选择使用两个部分、其中一个部分或者都不使用：

switch initializer; value {}

switch initializer {}

switch value {}

switch {}

很有趣，是吧？

使用字面布尔值的switch

有时候，可能会使用一个变量的switch语句，但这里有一种不同的方法。

考虑使用一个带有字面布尔值的switch语句。这种方法可以让我们检查多个条件，而不仅仅局限于一个变量的值。

func main() {
    var a int = 1
    var b int = 2

    switch true { // <--- use true literal
        case a == 1 && b == 2:
        fmt.Println("a is 1 and b is 2")
        case a == 3:
        fmt.Println("a is 3"):
        default:
        fmt.Println("a is not 1 or 3")
    }
}

乍一看，switch true可能似乎是多余和无意义的。

它感觉有点像在陈述显而易见的事实，但好消息是Go有一种更简化的处理方式，可以像这样简化它：

switch { // <--- just remove `true`
    case a == 1 && b == 2:
    ...
}

这种简化的方法同样有效。

另外，switch语句也可以与false字面值一起使用，提供了一种确定哪些条件未满足的方法。

Switch短赋值

我们经常忽视switch语句中的初始化器部分。

但它非常有用，与if语句或for循环中的初始化器类似。它允许你声明并赋值一个变量，然后立即使用它。

下面是一个例子来说明这一点：

switch a := 1; a {
    case 1:
    fmt.Println("a is 1")
}

// similar
if a := 1; a == 1 {
    fmt.Println("a is 1")
}

在这些情况下，变量a的作用域仅限于switch语句，意味着不能在外部使用a。

还记得我们可以忽略switch的两个部分吗？

你也可以选择只使用初始化器部分，当你这样做时，值部分被假定为true：

switch a := 1 {
    case a == 1:
    fmt.Println("a is 1")
    case a == 2:
    fmt.Println("a is 2")
}

到目前为止，我们已经看到了四种组织switch语句的方式：只使用初始化器、只使用值、两者都使用或者两者都不使用。但我们的重点主要在于switch本身。

接下来，我们将深入探讨case部分的作用以及如何在代码中充分利用它。

包含多个值的case

你可以在一个case中组合多个值。这种方法可以使你的代码更简洁易读：

switch a := 1; a {
    case 1, 2, 3: // <--
    fmt.Println("a is 1, 2 or 3")
}

很多Go的新手并不知道这个功能。相反，他们可能会写出这样的代码：

switch a := 1; a {
    case 1:
case 2:
case 3:
    fmt.Println("a is 1, 2 or 3")
}

但这种方法并不完全正确，因为switch在Go中的工作方式不同。

在这个例子中，打印语句只与最后一个case（case 3）相关联。所以，如果a是1或2，什么也不会发生，因为这些case后面没有指令，程序会直接跳过它们。

使用fallthrough关键字的case

这个关键字允许执行继续到下一个case而不检查其条件。这与大多数语言处理switch case的方式有些不同。

下面是一个例子来展示fallthrough的工作方式：

switch a := 1; a {
    case 1:
    fmt.Println("a is 1")
    fallthrough
    case 2:
    fmt.Println("Now in case 2")
    default:
    fmt.Println("Neither 1 nor 2")
}

输出会是什么？

在这种情况下，当a为1时，程序首先打印“a is 1”。然后，由于fallthrough关键字的存在，它会立即跳转到下一个case（case 2），而不检查a是否实际上为2。所以，它也会打印出“Now in case 2”。

你仍然可以在case 2中使用fallthrough关键字，程序会继续执行下一个case（default），并打印“Neither 1 nor 2”。

switch a := 1; a {
    case 1:
    fmt.Println("a is 1")
    fallthrough
    case 2:
    fmt.Println("Now in case 2")
    fallthrough
    default:
    fmt.Println("Neither 1 nor 2")
}

// Output:
// a is 1
// Now in case 2
// Neither 1 nor 2

但要记住，在Go中，fallthrough关键字绕过了下一个case的条件检查。因此，在switch语句的最后一个case中不使用它，因为没有后续的case可以过渡到。

默认情况和其细微差别

Go中的switch语句的默认情况类似于if语句中的else部分。

当没有任何其他case匹配时，它将执行默认情况，但是在Go中，默认情况有一些有趣的特点：

尽管在大多数编程语言中，默认情况通常放在末尾，但在Go中，它可以放置在switch语句的任何位置。大多数人为了清晰起见会把它放在末尾，但让我们看看当我们把它放在开头时会发生什么：

switch a := 1; a {
    default:
    fmt.Println("Neither 1 nor 2")
    case 1:
    fmt.Println("a is 1")
    case 2:
    fmt.Println("Now in case 2")
}

在这个例子中，即使默认情况首先出现，它仍然被视为最后的选择，只有在没有其他case匹配时才会执行。

但还有另一层可以探索。

如果我们将默认情况与fallthrough关键字混合使用会怎么样？让我们来看看：

switch a := 3; a {
    default:
    fmt.Println("Neither 1 nor 2")
    fallthrough
    case 1:
    fmt.Println("a is 1")
    case 2:
    fmt.Println("Now in case 2")
}

// Output:
// Neither 1 nor 2
// a is 1

在这种情况下，当a为3时，switch从默认情况开始，打印“Neither 1 nor 2”。然后，由于fallthrough的存在，它会移动到下一个case，打印“a is 1”。

带有类型断言的switch

switch语句不仅可以处理值，还可以处理类型。这在处理接口时特别有用。

类型断言是实现这一功能的特性，它允许检查接口值的类型，并根据该类型运行不同的代码段：

func main() {
    var i interface{} = "hello"

    switch v := i.(type) {
        case int:
        fmt.Println("i is an int and its value is", v)
        case string:
        fmt.Println("i is a string and its value is", v)
        default:
        fmt.Println("Unknown type")
    }
}

在这种情况下，i是一个存储字符串的接口变量。

switch语句使用i.(type)来确定i的类型，然后根据该类型选择要执行的case：