golang開發:channel使用

飛翔碼農發表於2020-09-27
Golang

channel主要是用於多個goroutine之間通訊

channel語法

channel是引用型別,需要實用make來建立channel,如下
make(chan Type, [buffer])
chan Type 通道的型別
buffer 是可選引數，代表通道緩衝區的大小(省略則代表無緩衝)
向channel裡面寫入資料使用 <- 符號

q := make(chan bool)
q<-true

從channel裡面讀取資料也是使用 <- 符號，只不過寫入的channel在右邊，讀取的時候channel在左邊。意思跟方向是一致的，一個是資料進入channel，一個是資料從channel出去

q := make(chan bool)
<-q

有緩衝channel的使用

我們一直使用的無緩衝的channel，今天主要學習下有快取的channel。
無緩衝的channel，寫入資料後一定要有goroutine 從channel讀取資料後再寫入，否則程式會panic。

func main() {
	ch := make(chan int)
	ch<-1
}

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan send]:

有緩衝的channel，可以寫入緩衝大小個資料，可以沒有goroutine從channel讀取。

func main() {
	ch := make(chan int,2)
	ch<-1
	ch<-2
}


Process finished with exit code 0

執行結果跟預期是一致的。

有緩衝大小的channel使用場景

假如我們有一個任務需要10個goroutine去排程執行，只要有一個goroutine執行完畢，排程就應該結束。我們看下虛擬碼

func test(wg *sync.WaitGroup,ch chan int,i int) {
	fmt.Println("test code")
	ch<-i
}

func main() {
	fmt.Println("start",runtime.NumGoroutine())
	ch := make(chan int)
	wg := new(sync.WaitGroup)
	for i:=0;i<10;i++ {
		wg.Add(1)
		go test(wg,ch,i)
	}
	fmt.Println(<-ch)
	fmt.Println("end",runtime.NumGoroutine())
	wg.Done()
}

start 1
test code
9
end 10

從執行結果上看，是第9個goroutine首先執行完畢了。程式也正常退出了。但是我們看到，加上主goroutine，記憶體中一共有11個goroutine，程式退出的時候還有10個goroutine，減去一個主goroutine，還有9個goroutine沒有退出，這個對程式來說是不允許，可能會洩漏或者長期佔用資源不釋放。

但是我們如果使用了有緩衝的channel，就可以利用channel的緩衝機制正常退出全部的goroutine了。
看下程式碼

func test(wg *sync.WaitGroup,ch chan int,i int) {
	ch<-i
}

func main() {
	fmt.Println("start",runtime.NumGoroutine())
	ch := make(chan int, 10)
	wg := new(sync.WaitGroup)
	for i:=0;i<10;i++ {
		wg.Add(1)
		go test(wg,ch,i)
	}
	fmt.Println(<-ch,"success")
	for i:= 0;i<9;i++ {
	 fmt.Println(<-ch)
	}
	fmt.Println("end",runtime.NumGoroutine())
	wg.Done()
}

start 1
0 success
1
2
4
3
5
6
7
8
9
end 1

Process finished with exit code 0

列印success的時候，第一個goroutine已經將任務完成，迴圈9次將channel裡面的資料讀取出來，保證建立的goroutine都不會阻塞能夠正常退出來。

相關文章