golang中select語句的簡單範例

2022-06-29 18:02:46

前言

在golang語言中，select語句就是用來監聽和channel有關的IO操作，當IO操作發生時，觸發相應的case動作。有了 select語句，可以實現 main主執行緒與 goroutine執行緒之間的互動。

select {
    case <-ch1 :     // 檢測有沒有資料可讀
        // 一旦成功讀取到資料，則進行該case處理語句
    case ch2 <- 1 :  // 檢測有沒有資料可寫
        // 一旦成功向ch2寫入資料，則進行該case處理語句
    default:
        // 如果以上都沒有符合條件，那麼進入default處理流程
}

注意事項：

select語句只能用於channel通道的IO操作，每個case都必須是一個通道。
如果不設定 default條件，當沒有IO操作發生時，select語句就會一直阻塞；
如果有一個或多個IO操作發生時，Go執行時會隨機選擇一個case執行，但此時將無法保證執行順序；
對於case語句，如果存在通道值為nil的讀寫操作，則該分支將被忽略，可以理解為相當於從select語句中刪除了這個case；
對於空的 select語句，會引起死鎖；
對於在 for中的select語句，不能新增 default，否則會引起cpu佔用過高的問題；

1.先舉個簡單例子

先建立兩個通道，並在 select 前往 c2 傳送資料

package main
 
import (
	"fmt"
)
 
//go的通道選擇器 讓你可以同時等待多個通道操作。go協程和通道以及選擇器的結合是go的一個強大特性。
 
func main() {
	// 在我們的例子中，我們將從兩個通道中選擇。
	c1 := make(chan string, 1)
	c2 := make(chan string, 1)
 
	c2 <- "nihao"
 
	//go func() {
	//	time.Sleep(time.Second * 1)
	//	c1 <- "one"
	//}()
	//
	//go func() {
	//	time.Sleep(time.Second * 2)
	//	c2 <- "two"
	//}()
 
	//我們使用 `select` 關鍵字來同時等待這兩個值，並列印各自接收到的值。
	//for i := 0; i < 2; i++ {
	select {
	case msg1 := <-c1:
		fmt.Println("received", msg1)
	case msg2 := <-c2:
		fmt.Println("received", msg2)
	default:
		fmt.Println("No data received")
	}
	//}
 
}

在執行 select 時，會遍歷所有（如果有機會的話）的 case 表示式，只要有一個通道有接收到資料，那麼 select 就結束，所以輸出如下

2. 避免造成死鎖

select 在執行過程中，必須命中其中的某一分支。

如果在遍歷完所有的 case 後，若沒有命中（命中：也許這樣描述不太準確，我本意是想說可以執行通道的操作語句）任何一個 case 表示式，就會進入 default 裡的程式碼分支。

package main
 
import (
	"fmt"
)
 
//go的通道選擇器 讓你可以同時等待多個通道操作。go協程和通道以及選擇器的結合是go的一個強大特性。
 
func main() {
	// 在我們的例子中，我們將從兩個通道中選擇。
	c1 := make(chan string, 1)
	c2 := make(chan string, 1)
 
	//c2 <- "nihao"
 
	//go func() {
	//	time.Sleep(time.Second * 1)
	//	c1 <- "one"
	//}()
	//
	//go func() {
	//	time.Sleep(time.Second * 2)
	//	c2 <- "two"
	//}()
 
	//我們使用 `select` 關鍵字來同時等待這兩個值，並列印各自接收到的值。
	//for i := 0; i < 2; i++ {
	select {
	case msg1 := <-c1:
		fmt.Println("received", msg1)
	case msg2 := <-c2:
		fmt.Println("received", msg2)
		//default:
		//	fmt.Println("No data received")
		//}
	}
}

但如果你沒有寫 default 分支，select 就會阻塞，直到有某個 case 可以命中，而如果一直沒有命中，select 就會丟擲 deadlock 的錯誤，就像下面這樣子。