第二题：
创建一个生产器和接收器，再建立一个无缓冲的channel。生产器负责把数据放进管道里，接收器负责把管道里面的数据打印出来。这里我们开5个协程把数据打印出来。
直接上代码！
package main
import (
"fmt"
)
func receive(c <-chan int) {
/*for v := range c {
fmt.Println("received:", v)
}*/
for i := 0; i <= 1; i++ {
go func() {
for v := range c {
fmt.Println(v)
}
}()
}
}
func generator() <-chan int {
c := make(chan int)
for i := 0; i <= 9; i++ {
go func(i int) {
for j := 0; j <= 9; j++ {
temp := i*100 + 20 + j
c <- temp
}
close(c)
}(i)
}
return c
}
func main() {
c := generator()
receive(c)
}埋了个小坑，跑上面的代码，在这里是不会有任何输出的。
原因是main函数结束时程序就退出了,没有给goroutine足够运行的时间来打印输出。
整个流程是并发执行的,main函数、generator的goroutine、receive的goroutine都是并发运行。
但是问题是main函数和generator很快就结束了,程序退出,receive的goroutine来不及打印数据。
解决方法就是让main函数等一等receive的goroutine。我们在main函数中加上一句：
time.Sleep(time.Second * 5) 这时看到可以顺利输出了。
但是...
但是却panic了。为什么呢？
因为generator()把消息发送到了关闭的管道。是因为生成器goroutine和接收goroutine的生命周期没有控制好导致的。
主要原因在于,接收的goroutine一旦从通道接收完所有的数据并退出,通道就会被关闭。
而此时,生成器goroutine可能还在向这个通道发送数据,于是产生了panic。
要避免这种情况,需要确保:
1、接收goroutine在最后一个生成器goroutine退出之前不能退出。
2、生成器goroutine在关闭通道之前,必须保证接收goroutine仍在运行。
问题出在生成器中close(c)这一行。这里每个goroutine都在自己完成后关闭了通道c。
按照程序逻辑,通道c应该在最后一个goroutine完成时关闭一次,而不是每个goroutine都关闭。所以应该只在主goroutine中关闭c。这里我们用WaitGroup来同步。
func generator() <-chan int {
c := make(chan int)
var wg sync.WaitGroup
wg.Add(10) // 添加10个goroutine
for i := 0; i < 10; i++ {
go func() {
// 生成数据
wg.Done() // goroutine结束
}()
}
go func() {
wg.Wait() // 等待所有goroutine完成
close(c) // 关闭通道,仅关闭一次
}()
return c
}
顺利输出！！
