Go语言select都能做什么?

2023-07-31 17:06:51 浏览数 (1)

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

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

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

基本语法

select语句的基本语法如下:

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

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

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

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

使用规则

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

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

多路复用

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

代码语言:txt复制
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语句,我们可以在没有数据完整性时默认执行的逻辑,避免程序陷入无限等待状态。

代码语言:txt复制
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.

超时处理

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

代码语言:txt复制
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.")
    }
}

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

输出:

Timeout occurred.

关于Masutaa

Masutaa是个互联网从业者自由协作交流平台,链接行业内TOP10%人才!目前平台上已经有将近400名互联网尖端人才,其中近70%的从业者从业年限超3年。加入Masutaa,加入自由生活!感兴趣的朋友可以搜索masutaa官网!

0 人点赞