文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

终止无限循环 goroutine 的安全方法?

2024-02-06 09:22

关注

问题内容

我有一个充当侦听器的 goroutine。输入流传入某个缓冲的 channel,我希望我的 goroutine 处理传入该通道的数据。然而,有时 channel 可能会暂时没有数据输入。如果 channel 一秒钟没有什么可提供的,我希望我的 goroutine 做一些不同的事情。该函数如下所示:

func main () {
    var wg sync.WaitGroup
    arr := make([]*myObject, 0)
    wg.Add(1)
    go listener(c, arr, &wg)
    for {
        // sending stuff to c
    }
}

func listener(c chan *myObject, arr []*myObject, wg *sync.WaitGroup) {
    for {
        select {
        case value := <- c:
            arr = append(arr, value)
        case <- time.After(1 * time.Second):
            fmt.Println(arr)
        }
}

问题是我想看到通过该通道的所有内容都打印出来。如果 main 突然结束,可能 arr 中还剩下一些东西还没有打印出来,我就看不到了。所以我需要确保这个goroutine在程序结束之前处理完通道中的所有数据。我认为,这意味着我需要使用 WaitGroup 并使用 Wait() 来确保程序在我的 goroutine 完成需要执行的操作之前不会关闭。 但我不知道在我的 WaitGroup 上调用 Done() 的位置。

基本上,我需要一种安全的方法来在程序结束之前“暂停”goroutine,并打印出剩下的内容。我怎样才能做到这一点?

更糟糕的是,对于单元测试之类的事情,我自己将数据发送到通道,在发送一定数量后,我想查看数组。但是,如果我只是在将数据发送到通道的代码之后立即检查数组,则 goroutine 可能还没有机会处理所有数据。在这种情况下,我想等待 goroutine 处理我发送的所有数据,然后我想暂停它并要求它向我显示数组。但我如何知道 goroutine 何时完成处理呢?我可以 sleep 一段时间,给它一个完成的机会,然后停下来看看,但这感觉相当黑客。我认为有一个最佳实践方法可以解决这个问题,但我还没有弄清楚。

这是我的一些想法,其中零个有效。

  1. 在无限 for 循环之外调用 Done()。这不起作用,因为据我所知,该代码无法访问。

  2. 在超时case中调用Done()。这是行不通的,因为超时后可能会有更多数据在路上,我希望我的 goroutine 继续监听。


正确答案


修改侦听器以在通道关闭时返回。返回时调用 wg.Done():

func listener(c chan *myObject, arr []*myObject, wg *sync.WaitGroup) {
    defer wg.Done()
    for {
        select {
        case value, ok := <- c:
            if !ok { return }
            arr = append(arr, value)
        case <- time.After(1 * time.Second):
            fmt.Println(arr)
        }
}

修改 main 以在发送完成后关闭通道。在从 main 返回之前等待 goroutine 完成。

var wg sync.WaitGroup
arr := make([]*myObject, 0)
wg.Add(1)
go listener(c, arr, &wg)
for {
    // sending stuff to c
}
close(c)
wg.Wait()

以上就是终止无限循环 goroutine 的安全方法?的详细内容,更多请关注编程网其它相关文章!

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 资料下载
  • 历年真题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯