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. 这个问题是不是经典的生产者/消费者问题
    原文作者:后端开发
    原文地址: https://segmentfault.com/a/1190000018527076
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞