当前位置： > > > > 如何让多个 go 例程等待另一个 goroutine 的输出

如何让多个 go 例程等待另一个 goroutine 的输出

来源：stackoverflow
2024-04-28 14:06:36
0浏览
收藏

Golang不知道大家是否熟悉？今天我将给大家介绍《如何让多个 go 例程等待另一个 goroutine 的输出》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

问题内容

main {
go routine_1(carryout a time consuming task and return output)
go routine_2(wait for output from routine_1, collect output and do task_2)
go routine_3(wait for output from routine_1, collect output and do task_3)
wait for all routines to complete
}

我使用了通道，但是一旦routine_2读取数据，routine_3就无法获取数据。

我不想让主线程进行同步，即知道routine_2、routine_3到routine_1之间的依赖关系

事实证明，关闭通道将为其他例程提供所需的广播，以了解结果已准备好。

感谢您抽出时间。（这是我在 stackoverflow 中的第一个问题，很高兴看到如此好的质量和快速的回​​复）谢谢大家。

更新我选择的答案。再次感谢。

package main

import (
    "fmt"
    "sync"
)

func computeResult() int {
    return 100
}

func main() {

    var wg sync.WaitGroup
    wg.Add(2)
    var output int

    ch := make(chan struct{})

    // Compute result
    go func() {
        defer wg.Done()
        output = computeResult()
        fmt.Println("closing channel to signal task done")
        close(ch)
        fmt.Println("channel closed")

    }()

    go func() {
        defer wg.Done()
        // Wait for ch close
        <-ch
        fmt.Println(output)
    }()

    wg.Wait()

    var wg2 sync.WaitGroup
    wg2.Add(1)

    go func() {
        defer wg2.Done()
        fmt.Println("wait on closed channel")
        // Wait for ch close
        <-ch
        fmt.Println(output)
    }()

    wg2.Wait()

}

解决方案

您可以使用通道的关闭来广播结果准备情况，并使用共享变量来存储结果：

func() {
   var output sometype
   ch:=make(chan struct{})

   // compute result
   go func() {
      output=computeresult()
      close(ch)
   }()

   go func() {
     // wait for ch close
     <-ch
      useoutput()
    }()
}

您可以有许多 goroutine 等待通道关闭。当通道关闭时，所有 goroutine 将被启用。

您可以在第一个例程完成后调用其他两个例程：

var wg sync.WaitGroup
wg.Add(2)
go func() {
    output := task1()
    go func() {
        defer wg.done()
        task2(output)
    }
    go func() {
        defer wg.done()
        task3(output)
    }
}()
wg.Wait()

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何让多个 go 例程等待另一个 goroutine 的输出》文章吧，也可关注米云公众号了解相关技术文章。