登录
首页 >  Golang >  Go问答

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

来源:Golang技术栈

时间:2023-04-15 12:02:59 344浏览 收藏

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

问题内容

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

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

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

package main

import (
    "time"
    "fmt"
)


func ProcessToDo(done chan struct{}, todo chan string) {
    for {
        select {
        case work, ok := 

正确答案

done在您的情况下,通道是完全没有必要的,因为您可以通过关闭todo通道本身来发出关闭信号。

for range在通道上使用 ,它将迭代直到通道关闭并且其缓冲区为空。

你应该有一个done通道,但只是为了让 goroutine 本身可以发出它完成工作的信号,以便主 goroutine 可以继续或退出。

这个变体等同于你的变体,更简单,不需要time.Sleep()调用等待其他 goroutines(无论如何这太错误和不确定)。在Go Playground上尝试一下:

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 

还要注意,worker goroutines 应该使用完成发出信号,defer这样如果主 goroutine 以某种意外的方式返回或恐慌,它就不会卡在等待 worker 的等待中。所以它应该像这样开始:

defer func() {
    done 

您还可以使用sync.WaitGroup将主 goroutine 同步到工作程序(等待它)。事实上,如果你打算使用多个工作 goroutine,那比从done通道中读取多个值更干净。WaitGroup此外,由于它有一个方法(这是一个函数调用),因此它更简单地发出完成信号,Done()因此您不需要匿名函数:

defer wg.Done()

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

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

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

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

for i := 0; i 

无需更改任何其他内容,您现在拥有一个正确的并发应用程序,它使用 10 个并发 goroutines 接收和处理您的作业。而且我们没有使用任何“丑陋” time.Sleep()(我们使用了一个但只是为了模拟慢速处理,而不是等待其他 goroutines),并且您不需要任何额外的同步。

以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于Golang的相关知识,也可关注golang学习网公众号。

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表