Làm thế nào để chờ tất cả các goroutin kết thúc mà không cần sử dụng time.Sleep?

Question

Mã này chọn tất cả các tệp xml trong cùng một thư mục, như thực thi được gọi và áp dụng không đồng bộ cho mỗi kết quả trong phương thức gọi lại (trong ví dụ bên dưới, chỉ tên của tệp được in ra).

Làm cách nào để tránh sử dụng phương pháp ngủ để giữ phương pháp chính thoát khỏi? Tôi có vấn đề bao quanh đầu của tôi xung quanh các kênh (tôi giả định đó là những gì nó cần, để đồng bộ hóa kết quả) để giúp đỡ bất kỳ được đánh giá cao!

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) 
} 

Answer 1

Bạn có thể sử dụng sync.WaitGroup. Trích dẫn ví dụ được liên kết:

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() 
} 

Answer 2

WaitGroups chắc chắn là cách kinh điển để thực hiện việc này. Tuy nhiên, vì lợi ích của sự hoàn chỉnh, đây là giải pháp thường được sử dụng trước khi WaitGroups được giới thiệu. Ý tưởng cơ bản là sử dụng một kênh để nói "Tôi đã hoàn thành" và đợi goroutine chính cho đến khi mỗi thói quen sinh sản đã báo cáo sự hoàn thành của nó.

func main() { 
    c := make(chan struct{}) // We don't need any data to be passed, so use an empty struct 
    for i := 0; i < 100; i++ { 
     go func() { 
      doSomething() 
      c <- struct{}{} // signal that the routine has completed 
     }() 
    } 

    // Since we spawned 100 routines, receive 100 messages. 
    for i := 0; i < 100; i++ { 
     <- c 
    } 
}