确保在 goroutine 完成之前不退出的 Go 程序

golang学习网今天将给大家带来《确保在 goroutine 完成之前不退出的 Go 程序》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

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

我尝试了一些方法，包括使用等待组，但问题仍然存在。我注意到，通过添加 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://go.dev/play/p/nzkiyoqaggd

读取关闭通道的示例 https://go.dev/play/p/hyzofbz0xc5

输出

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

理论要掌握，实操不能落！以上关于《确保在 goroutine 完成之前不退出的 Go 程序》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！