Go语言select都能做什么？

Masutaa大师

发布于 2023-07-31 17:06:51

1740

发布于 2023-07-31 17:06:51

文章被收录于专栏：Masutaa大师

在Go语言中，select是一个关键字，用于监听和channel有关的IO操作。

通过select语句，我们可以同时监听channel，并在其中任意一个channel就绪多个时进行相应的处理。

本文将总结一下select语句的常见用法，以及在使用过程中的注意事项。

基本语法

select语句的基本语法如下：

select {
case <-channel1:
    // 通道 channel1 就绪时的处理逻辑
case data := <-channel2:
    // 通道 channel2 就绪时的处理逻辑
default:
    // 当没有任何通道就绪时的默认处理逻辑
}

看到这个语法，很容易想到switch语句。

虽然select语句和switch语句表面上有些相似，但它们的用途和功能是不同的。

switch用于条件判断，而select用于通道操作。不能在select语句中使用任何类型的条件表达式，只能对通道进行操作。

使用规则

虽然语法简单，但是在使用过程中，还是有一些地方需要注意，我总结了以下四点：

select该语句仅用于通道操作选择，用于在多个通道之间进行，以监听通道的就绪状态，而不用于其他类型的条件判断。
select语句可以包含多个case子句，每个case子句对应一个通道操作。当其中任意一个通道适合时，相应的case子句就会被执行。
如果多个通道都已经就绪，select语句会随机选择一个通道来执行。这样保证了多个通道之间的公平竞争。
select语句的执行可能是阻塞的，也可能是非阻塞的。如果没有任何一个通道就绪且没有默认的default子句，select语句会阻塞，直到有一个通道就绪。如果有default子句，且没有任何通道就绪，则语句会阻塞，直到有一个通道就绪select。会执行default子句，从而避免阻塞。

多路复用

select最常见的用途之一，同时监听通道，并根据它们的多个就绪状态执行不同的操作。

package main

import (
    "fmt"
    "time"
)

func main() {
    c1 := make(chan string)
    c2 := make(chan string)

    go func() {
        time.Sleep(3 * time.Second)
        c1 <- "one"
    }()

    go func() {
        time.Sleep(3 * time.Second)
        c2 <- "two"
    }()

    select {
    case msg := <-c1:
        fmt.Println(msg)
    case msg := <-c2:
        fmt.Println(msg)
    }
}

执行上面的代码，程序会随机打印one或者two，如果通道为空的话，程序就会一直阻塞在那里。

阻碍非通信

当通道中没有数据中断或者没有缓冲空间可写入时，普通的读写器操作将会阻塞。

但通过select语句，我们可以在没有数据完整性时默认执行的逻辑，避免程序陷入无限等待状态。

package main

import (
    "fmt"
)

func main() {
    channel := make(chan int)

    select {
    case data := <-channel:
        fmt.Println("Received:", data)
    default:
        fmt.Println("No data available.")
    }
}

执行上面的代码，程序会执行default分支。

输出：

No data available.

超时处理

通过结合select和time.After函数，我们可以在指定期限等待通道就绪，超过时间后执行相应的逻辑。

package main

import (
    "fmt"
    "time"
)

func main() {
    channel := make(chan int)

    select {
    case data := <-channel:
        fmt.Println("Received:", data)
    case <-time.After(3 * time.Second):
        fmt.Println("Timeout occurred.")
    }
}

执行上面代码，如果channel在3秒内没有数据裁决，select会执行time.After分支。

输出：

Timeout occurred.