使用 Goroutines 在 Golang 中并行化文件下载
问题:
我们可以利用 Goroutines同时下载多个文件?
代码上下文:
<code class="go">package main import ( "fmt" "io" "io/ioutil" "net/http" "net/url" "os" "path/filepath" "sync" ) // ... (existing Dropbox access token handling code) var wg sync.WaitGroup func downloadFile(file File, token TokenResponse) { // Acquire WaitGroup counter wg.Add(1) defer wg.Done() // Release counter when function returns downloadURL := fmt.Sprintf("https://api-content.dropbox.com/1/files/dropbox/%s?access_token=%s", file.Path, token.AccessToken) resp, err := http.Get(downloadURL) if err != nil { panic(err) } defer resp.Body.Close() filename := filepath.Base(file.Path) outFile, err := os.Create(filename) if err != nil { panic(err) } defer outFile.Close() io.Copy(outFile, resp.Body) } func main() { // ... (existing Dropbox authorization and file list code) // Spawn goroutines for file downloads for _, file := range flr.FileList { go downloadFile(file, tr) if count >= 2 { break } } // Wait for all goroutines to complete before exiting wg.Wait() }</code>
此修改后的代码使用sync.WaitGroup来确保主goroutine不会退出,直到所有文件下载完成。这允许 goroutine 并行下载文件,从而提高性能。
以上是## Goroutines 可以加快 Golang 中的文件下载速度吗?的详细内容。更多信息请关注PHP中文网其他相关文章!