如何在不使用 time.Sleep 的情况下等待所有 goroutines 完成？

一尘不染

如何在不使用 time.Sleep 的情况下等待所有 goroutines 完成？

此代码选择同一文件夹中的所有 xml 文件，作为调用的可执行文件，并异步将处理应用于回调方法中的每个结果（在下面的示例中，仅打印出文件的名称）。

如何避免使用 sleep 方法来防止 main 方法退出？我在处理频道时遇到了问题（我认为这就是同步结果所需要的），因此感谢您的帮助！

package main

import (
    "fmt"
    "io/ioutil"
    "path"
    "path/filepath"
    "os"
    "runtime"
    "time"
)

func eachFile(extension string, callback func(file string)) {
    exeDir := filepath.Dir(os.Args[0])
    files, _ := ioutil.ReadDir(exeDir)
    for _, f := range files {
            fileName := f.Name()
            if extension == path.Ext(fileName) {
                go callback(fileName)
            }
    }
}


func main() {
    maxProcs := runtime.NumCPU()
    runtime.GOMAXPROCS(maxProcs)

    eachFile(".xml", func(fileName string) {
                // Custom logic goes in here
                fmt.Println(fileName)
            })

    // This is what i want to get rid of
    time.Sleep(100 * time.Millisecond)
}

阅读 181

2021-11-18

共1个答案

一尘不染

您可以使用sync.WaitGroup。引用链接的例子：

package main

import (
        "net/http"
        "sync"
)

func main() {
        var wg sync.WaitGroup
        var urls = []string{
                "http://www.golang.org/",
                "http://www.google.com/",
                "http://www.somestupidname.com/",
        }
        for _, url := range urls {
                // Increment the WaitGroup counter.
                wg.Add(1)
                // Launch a goroutine to fetch the URL.
                go func(url string) {
                        // Decrement the counter when the goroutine completes.
                        defer wg.Done()
                        // Fetch the URL.
                        http.Get(url)
                }(url)
        }
        // Wait for all HTTP fetches to complete.
        wg.Wait()
}

2021-11-18