鉴于您在一个频道上确实有多个作者,因此您会遇到一些挑战,因为在Go中执行此 *** 作的简单方法通常是在一个频道上拥有一个作者,然后让该作者关闭频道。发送最后一个数据时的通道:
func produce(... args including channel) { defer close(ch) for stuff_to_produce { ch <- item }}
这种模式具有很好的特性,无论您如何退出
produce,通道都会关闭,从而指示生产已结束。
您没有使用这种模式,而是向多个goroutine传递了一个通道,每个goroutine都可以发送 一条
消息,因此您需要移动
close(当然,也可以使用其他模式)。表达所需模式的最简单方法是:
func overall_produce(... args including channel ...) { var pg sync.WaitGroup defer close(ch) for stuff_to_produce { pg.Add(1) go produceInParallel(ch, &pg) // add more args if appropriate } pg.Wait()}
该
pg计数器累计活跃的生产者。每个调用都必须使用
pg.Done()来表明已完成
ch。总制片人现在等待他们全部完成,那么 它
关闭的道路上走出通道。
(如果将内部
produceInParallel函数编写为闭包,则无需显式传递
ch并传递
pg给它。也可以将其编写
overallProducer为闭包。)
请注意,单个使用者的循环可能最好使用以下
for ... range结构来表示:
func receive(msg <-chan message, wg *sync.WaitGroup) { for m := range msg { fmt.Println("Received:", m) } wg.Done()}
(您提到了
select向循环添加a的意图,以便在消息尚未准备好时可以执行其他一些计算。如果无法将该代码分解为独立的goroutine,则实际上您将需要更高级的
m,ok := <-msg构造。)
还要注意,
wgfor
receive(取决于您构造其他事物的方式可能是不必要
pg的)与生产者的等待组非常独立。诚然,按照书面说明,只有在所有生产者都完成之后才能完成消费者的工作,但我们希望独立等待生产者完成,以便我们可以关闭整体生产者包装中的渠道。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)