go语言怎么实时显示下载进度

本示例说明如何从网上将文件下载到本地计算机。通过io.Copy()直接使用并传递响应主体,我们将数据流式传输到文件中,而不必将其全部加载到内存中-小文件不是问题,但下载大文件时会有所不同。

package mainimport (    "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则可以传递计数器来跟踪进度。在下载时,我们还将文件另存为临时文件,因此在完全下载文件之前,我们不会覆盖有效文件。

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 := "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}

原创文章,作者:XFQFV,如若转载,请注明出处:https://www.beidanyezhu.com/a/28904.html

(0)
XFQFV的头像XFQFV
上一篇 2025-02-01
下一篇 2025-02-01

相关推荐

  • Go语言与鸭子类型的关系是什么

    Go语言与鸭子类型的关系   先直接来看维基百科里的定义:   If it looks like a duck, swims like a duck, and quacks lik…

  • Go语言的重要性

    一、Go 开发者 数量 & 所处地区 全球大约有 110 万名职业 Go 开发者(特指在工作中专门将 Go 作为主力编程语言的群体),如果把主要使用其他编程语言但同时兼职使…

    2025-02-05
  • Go语言中CGO怎么用

    1. Go语言调用C函数例子: package main   // // 引用的C头文件需要在注释中声明,紧接着注释需要有import &q…

    2025-02-05
  • Go语言能否取代Linux脚本

      在Cloudflare的人们都非常喜欢Go语言。我们在许多内部软件项目以及更大的管道系统中使用它。但是,我们能否进入下一个层次并将其用作我们最喜欢的操作系统Linux的脚本语言…

  • 什么是Go语言的字符串

      一个Go语言字符串是一个任意字节的常量序列。 Go语言字符串与其他语言不同点   Go语言字符串与其他语言(Java,C,Python)字符串的不同点Go语言中字符串的字节使用…

  • 为什么Go语言能够成功

      常言道,历史不会重演,但总会惊人的相似。   如果您想创建一种编程语言,多向那些有经验的人士学习,他们有很多可取之处。在《GoTime》第100期节目中,两位Go语言的创造者R…

  • go语言中函数与方法是什么

      如果你遇到没有函数体的函数声明,表示该函数不是以Go实现的。   package math   func Sin(x float64) float //implemented …

  • GO语言的类型有哪些

    1、值的类型给编译器提供两部分信息:一是,需要分配多少内存给这个值(即值的规模);二是这段内存表示什么。 2、用户自定义类型有两种方法。一是使用关键字 struct ,来创建一个结…

    2025-02-05
  • 如何安装和使用Go语言集成开发环境的VS Code

    目录 Go语言集成开发环境之VS Code安装使用 下载与安装 安装中文简体插件 安装Go开发扩展 变更编辑器主题 安装Go语言开发工具包 配置VSCode开启自动保存 配置代码片…

    2025-02-05
  • Go语言的接口的介绍以及作用是什么

    接口就是一系列方法的集合(规范行为) 在面向对象的领域里,接口一般这样定义:接口定义一个对象的行为,规范子类对象的行为。 在 Go 语言中的接口是非侵入式接口(接口没了,不影响代码…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

分享本页
返回顶部