快捷導(dǎo)航

golang中select語句的簡單實例

更新時間：2022年06月29日 11:56:39 作者：燕雙鷹...

Go的select語句是一種僅能用于channl發(fā)送和接收消息的專用語句,此語句運行期間是阻塞的,下面這篇文章主要給大家介紹了關(guān)于golang中select語句的相關(guān)資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下

前言

在golang語言中，select語句就是用來監(jiān)聽和channel有關(guān)的IO操作，當IO操作發(fā)生時，觸發(fā)相應(yīng)的case動作。有了 select語句，可以實現(xiàn) main主線程與 goroutine線程之間的互動。

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

注意事項：

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

1.先舉個簡單例子

先創(chuàng)建兩個信道，并在 select 前往 c2 發(fā)送數(shù)據(jù)

package main
 
import (
	"fmt"
)
 
//go的通道選擇器 讓你可以同時等待多個通道操作。go協(xié)程和通道以及選擇器的結(jié)合是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` 關(guān)鍵字來同時等待這兩個值，并打印各自接收到的值。
	//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 表達式，只要有一個信道有接收到數(shù)據(jù)，那么 select 就結(jié)束，所以輸出如下

2. 避免造成死鎖

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

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

package main
 
import (
	"fmt"
)
 
//go的通道選擇器 讓你可以同時等待多個通道操作。go協(xié)程和通道以及選擇器的結(jié)合是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` 關(guān)鍵字來同時等待這兩個值，并打印各自接收到的值。
	//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 的錯誤，就像下面這樣子。