跳到主要内容

Go select 学习

· 阅读需 3 分钟
ahKevinXy
ahKevinXy
作者

预览图 在 Go 语言内 select 是一个关键字,用于监听 和channel 有关的IO 操作

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

基本语法​

select 语句基本语法:

select {
case <-channel1:
	
	// do channel1 逻辑
case data:= <-channel2:
	// 通过channel 就绪时候处理
default:
	// 没有任何通道就绪时进行处理
}

语法类型 switch 相识

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

使用规则​

  1. select 语句只能使用过程中,用于多个通道之间进行选择,以监听通道的就绪昨天,而不是用于其它类型的条件判断
  2. select 语句可以保护多个case语句,每个case子句对应一个通道操作,当其中任意一个通道就绪时,相应的case 子句将会执行
  3. 如果多个通路都就绪,select 语句会随机一个通道执行,这样保证多个通道公平竞争
  4. select 语句的执行可能是阻塞,也可以是非阻塞.如果没有任何一个通道就绪没有默认的 default 子句,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 分支。

超时处理​

通过结合 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 分支。

标签：