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. 这个问题是否是经典的生产者/消费者问题