卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34515本站已运行393

Go 程序在 goroutine 工作完成之前退出

go 程序在 goroutine 工作完成之前退出

php小编小新在这篇文章中将介绍一个关于Go程序的重要问题:在goroutine工作完成之前退出的情况。在Go语言中,goroutine是一种轻量级的线程,可以并发执行任务。然而,当我们的程序可能在goroutine工作完成之前退出时,我们需要了解如何处理这种情况,以确保我们的程序能够正确地完成任务。在接下来的内容中,我们将探讨这个问题,并提供一些解决方案来解决它。

问题内容

我在了解如何正确阻止和关闭频道时遇到问题。我正在启动任意数量的工作人员,我发现我的主要功能要么在工作人员完成之前退出,要么由于未关闭的通道而挂起。我需要一种更好的方法来阻止工作人员在不退出主通道的情况下读取通道,然后在完成后优雅地关闭通道以结束循环。我所做的任何尝试都以僵局告终。

我尝试了一些方法,包括使用等待组,但问题仍然存在。我注意到,通过添加 time.sleep,程序按预期工作,但将其注释掉会导致没有完成任何工作。

time.sleep(time.duration(10 * time.second))

这是一个可运行的示例 https://go.dev/play/p/qhqnj-ajqbi,其中保留了 sleep。这是注释掉睡眠超时的损坏代码。

package main

import (
    "fmt"
    "sync"
    "time"
)

// some complicated work
func do(num int, ch chan<- int) {
    time.sleep(time.duration(500 * time.millisecond))
    ch <- num
}

func main() {

    results := make(chan int)

    // for some number of required complicated work
    for i := 0; i < 53; i++ {
        go do(i, results)
    }

    var wg sync.waitgroup

    // start 3 workers which can process results
    for i := 0; i < 3; i++ {
        wg.add(1)
        go func(id int) {
            defer wg.done()
            worker(id, results)
        }(i)
    }

    // handle closing the channel when all workers complete
    go func() {
        wg.wait()
        close(results)
    }()

    //time.sleep(time.duration(10 * time.second))

    fmt.println("donezo")
}

// process the results of do() in a meaningful way
func worker(id int, ch <-chan int) {
    fmt.println("starting worker", id)

    for i := range ch {
        fmt.println("channel val:", i)
    }
}

我还尝试将 defer wg.done() 移动到 worker() func 内部,但这是同样的问题,并且在没有睡眠的情况下无法工作。

// process the results of do() in a meaningful way
func worker(wg *sync.WaitGroup, id int, ch <-chan int) {
    fmt.Println("starting worker", id)

    defer wg.Done()

    for i := range ch {
        fmt.Println("channel val:", i)
    }
}

我是否选择了错误的范式,或者我只是使用了错误的范式?

解决方法

我最初问“我可以对我的代码进行一些小调整来使其工作吗?还是我必须重新考虑这个问题?”我发现的答案是,是的,有是一个小调整。

我必须学习一个关于通道的有趣的基本概念:您可以从封闭的通道中读取数据,即排空通道。正如我最初的示例中提到的 range 永远不会终止,因为我找不到关闭通道的好地方,即使当我以其他创造性的方式强制它时,程序也会表现出不良行为

  • 未处理完频道内的所有内容而退出
  • 死锁或在封闭通道上发送

这是因为“真实”代码的细微差别,其中处理通道内容所需的时间比填充所需的时间更长频道和事物不同步。

由于我的发送者中没有明确的实用方法来关闭通道(99% 的通道教程中都建议这样做),因此当您有多个工作人员正在读取通道并且工作人员不知道时,通过 goroutine 在 main 中执行此操作实际上是可以接受的其中哪些读取了最后一个值。

解决方案

我将工作人员包装在自己的 sync.waitgroup 中,并使用 worker.wait()阻止程序退出,从而允许工作“完成” ”。当没有更多数据要发送时,我独立地 close() 通道,即我通过使用他们自己的等待组等待编写者完成来阻止。 close 为范围循环提供了一个终止情况,因为当返回通道的默认值时,即到达通道结束时返回 eof 类型,它将结束。阻塞交会通道没有终点,直到它被关闭。

我对此的看法是,如果您不知道将并行推送多少个值,则 go 无法知道无缓冲通道的长度,因为它正在范围内,直到您关闭它。。由于关闭,它表示读取剩下的任何内容,直到终止值或结束。 workers.wait() 会阻塞,直到完成。

已解决操作的示例 https://www.php.cn/link/2bf0ccdbb4d3ebbcb990af74bd78c658

读取关闭通道的示例 https://www.php.cn/link/d5397f1497b5cdaad7253fdc92db610b

输出

filling 0
filling 1
filling 2
filling 3
filling 4
filling 5
filling 6
filling 7
filling 8
filling 9
closed
empyting 0
empyting 1
empyting 2
empyting 3
empyting 4
empyting 5
empyting 6
empyting 7
empyting 8
empyting 9
卓越飞翔博客
上一篇: 如何在 Go 中运行前台或后台 shell 命令
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏