如何在不使用 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
|
|
|
|
|
|