前言:
眼前各位老铁们对“golang 文件下载”大致比较看重,各位老铁们都需要分析一些“golang 文件下载”的相关资讯。那么小编也在网摘上汇集了一些有关“golang 文件下载””的相关内容,希望朋友们能喜欢,姐妹们一起来学习一下吧!普通文件下载
package mainimport ("io""net/http""os")func main() { // 下载地址: fileUrl := "; if err := DownloadFile("保存文件名.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}
带进度条的大文件下载
package mainimport ( "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 := "文件名称"" err := DownloadFile("保存文件名.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{} /* io.TeeReader则可以传递计数器来跟踪进度 io.TeeReader返回一个将其从r读取的数据写入w的Reader接口。所有通过该接口对r的读取都会执行对应的对w的写入。没有内部的缓冲; 写入必须在读取完成前完成。写入时遇到的任何错误都会作为读取错误返回。 */ 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}
版权声明:
本站文章均来自互联网搜集,如有侵犯您的权益,请联系我们删除,谢谢。
标签: #golang 文件下载