回答

收藏

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

技术问答 技术问答 311 人阅读 | 0 人回复 | 2023-09-12

代码选择同一文件夹中的所有 xml 文件作为调用的可执行文件,并异步处理应用于回调方法的每个结果(文件名称仅在以下示例中打印)。4 ], s( a$ S$ x% V/ U
如何避免使用 sleep 防止 main 退出方法?我在处理频道时遇到了问题(我认为这是同步结果所需要的),所以谢谢你的帮助!& D0 O1 U" S. ^9 n
    package mainimport  "fmt"    "io/ioutil"    "path"    "path/filepath"    "os"    "runtime"    "time")func eachFile(extension string,callback func(file string))    exeDir := filepath.Dir(os.Args    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)}* }4 r7 a* ?/ b- c5 U; r! b
               
8 V/ c0 I' [% X, Z    解决方案:                                                                " {" C8 Y) n" b5 ]( B% t9 Y
                                                                您可以使用sync.WaitGroup。引用链接的例子:4 V3 ~0 k' R( e6 g. |
    package mainimport (       "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()}- i$ l& L) s3 e/ i
分享到:
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则