缓冲区为空后关闭“工人”执行例行程序

一分耕耘，一分收获！既然打开了这篇文章《缓冲区为空后关闭“工人”执行例行程序》，就坚持看下去吧！文中内容包含golang等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

问题内容

我希望我的日常工作人员（ProcessToDo()在下面的代码中）等到所有“排队”的工作都处理完后再关闭。

工作例程有一个“待办事项”通道（缓冲），工作通过该通道发送给它。它有一个“完成”通道来告诉它开始关闭。该文档说，如果满足多个选择，则通道上的选择将选择一个“伪随机值”......这意味着在所有缓冲工作完成之前触发关闭（返回）。

在下面的代码示例中，我希望打印所有 20 条消息...

package main

import (
    "time"
    "fmt"
)


func ProcessToDo(done chan struct{}, todo chan string) {
    for {
        select {
        case work, ok := 
<h2 class="daan">
    正确答案
</h2>
<p><code>done</code>在您的情况下，通道是完全没有必要的，因为您可以通过关闭<code>todo</code>通道本身来发出关闭信号。</p>
<p>并<code>for range</code>在通道上使用 ，它将迭代直到通道关闭并且其缓冲区为空。</p>
<p>你应该有一个<code>done</code>通道，但只是为了让 goroutine 本身可以发出它完成工作的信号，以便主 goroutine 可以继续或退出。</p>
<p>这个变体等同于你的变体，更简单，不需要<code>time.Sleep()</code>调用等待其他 goroutines（无论如何这太错误和不确定）。<a target='_blank'  href='https://www.17golang.com/gourl/?redirect=MDAwMDAwMDAwML57hpSHp6VpkrqbYLx2eayza4KafaOkbLS3zqSBrJvPsa5_0Ia6sWuR4Juaq6t9nq5roGCUgXpusdyfr5Z5fZqwoZXamqqqrpLfZabFn5OqyHxxhommpYrJuJ99h3inuq2tfs6B3LKkkrqBZK92gqC0jYqbfaN-a77Qs7GDhp6as3uG3oaVsbOFmIVhu6yKnrSKdW0' rel='nofollow'>在Go
Playground</a>上尝试一下：</p>
<pre class="brush:go;toolbar:false">func ProcessToDo(done chan struct{}, todo chan string) {
    for work := range todo {
        fmt.Printf("todo: %q\n", work)
        time.Sleep(100 * time.Millisecond)
    }
    fmt.Printf("Shutting down ProcessToDo - todo channel closed!\n")
    done 
<p>还要注意，worker goroutines 应该使用完成发出信号，<code>defer</code>这样如果主 goroutine
以某种意外的方式返回或恐慌，它就不会卡在等待 worker 的等待中。所以它应该像这样开始：</p>
<pre class="brush:go;toolbar:false">defer func() {
    done 
<p>您还可以使用<a target='_blank'  href='https://www.17golang.com/gourl/?redirect=MDAwMDAwMDAwML57hpSHp6VpkrqbYLx2eayza4KafaOkbLS3zqSBrJvPsa5_0Ia6sWuR4Juaq6t9nq5roGCUgXpusdyfppZkr83GoZTZmpaup4SZcqK8q2muypBtmnyyh5C-utJohmWB28iMcs6B3LGjgZiCnLB2l6uzgI1lfoCKsbK3yqGNrICas3icmJK3sWyR4H2qvIaGnrOmhWU' rel='nofollow'><code>sync.WaitGroup</code></a>将主 goroutine
同步到工作程序（等待它）。事实上，如果你打算使用多个工作
goroutine，那比从<code>done</code>通道中读取多个值更干净。<code>WaitGroup</code>此外，由于它有一个方法（这是一个函数调用），因此它更简单地发出完成信号，<a target='_blank'  href='https://www.17golang.com/gourl/?redirect=MDAwMDAwMDAwML57hpSHp6VpkrqbYLx2eayza4KafaOkbLS3zqSBrJvPsa5_0Ia6sWuR4Juaq6t9nq5roGCUgXpusdyfppZkr83GoZTZmpaup4SZcqK8q2muypBtmnyyh5C-utJohmWB28iMctmKqp2ukrl9mqurfZ6_gI1gf316aL-30WiOdoDdsmZ7zoXNz22Hupicr6yUnrKzhpx-kIZqvt3RdQ' rel='nofollow'><code>Done()</code></a>因此您不需要匿名函数：</p>
<pre class="brush:go;toolbar:false">defer wg.Done()

有关完整示例，请参阅JimB 的答案WaitGroup。

如果你想使用多个工作 goroutine，使用for range也是惯用的：通道是同步的，所以你不需要任何额外的代码来同步对todo通道的访问或从它接收的作业。如果你关闭 中的todo通道main()，这将正确地向所有工作 goroutine 发出信号。但当然，所有排队的作业都将被接收和处理一次。

现在采用WaitGroup 用于使主 goroutine 等待工作者的变体（JimB 的回答）：如果你想要超过 1 个工作者 goroutine 怎么办？同时处理您的工作（并且很可能是并行处理）？

您唯一需要在代码中添加/更改的是：真正启动其中的多个：

for i := 0; i 
<p>无需更改任何其他内容，您现在拥有一个正确的并发应用程序，它使用 10 个并发 goroutines 接收和处理您的作业。而且我们没有使用任何“丑陋”
<code>time.Sleep()</code>（我们使用了一个但只是为了模拟慢速处理，而不是等待其他 goroutines），并且您不需要任何额外的同步。</p><p>以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。</p>