golang实现带进度条大文件下载

lu569368 · · 1060 次点击 · 开始浏览    置顶
这是一个创建于 的主题,其中的信息可能已经有所发展或是发生改变。

本示例说明如何从网上将文件下载到本地计算机。通过io.Copy()直接使用并传递响应主体,我们将数据流式传输到文件中,而不必将其全部加载到内存中-小文件不是问题,但下载大文件时会有所不同。 **本文转自:[http://www.topgoer.com](http://www.topgoer.com/?fr=studygolang "http://www.topgoer.com")** ```go package main import ( "io" "net/http" "os" ) func main() { fileUrl := "http://topgoer.com/static/2/9.png" if err := DownloadFile("9.png", fileUrl); err != nil { panic(err) } } // download file会将url下载到本地文件,它会在下载时写入,而不是将整个文件加载到内存中。 func DownloadFile(filepath string, url string) error { // Get the data resp, err := http.Get(url) if err != nil { return err } defer resp.Body.Close() // Create the file out, err := os.Create(filepath) if err != nil { return err } defer out.Close() // Write the body to file _, err = io.Copy(out, resp.Body) return err } ``` ### 带进度条的大文件下载 下面的示例是带有进度条的大文件下载,我们将响应主体传递到其中,io.Copy()但是如果使用a,TeeReader则可以传递计数器来跟踪进度。在下载时,我们还将文件另存为临时文件,因此在完全下载文件之前,我们不会覆盖有效文件。 ```go package main import ( "fmt" "io" "net/http" "os" "strings" "github.com/dustin/go-humanize" ) type WriteCounter struct { Total uint64 } func (wc *WriteCounter) Write(p []byte) (int, error) { n := len(p) wc.Total += uint64(n) wc.PrintProgress() return n, nil } func (wc WriteCounter) PrintProgress() { fmt.Printf("\r%s", strings.Repeat(" ", 35)) fmt.Printf("\rDownloading... %s complete", humanize.Bytes(wc.Total)) } func main() { fmt.Println("Download Started") fileUrl := "http://topgoer.com/static/2/9.png" err := DownloadFile("9.png", fileUrl) if err != nil { panic(err) } fmt.Println("Download Finished") } func DownloadFile(filepath string, url string) error { out, err := os.Create(filepath + ".tmp") if err != nil { return err } resp, err := http.Get(url) if err != nil { out.Close() return err } defer resp.Body.Close() counter := &WriteCounter{} if _, err = io.Copy(out, io.TeeReader(resp.Body, counter)); err != nil { out.Close() return err } fmt.Print("\n") out.Close() if err = os.Rename(filepath+".tmp", filepath); err != nil { return err } return nil } ```

有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1060 次点击  ∙  1 赞  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传