48

如何将以下代码转换为使用流/管道,这样我就不需要将全部内容读入内存?就像是: http.Get("http://example.com/").Pipe("./data.txt")

package main
import ("net/http";"io/ioutil")

func main() {
        resp, err := http.Get("http://example.com/")
        check(err)
        defer resp.Body.Close()
        body, err := ioutil.ReadAll(resp.Body)
        check(err)
        err = ioutil.WriteFile("./data.txt", body, 0666)
        check(err)
}
func check(e error) {
        if e != nil {
                panic(e)
        }
}
4

2 回答 2

73

怎么样io.Copy()?可以在以下位置找到它的文档:http: //golang.org/pkg/io/#Copy

不过,这很简单。给它一个io.Reader和一个io.Writer,它会复制数据,一次一小块(例如,不是一次全部在内存中)。

因此,您可以尝试编写如下内容:

func main() {
  resp, err := http.Get("...")
  check(err)
  defer resp.Body.Close()
  out, err := os.Create("filename.ext")
  if err != nil {
    // panic?
  }
  defer out.Close()
  io.Copy(out, resp.Body)
}

我还没有测试过上述内容;我只是从您上面的示例中快速将其破解,但如果不赚钱,它应该很接近。

于 2013-05-01T01:13:04.537 回答
3

另一种选择是File.ReadFrom

package main

import (
   "net/http"
   "os"
)

func main() {
   r, e := http.Get("http://speedtest.lax.hivelocity.net")
   if e != nil {
      panic(e)
   }
   defer r.Body.Close()
   f, e := os.Create("index.html")
   if e != nil {
      panic(e)
   }
   defer f.Close()
   f.ReadFrom(r.Body)
}
于 2020-12-17T23:30:31.003 回答