如何在主动从通道读取数据的同时检查 goroutine 是否完成？

珍惜时间，勤奋学习！今天给大家带来《如何在主动从通道读取数据的同时检查 goroutine 是否完成？》，正文内容主要涉及到等等，如果你正在学习Golang，或者是对Golang有疑问，欢迎大家关注我！后面我会持续更新相关内容的，希望都能帮到正在学习的大家！

很难用一句话来表达这个问题。情况如下：

我正在尝试生成一组 goroutine 来递归目录并查找匹配的文件。然后我收集这些文件并继续处理它们。然而，问题是我不知道每个例程会找到多少个文件，所以我很难弄清楚如何在所有例程完成后让主线程退出

我可以让通道缓冲区变得非常大，但这不是一个好的解决方案，这个工具不需要 100% 健壮，但足够好，不会一直损坏。另外，它有可能会出现很多文件

// start a routine to traverse each directory
fpchan := make(chan string, 100)
for _, dir := range numdirs {
    fmt.printf("searching for file in %s\n", dir)
    go findlogs(searchstring, dir, fpchan)
}

// collect filepaths from channel
files := make([]string, 0, maxlogs)
for file := range fpchan { // i'll get stuck when everything completes, nothing to receive
    if len(files) <= cap(files) {
        files.append(file)
    } else {
        fmt.println("reached max logfile count of %d\n", maxlogs)
    }
}

等待组实际上不起作用，因为通道可能会填满并且例程会被卡住（因为我不知道提前会有多少结果）

在通道上发送空字符串作为 goroutine 发出“完成”信号的方式是否合规？就像下面这样：

// collect filepaths from channel
files := make([]string, 0, maxLogs)
for file := range fpchan {
    if file == "" {
        fmt.Println("goroutine finished")
        numRunning--
        if numRunning == 0 {
             break
        }
        continue
    }
    if len(files) <= cap(files) {
        files.append(file)
    } else {
        fmt.Println("Reached max logfile count of %d\n", maxLogs)
    }
}

对于这种情况，由于空文件路径无效，因此它可以作为穷人的信号。这感觉就像是一个黑客，我认为应该有更好的解决方案

有什么方法可以以一种不太复杂的方式做到这一点（一堆额外的通道、非阻塞接收等）？如果比这复杂得多，我只会按顺序执行它们，但我认为这是利用并发的好机会

正确答案

您可以在此处使用 waitgroup。但是您需要在 goroutine 中等待 waitgroup，并且在该 goroutine 中，当 waitgroup 完成时关闭通道，以便主循环终止：

var wg sync.WaitGroup
// start a routine to traverse each directory
fpchan := make(chan string, 100)
for _, dir := range numDirs {
    fmt.Printf("Searching for file in %s\n", dir)
    wg.Add(1)
    go func(dir string) {
        findLogs(searchString, dir, fpchan)
        wg.Done()
    }(dir)
}

go func() {
    wg.Wait()
    close(fpchan)
}()

（您在 fpchan 上的收集循环保持不变）

到这里，我们也就讲完了《如何在主动从通道读取数据的同时检查 goroutine 是否完成？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！