golang 下载文件 带进度

Go是一种开源编程语言，由Google开发。它以其简洁、高效的设计和并发支持而闻名。Go语言旨在提供一种更好的开发体验，使软件开发变得更加简单、快速和可靠。在这篇文章中，我们将讨论如何使用Go语言下载文件并实现带进度的功能。

使用Go语言下载文件

在Go语言中，要下载文件需要使用http包提供的Get方法来发送HTTP请求。以下是一个简单的示例代码：

```go package main import ( "fmt" "io" "net/http" "os" ) func downloadFile(url string, filepath string) error { resp, err := http.Get(url) if err != nil { return err } defer resp.Body.Close() file, err := os.Create(filepath) if err != nil { return err } defer file.Close() _, err = io.Copy(file, resp.Body) if err != nil { return err } return nil } func main() { url := "https://example.com/file.jpg" filepath := "./file.jpg" err := downloadFile(url, filepath) if err != nil { fmt.Println("Error:", err) return } fmt.Println("File downloaded successfully!") } ```

以上代码定义了一个downloadFile函数，它接收文件的URL和保存路径作为参数。函数内部使用http.Get方法发送HTTP请求获取文件内容，并使用io.Copy方法将内容写入到本地文件中。

实现带进度的下载

在实际应用中，用户可能需要了解文件下载的进度以及剩余时间等信息。为了实现带进度的下载，我们可以使用http包提供的Transport和Client结构体，并结合io.TeeReader来实现。

首先，我们需要创建一个实现io.Writer接口的自定义结构体，并在Write方法中打印出进度信息：

```go type progressWriter struct { total int64 current int64 } func (pw *progressWriter) Write(p []byte) (int, error) { n := len(p) pw.current += int64(n) progress := float64(pw.current) / float64(pw.total) * 100 fmt.Printf("\rDownloading... %.2f%%", progress) return n, nil } ```

然后，我们修改downloadFile函数的实现，将下载文件时使用的io.Copy方法改为io.CopyBuffer，并传入自定义的progressWriter作为第三个参数：

```go func downloadFileWithProgress(url string, filepath string) error { client := http.Client{ Transport: &http.Transport{}, } resp, err := client.Get(url) if err != nil { return err } defer resp.Body.Close() file, err := os.Create(filepath) if err != nil { return err } defer file.Close() // 获取文件总大小 total := resp.ContentLength // 使用8KB缓冲区 buffer := make([]byte, 8*1024) // 初始化progressWriter pw := &progressWriter{total: total} // 使用io.CopyBuffer，并传入progressWriter _, err = io.CopyBuffer(io.MultiWriter(file, pw), resp.Body, buffer) if err != nil { return err } fmt.Println("\nFile downloaded successfully!") return nil } ```

使用示例

现在我们可以使用downloadFileWithProgress函数来下载文件并显示下载进度了。以下是一个简单的示例代码：

```go func main() { url := "https://example.com/file.jpg" filepath := "./file.jpg" err := downloadFileWithProgress(url, filepath) if err != nil { fmt.Println("Error:", err) return } } ```

通过以上代码我们就可以实现在终端中显示文件下载的进度了，用户可以清晰地了解到文件下载的进展情况。

总结起来，本文介绍了如何使用Go语言下载文件并实现带进度的功能。通过使用http包提供的Get方法发送HTTP请求，并结合io.CopyBuffer和自定义的progressWriter，我们可以轻松地实现带进度的下载。希望本文对你有所帮助，如果你有任何疑问或建议，请随时在下方留言。