Go Channel 面試題解析

有一道這樣的面試題目：

寫代碼實現兩個 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)
}

如今至少輸出達到要求了，可是還有問題。

1. 等待 3 秒這個時間是拍腦殼寫的，並不科學。生產條件下你怎麼知道須要等多久呢？理想狀況下是打印完 5 個數字兩個 goroutine 都退出，因此咱們須要一個更完善的退出機制。
2. 退出機制由誰來觸發？

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 的思惟模式。

思考：

1. 爲何推薦匿名方法建立 goroutine？
2. 除了 done channel 還有什麼辦法在 goroutine 間傳遞信息？
3. 這個問題是否是經典的生產者/消費者問題