执行例程从 URL 下载到文件

Question

我正在学习 GO。在学习了一些基础知识之后，我一直在尝试编写一个小程序，它可以使用 goroutines 将网页（切片中的 url）同时下载到不同的文件中。这是我写的一些代码：

func downloadFromUrl(url string) {
    tokens := strings.Split(url, "/")
    fileName := tokens[len(tokens)-1]
    // I took out the bit that download the file for testing.
    fmt.Println("Downloading", url, "to", fileName)

}

我注释掉了实际下载页面进行测试的部分。在我的主要功能中，我正在这样做：

func main() {
    urls := []string{"http://www.google.com", "http://www.yahoo.com", "http://www.google.com"}

    for _, url := range urls {
        fmt.Println(url);
        go downloadFromUrl(url);
    }
}

问题是当我使用表达式go downloadFromUrl(url); 时，函数downloadFromUrl 没有运行。但如果我只是在循环中使用downloadFromUrl(url)，它就可以正常工作。我究竟做错了什么？我必须在例程中使用通道吗？

Answer 1

问题是在你的 goroutine 返回之前 main 退出，最简单的解决方案是使用 sync.WaitGroup。

func main() {
    urls := []string{"http://www.google.com", "http://www.yahoo.com", "http://www.google.com"}
    var wg sync.WaitGroup
    for _, url := range urls {
        wg.Add(1)
        log.Println(url)
        go downloadFromUrl(url, &wg)
    }
    wg.Wait()
}

func downloadFromUrl(url string, wg *sync.WaitGroup) {
    defer wg.Done()
    ....
}