一手掌握 Go 入门：巧用 select 切换协程

2023-12-30 16:18:23

Go 中的并发编程

Go 语言以其出色的并发编程能力著称，它提供了多种机制来支持并发，其中最核心的概念之一便是协程（goroutine）。协程是 Go 中轻量级的执行单元，它与线程类似，但更加轻巧高效。

在 Go 中，我们可以使用 go 来启动一个协程，如下所示：

go func() {
    // 协程代码
}

这个协程将在当前 goroutine 之外执行，它拥有自己的栈空间和局部变量，互不干扰。我们可以通过 select 关键字来实现多个协程之间的通信和同步。

select 关键字

select 关键字类似于 switch 语句，但它专门用于处理协程通信。它的基本语法如下：

select {
    case <case1>:
        // 代码块 1
    case <case2>:
        // 代码块 2
    ...
    default:
        // 默认代码块
}

在 select 语句中，我们可以使用 case 语句来指定多个协程通信的通道，当某个通道有值可读时，对应的 case 语句就会被执行。如果没有任何通道有值可读，则执行 default 代码块。

例如，以下代码演示了如何使用 select 语句从两个通道中获取值：

package main

import (
    "fmt"
    "time"
)

func main() {
    // 创建两个通道
    ch1 := make(chan int)
    ch2 := make(chan string)

    // 启动两个协程，分别向通道中发送值
    go func() {
        time.Sleep(1 * time.Second)
        ch1 <- 10
    }()
    go func() {
        time.Sleep(2 * time.Second)
        ch2 <- "Hello, World!"
    }()

    // 使用 select 从通道中获取值
    select {
    case v := <-ch1:
        fmt.Println("Received from ch1:", v)
    case s := <-ch2:
        fmt.Println("Received from ch2:", s)
    }
}

运行这段代码，我们会看到以下输出：

Received from ch2: Hello, World!

这意味着从通道 ch2 中获取值需要花费更多时间，因此 select 语句选择了从 ch2 中获取值。

灵活运用 select

select 关键字非常灵活，它可以用于各种复杂的并发编程场景。例如，我们可以使用 select 来实现超时机制，或者实现多个协程之间的同步。

以下代码演示了如何使用 select 来实现超时机制：

package main

import (
    "fmt"
    "time"
)

func main() {
    // 创建一个通道
    ch := make(chan int)

    // 启动一个协程，向通道中发送值
    go func() {
        time.Sleep(1 * time.Second)
        ch <- 10
    }()

    // 使用 select 来实现超时机制
    select {
    case v := <-ch:
        fmt.Println("Received from ch:", v)
    case <-time.After(2 * time.Second):
        fmt.Println("Timeout!")
    }
}