一个完成后如何完成所有的goroutine

Question

我想运行多个goroutine，进行一些处理，将结果放入channel，当至少有一个goroutine完成时，完成所有其他goroutine并从channel返回结果。

所以，我尝试使用等待组来实现这一点，但似乎我没有正确使用等待组。

    package optimizer

    import (
        "github.com/tevino/abool"
        "errors"
        "sync"
        "runtime"
        "log"
    )

    type Optimizer struct {
        Handlers      []ofdHandler.Handler
    }

    func Make(handlers []ofdHandler.Handler, maxProcs int) Optimizer {
        runtime.GOMAXPROCS(maxProcs)
        return Optimizer{Handlers: handlers}
    }

    func (o Optimizer) Optimize(params operations.GetV1ReceiptsParams) (*models.Receipt, error) {
        var wg sync.WaitGroup
        wg.Add(len(o.Handlers))

        results := make(chan *models.Receipt)
        isCalculated := abool.NewBool(false)

        for _, handler := range o.Handlers {
            go func(handler ofdHandler.Handler) {
                log.Println("Starting handler: ", handler.GetName())
                defer wg.Done()

                if isCalculated.IsSet() {
                    log.Println("Result is calculated, exiting goroutine...")
                    return
                }

                receipt, err := handler.Handle(params)
                if err != nil {
                    log.Println(err)
                    return
                }

                if isCalculated.IsSet() {
                    log.Println("Result is calculated, exiting goroutine...")
                    return
                }

                log.Println("Writing result to channel...")
                isCalculated.Set()
                results <- receipt
            }(handler)
        }

        log.Println("Waiting...")
        wg.Wait()

        if receipt, ok := <-results; ok {
            return receipt, nil
        }

        return nil, errors.New("couldn't optimize with current list of Handlers")
    }

Answer 1

最好的方法是使用context。在Go blog 上有一个很好的示例，说明如何针对您的问题类型执行此操作。

Answer 2

您发布的代码不能按原样编译，您也没有描述问题（发生了什么而不是预期结果），但是通过查看代码我有两个建议：

使结果通道缓冲，长度为1：

results := make(chan *models.Receipt, 1)

这允许首先完成的 goroutine 将结果写入通道并退出，从而减少等待组的计数器。

最后，你应该使用SetToIf 而不是isCalculated.Set()，这样你就不会设置标志/结果两次（因此停滞不前，因为读者仍在等待等待组“完成”等等第二个 goroutine 将无法写入通道，这意味着等待组永远不会达到零）：

log.Println("Writing result to channel...")
if isCalculated.SetToIf(false, true) {
   results <- receipt
}