go - 如何使用 Go 有效地下载大文件？

Question

有没有办法使用 Go 下载大文件，将内容直接存储到文件中，而不是在将内容写入文件之前将其全部存储在内存中？因为文件太大，所以在将其写入文件之前将其全部存储在内存中会耗尽所有内存。

Answer 1

我假设您的意思是通过 http 下载（为简洁起见，省略了错误检查）：

import ("net/http"; "io"; "os")
...
out, err := os.Create("output.txt")
defer out.Close()
...
resp, err := http.Get("http://example.com/")
defer resp.Body.Close()
...
n, err := io.Copy(out, resp.Body)

http.Response 的主体是一个 Reader，因此您可以使用任何带有 Reader 的函数，例如一次读取一个块而不是一次读取所有块。在这种特定情况下，io.Copy()会为您完成繁重的工作。

Answer 2

史蒂夫 M 答案的更具描述性的版本。

import (
    "os"
    "net/http"
    "io"
)

func downloadFile(filepath string, url string) (err error) {

  // Create the file
  out, err := os.Create(filepath)
  if err != nil  {
    return err
  }
  defer out.Close()

  // Get the data
  resp, err := http.Get(url)
  if err != nil {
    return err
  }
  defer resp.Body.Close()

  // Check server response
  if resp.StatusCode != http.StatusOK {
    return fmt.Errorf("bad status: %s", resp.Status)
  }

  // Writer the body to file
  _, err = io.Copy(out, resp.Body)
  if err != nil  {
    return err
  }

  return nil
}

Answer 3

上面使用的答案io.Copy正是您所需要的，但如果您对恢复中断下载、自动命名文件、校验和验证或监控多个下载的进度等附加功能感兴趣，请查看抓取包。

Answer 4

1. 这是一个示例。https://github.com/thbar/golang-playground/blob/master/download-files.go

2. 另外我给你一些代码可能会对你有所帮助。

代码：

func HTTPDownload(uri string) ([]byte, error) {
    fmt.Printf("HTTPDownload From: %s.\n", uri)
    res, err := http.Get(uri)
    if err != nil {
        log.Fatal(err)
    }
    defer res.Body.Close()
    d, err := ioutil.ReadAll(res.Body)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("ReadFile: Size of download: %d\n", len(d))
    return d, err
}

func WriteFile(dst string, d []byte) error {
    fmt.Printf("WriteFile: Size of download: %d\n", len(d))
    err := ioutil.WriteFile(dst, d, 0444)
    if err != nil {
        log.Fatal(err)
    }
    return err
}

func DownloadToFile(uri string, dst string) {
    fmt.Printf("DownloadToFile From: %s.\n", uri)
    if d, err := HTTPDownload(uri); err == nil {
        fmt.Printf("downloaded %s.\n", uri)
        if WriteFile(dst, d) == nil {
            fmt.Printf("saved %s as %s\n", uri, dst)
        }
    }
}