前言

在 Go 中，可以通過(guò)關(guān)鍵字 select 來(lái)完成從不同的并發(fā)執(zhí)行的協(xié)程中獲取值，它和 switch 控制語(yǔ)句非常相似，也被稱作通信開(kāi)關(guān)；它的行為像是“你準(zhǔn)備好了嗎”的輪詢機(jī)制；

select 監(jiān)聽(tīng)進(jìn)入通道的數(shù)據(jù)，也可以是用通道發(fā)送值的時(shí)候。

select 是 Go 在語(yǔ)言層面提供的多路 I/O 復(fù)用機(jī)制，用于檢測(cè)多個(gè)管道是否就緒（即可讀或可寫(xiě)），其特性與管道息息相關(guān)。

語(yǔ)法格式：

select {
case u:= <- ch1:
        ...
case v:= <- ch2:
        ...
        ...
default: // no value ready to be received
        ...
}

default 語(yǔ)句是可選的；fallthrough 行為，和普通的 switch 相似，是不允許的。在任何一個(gè) case 中執(zhí)行 break 或者 return，select 就結(jié)束了。

select 做的就是：選擇處理列出的多個(gè)通信情況中的一個(gè)。

• 如果都阻塞了，會(huì)等待直到其中一個(gè)可以處理
• 如果多個(gè)可以處理，隨機(jī)選擇一個(gè)
• 如果沒(méi)有通道操作可以處理并且寫(xiě)了default 語(yǔ)句，它就會(huì)執(zhí)行：default 永遠(yuǎn)是可運(yùn)行的（這就是準(zhǔn)備好了，可以執(zhí)行）。

在 select 中使用發(fā)送操作并且有 default 可以確保發(fā)送不被阻塞！如果沒(méi)有 default，select 就會(huì)一直阻塞。default 不能處理管道讀寫(xiě)操作，

select 語(yǔ)句實(shí)現(xiàn)了一種監(jiān)聽(tīng)模式，通常用在（無(wú)限）循環(huán)中；在某種情況下，通過(guò) break 語(yǔ)句使循環(huán)退出。

程序示例

package main
import (
    "fmt"
    "time"
)
func main() {
    ch1 := make(chan int)
    ch2 := make(chan int)
    go pump1(ch1)
    go pump2(ch2)
    go suck(ch1, ch2)
    time.Sleep(1e9)
}
func pump1(ch chan int) {
    for i := 0; ; i++ {
        ch <- i * 2
    }
}
func pump2(ch chan int) {
    for i := 0; ; i++ {
        ch <- i + 5
    }
}
func suck(ch1, ch2 chan int) {
    for {
        select {
        case v := <-ch1:
            fmt.Printf("Received on channel 1: %d\n", v)
        case v := <-ch2:
            fmt.Printf("Received on channel 2: %d\n", v)
        }
    }
}

在程序 goroutine_select.go 中有 2 個(gè)通道 ch1 和 ch2，

三個(gè)協(xié)程 pump1()、pump2() 和 suck()。

這是一個(gè)典型的生產(chǎn)者消費(fèi)者模式。在無(wú)限循環(huán)中，ch1 和 ch2 通過(guò) pump1() 和 pump2() 填充整數(shù)；suck() 也是在無(wú)限循環(huán)中輪詢輸入的，通過(guò) select 語(yǔ)句獲取 ch1 和 ch2 的整數(shù)并輸出。選擇哪一個(gè) case 取決于哪一個(gè)通道收到了信息。程序在 main 執(zhí)行 1 秒后結(jié)束。

運(yùn)行結(jié)果：

Received on channel 2: 148120
Received on channel 2: 148121
Received on channel 2: 148122
Received on channel 2: 148123
Received on channel 2: 148124
Received on channel 2: 148125
Received on channel 2: 148126
Received on channel 1: 296784
Received on channel 2: 148127
Received on channel 2: 148128
Received on channel 2: 148129
Received on channel 1: 296786
Received on channel 1: 296788

一秒內(nèi)的輸出非常驚人，如果我們給它計(jì)數(shù)（goroutine_select2.go），得到了 296788 個(gè)左右的數(shù)字。

select 特性預(yù)覽

管道讀寫(xiě)

select 只能作用于管道，包括數(shù)據(jù)的讀取和寫(xiě)入。例如：

package main
import "fmt"
func selectDemo(c chan string) {
	recv := ""
	send := "Hello"
	select {
	case recv = <-c:
		fmt.Printf("Received %s\n", recv)
	case c <- send:
		fmt.Printf("Sent %s\n", send)
	}
}

• 如果管道中沒(méi)有緩存，如下：

func main() {
	c := make(chan string)
	selectDemo(c)
}

此時(shí)管道既不能讀也不能寫(xiě)，兩個(gè) case 語(yǔ)句都不執(zhí)行，select 陷入阻塞

• 如果管道中有緩沖區(qū)且還可以存放至少一個(gè)數(shù)據(jù)，如下：

func main() {
	c := make(chan string, 1)
	selectDemo(c)
}

此時(shí)，管道可以寫(xiě)入，寫(xiě)操作對(duì)應(yīng)的 case 語(yǔ)句得到執(zhí)行，且執(zhí)行結(jié)束后函數(shù)退出。

• 如果管道有緩沖區(qū)且緩沖區(qū)中已放滿數(shù)據(jù)，如下：

func main() {
	c := make(chan string, 1)
	c <- "你好，向你說(shuō)再見(jiàn)！"
	selectDemo(c)
}

此時(shí)，管道可以讀取，讀操作對(duì)應(yīng)的 case 語(yǔ)句得到執(zhí)行，且執(zhí)行結(jié)束后函數(shù)退出。

• 管道有緩沖區(qū)，緩沖區(qū)中已有部分?jǐn)?shù)據(jù)還可以存入數(shù)據(jù)，如下：

func main() {
	c := make(chan string, 2)
	c <- "你好，向你說(shuō)再見(jiàn)！"
	selectDemo(c)
}

管道的緩沖區(qū)有部分且還可以存入數(shù)據(jù)，此時(shí)管道既可以讀取也可以寫(xiě)入，select 將選取一個(gè) case 語(yǔ)句執(zhí)行，任意一個(gè) case 語(yǔ)句執(zhí)行結(jié)束后函數(shù)就退出。

總結(jié)

select 的每個(gè) case 語(yǔ)句只能操作一個(gè)管道，要么寫(xiě)入數(shù)據(jù)，要么讀取數(shù)據(jù)；

如果管道中沒(méi)有數(shù)據(jù)讀取操作則會(huì)阻塞，如果管道中沒(méi)有空余的緩沖區(qū)則寫(xiě)入操作會(huì)阻塞；

當(dāng) select 的多個(gè) case 語(yǔ)句中的管道均阻塞時(shí)，整個(gè) select 語(yǔ)句也會(huì)陷入阻塞，直到任意一個(gè)管道解除阻塞；

如果多個(gè) case 語(yǔ)句均沒(méi)有阻塞，那么 select 將隨機(jī)挑選一個(gè) case 執(zhí)行。

以上就是Go使用select切換協(xié)程入門詳解的詳細(xì)內(nèi)容，更多關(guān)于Go select 切換協(xié)程的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論