登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-22 12:31:28 457浏览 收藏

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学习网公众号了解相关技术文章。

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