Go Channel 面試題解析
摘要：
有一道這樣的面試題目：
寫程式碼實現兩個 goroutine，其中一個產生隨機數並寫入到 go channel 中，另外一個從 channel 中讀取數字並列印到標準輸出。最終輸出五個隨機數。
憑直覺開始擼了以下程式碼：
func getRand(ch chan i...
有一道這樣的面試題目：
寫程式碼實現兩個 goroutine，其中一個產生隨機數並寫入到 go channel 中，另外一個從 channel 中讀取數字並列印到標準輸出。最終輸出五個隨機數。
憑直覺開始擼了以下程式碼：
func getRand(ch chan int) {
ch <- rand.Intn(5)
}
func printRand(ch chan int) {
fmt.Println("Rand Number = ", <-ch)
}
func main() {
rand.Seed(63)
ch := make(chan int)
for i := 1; i <= 5; i++ {
go getRand(ch)
go printRand(ch)
}
}
這個是大部分人憑直覺寫出來的，稍微有經驗的就會發現問題。執行起來會什麼都不列印程式就退出了，因為 main 方法沒有等 goroutine 退出就結束了。下面改造一下 main 方法。
func main() {
rand.Seed(63)
ch := make(chan int)
for i := 1; i <= 5; i++ {
go getRand(ch)
go printRand(ch)
}
time.Sleep(3 * time.Second) // 加這一行等待 goroutine
close(ch)
}
現在至少輸出達到要求了，但是還有問題。
- 等待 3 秒這個時間是拍腦袋寫的，並不科學。生產條件下你怎麼知道需要等多久呢？理想情況下是列印完 5 個數字兩個 goroutine 都退出，所以我們需要一個更完善的退出機制。
- 退出機制由誰來觸發？
func main() {
ch := make(chan int)
done := make(chan bool)
go func() {
for {
select {
case ch <- rand.Intn(5): // Create and send random number into channel
case <-done: // If receive signal on done channel - Return
return
default:
}
}
}()
go func() {
for i := 0; i < 5; i++ {
fmt.Println("Rand Number = ", <-ch) // Print number received on standard output
}
done <- true // Send Terminate Signal and return
return
}()
<-done // Exit Main when Terminate Signal received
}
上面是一個比較完美的答案，一般推薦使用匿名方法建立 goroutine，通過 Done channel 來實現 goroutine 之間的通知符合 Golang 的思維模式。
思考：
- 為什麼推薦匿名方法建立 goroutine？
- 除了 done channel 還有什麼辦法在 goroutine 間傳遞資訊？
- 這個問題是不是經典的生產者/消費者問題