当前位置： 技术文章>> 100道Go语言面试题之-Go语言中的select语句是如何工作的？请给出一个使用select语句的示例。

文章标题：100道Go语言面试题之-Go语言中的select语句是如何工作的？请给出一个使用select语句的示例。
  • 文章分类: 后端
  • 6420 阅读

Go语言中的select语句是一种特殊的语句，用于同时等待多个通信操作。当多个goroutine需要同时等待多个通信操作（比如多个channel的读或写操作）时，select会阻塞，直到某个通信操作可以进行。这使得select成为处理多个channel的并发操作时的强大工具。

select的工作机制：

  1. 等待多个通信操作：select会阻塞，直到其中一个case可以进行（即channel可读或可写）。
  2. 随机选择：如果有多个case都准备好了，select会随机选择一个执行。
  3. default case：select中可以包含一个default分支，这个分支在没有任何其他case准备好时就会执行。这使得select可以非阻塞地运行。

示例代码：

下面是一个使用select语句的示例，这个例子中，我们创建了两个channel，分别用于接收不同的消息。然后，使用select来同时等待这两个channel上的消息。

package main

import (
    "fmt"
    "time"
)

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

    // 启动两个goroutine发送数据到channel
    go func() {
        time.Sleep(1 * time.Second) // 模拟耗时操作
        ch1 <- "from ch1"
    }()

    go func() {
        time.Sleep(2 * time.Second) // 模拟耗时比ch1长的操作
        ch2 <- "from ch2"
    }()

    // 使用select等待两个channel上的消息
    for i := 0; i < 2; i++ {
        select {
        case msg1 := <-ch1:
            fmt.Println("Received", msg1)
        case msg2 := <-ch2:
            fmt.Println("Received", msg2)
        }
    }

    // 注意：由于两个goroutine都已经启动，并且它们会向各自的channel发送消息，
    // select会依次打印出这两个消息，但顺序取决于哪个goroutine先完成。
}

在这个例子中，我们启动了两个goroutine，每个都向不同的channel发送一条消息。主goroutine中使用select语句等待这些消息。由于ch1的goroutine先完成，所以"from ch1"会被首先打印出来，随后是"from ch2"。然而，因为select的随机性（虽然在这个例子中由于时间差的存在，结果看起来是确定的），如果你多次运行这个程序，理论上可能会看到不同的输出顺序（尽管在这个特定例子中几乎不可能，因为时间差是固定的）。

注意：在上面的例子中，select循环了两次以接收两个channel中的消息。在实际应用中，你可能需要根据你的具体需求来设计select的使用方式。

推荐文章