登录
首页 >  Golang >  Go问答

如何在 goroutine 函数中等待多个其他函数返回的值?

来源:stackoverflow

时间:2024-03-01 19:18:24 304浏览 收藏

大家好,我们又见面了啊~本文《如何在 goroutine 函数中等待多个其他函数返回的值?》的内容中将会涉及到等等。如果你正在学习Golang相关知识,欢迎关注我,以后会给大家带来更多Golang相关文章,希望我们能一起进步!下面就开始本文的正式内容~

问题内容

我创建了一个 go 演示示例来说明我正在谈论的内容。

在示例中,我创建了一个 goroutine func2,我只想等待通过其通道的输入,并在值到达后打印该值。

func func2(ch chan int) {
    fmt.println("func2")
    v:=<-ch
    fmt.println(v)
}

然后,在循环中,我为另一个函数创建 goroutine,它们是 waitgroup 的一部分。

func func1(ch chan int, wg *sync.WaitGroup) {
    fmt.Println("func1")
    ch <- 11032
    wg.Done()
}

在 main 中,我等待 waitgroup。我遇到了僵局,我不知道如何解决它。为了清楚地表明我想要实现的目标,我希望 func2 在调用它后作为线程保持打开状态,以处理 n 个值,其中 n 是我为 func1 调用的 goroutine 数量。我考虑过在 func2 内部使用 waitgroup wait,但我不希望它阻塞,因为它需要在发送时处理来自 func1 的新数据。


解决方案


我认为您陷入了僵局,因为您的 func2 仅消耗了 ch 中的 1 值,然后完成。然后其他 func1 goroutines 就会卡住,等待 ch 可供写入,但它们无法执行此操作,因为没有其他 goroutine 可以从另一端的 ch 读取。

由于您希望 func2 不断消耗 ch 中的值,直到 ch 关闭,因此您需要在 func2 中创建一个循环,如下所示:

func func2(ch chan int) {
    fmt.println("func2")
    for v := range ch {
        fmt.println(v)
    }
}

这将使 func2 保持“活动”并从 ch 读取数据,直到您在其他地方执行 close(ch) 为止。在您的示例中关闭 ch 的适当位置可能是在 wg.wait() 之后的 main 中。

如果您想确保在程序完成之前看到所有 println 语句的结果,您还应该使用某种同步机制来等待 func2 完成。否则,main 将在 close(ch) 之后立即结束,这可能会也可能不会func2 打印它收到的每个值之前。

实现此目的的常用技术是“完成”通道。例如:

func func2(ch chan int, done chan bool) {
    fmt.println("func2")
    for v := range ch {
        fmt.println(v)
    }
    done <- true
}

main 中:

done := make(chan bool)
go func2(ch, done)
...
wg.wait()
close(ch)
<-done

使用 chan struct{}(空结构)也很常见,因为空结构不需要内存。

您正在寻找的是一种在所有值完成后关闭通道的干净方法流过。当新值出现时,您可以创建一个新通道并开始在该通道上传输值。这就是我的做法,也许有帮助

var wg sync.WaitGroup

toUpdate := make(chan *someType, BufferSize)

for i := 0; i < BufferSize; i++ {
    go processEvents(toUpdate, &wg)
}

// // wait till all the checks have come back
go func(toUpdate chan * someType, group *sync.WaitGroup) {
    group.Wait()
    close(toCreate)
}(toCreate, toUpdate, &wg)

今天关于《如何在 goroutine 函数中等待多个其他函数返回的值?》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

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