我使用sync.WaitGroup,defer wg.Close()并wg.Wait()等待goroutine完成。
sync.WaitGroup
defer wg.Close()
wg.Wait()
该程序确实会等待,但是它永远不会退出。
这是我的程序(可运行):
package main import ( "fmt" "io" "log" "net/http" "os" "sync" ) var symbols = []string{ "ASSA-B.ST", "ELUX-B.ST", "HM-B.ST", } func main() { fmt.Println("fetching quotes...") fetchedSymbols := make(chan string) var wg sync.WaitGroup wg.Add(len(symbols)) for _, symbol := range symbols { go fetchSymbol(symbol, &wg, fetchedSymbols) } for response := range fetchedSymbols { fmt.Println("fetched " + response) } wg.Wait() fmt.Println("done") } func fetchSymbol(symbol string, wg *sync.WaitGroup, c chan<- string) { defer wg.Done() resp, err := http.Get("http://ichart.yahoo.com/table.csv?s=" + symbol + "&a=0&b=1&c=2000") defer resp.Body.Close() if err != nil { log.Fatal(err) } out, err := os.Create("./stock-quotes/" + symbol + ".csv") defer out.Close() if err != nil { log.Fatal(err) } io.Copy(out, resp.Body) c <- symbol }
下载所有报价后,该程序是否应该退出?(仅供参考:我刚刚开始学习GO)
您永远不会关闭fetchedSymbols通道,因此范围循环永远不会退出。
fetchedSymbols
解决此问题的一种方法是使用WaitGroup,您已经必须在关闭通道时发出信号。调整范围fetchedSymbols足以阻止main中的进度,并且您不需要其他通道或WaitGroup。
... go func() { wg.Wait() close(fetchedSymbols) }() for response := range fetchedSymbols { fmt.Println("fetched " + response) } ...