尝试阻止goroutine无限创建引发恐慌

Golang不知道大家是否熟悉？今天我将给大家介绍《尝试阻止goroutine无限创建引发恐慌》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

我正在尝试并行化对 api 的调用以加快速度，但我遇到了一个问题，如果我收到来自其中一个 goroutine 调用的错误，我需要停止旋转 goroutine 来调用 api。由于我关闭通道两次（一次在错误处理部分，一次在执行完成时），所以我收到 panic: close of returned channel 错误。有没有一种优雅的方法来处理这个问题而不会让程序惊慌？任何帮助将不胜感激！

以下是伪代码片段。

for i := 0; i < somenumber; i++ {
    go func(num int, q chan<- bool) {
        value, err := callanapi()
        if err != nil {
            close(q)//exit from the for-loop
        }
        // process the value here
        wg.done()
    }(i, quit)
}
close(quit)

为了模拟我的场景，我编写了以下程序。一旦满足条件（注释掉），有没有办法优雅地退出for循环？

package main

import (
    "fmt"
    "sync"
)

func receive(q <-chan bool) {
    for {
        select {
        case <-q:
            return
        }
    }
}

func main() {
    quit := make(chan bool)

    var result []int
    wg := &sync.WaitGroup{}
    wg.Add(10)
    for i := 0; i < 10; i++ {
        go func(num int, q chan<- bool) {
            //if num == 5 {
            //  close(q)
            //}
            result = append(result, num)

            wg.Done()
        }(i, quit)
    }
    close(quit)
    receive(quit)

    wg.Wait()

    fmt.Printf("Result: %v", result)
}

解决方案

您可以使用定义 context 类型的 context 包，该类型跨 api 边界和进程之间携带截止日期、取消信号和其他请求范围的值。

package main

import (
    "context"
    "fmt"
    "sync"
)

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel() // cancel when we are finished, even without error

    wg := &sync.WaitGroup{}

    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(num int) {
            defer wg.Done()
            select {
            case <-ctx.Done():
                return // Error occured somewhere, terminate
            default: // avoid blocking
            }

            // your code here
            // res, err := callAnAPI()
            // if err != nil {
            //  cancel()
            //  return
            //}

            if num == 5 {
                cancel()
                return
            }

            fmt.Println(num)

        }(i)
    }

    wg.Wait()

    fmt.Println(ctx.Err())
}

试穿：Go Playground

您还可以查看此 answer​​ 以获取更详细的说明。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《尝试阻止goroutine无限创建引发恐慌》文章吧，也可关注golang学习网公众号了解相关技术文章。