Go语言：使用io.Reader包装器实时监控文件下载进度

package main

import (
    "fmt"
    "io"
    "net/http"
    "os"
)

// ProgressReader 结构体包装了一个 io.Reader，并跟踪读取进度。
type ProgressReader struct {
    io.Reader
    total int64 // 已读取的总字节数
    size  int64 // 文件总大小 (可选，用于计算百分比)
}

// NewProgressReader 创建一个新的 ProgressReader 实例。
func NewProgressReader(reader io.Reader, totalSize int64) *ProgressReader {
    return &ProgressReader{
        Reader: reader,
        size:   totalSize,
    }
}

// Read 方法实现了 io.Reader 接口，并在每次读取时更新并打印进度。
func (pr *ProgressReader) Read(p []byte) (int, error) {
    n, err := pr.Reader.Read(p) // 调用底层 Reader 的 Read 方法
    pr.total += int64(n)        // 累加已读取的字节数

    // 实时打印进度
    if err == nil { // 只有在没有错误时才更新和打印进度
        if pr.size > 0 {
            // 如果已知总大小，可以计算并打印百分比
            progress := float64(pr.total) / float64(pr.size) * 100
            // 使用 \r 回到行首，实现同一行刷新输出，模拟进度条效果
            fmt.Printf("\rDownloading: %d/%d bytes (%.2f%%)", pr.total, pr.size, progress)
        } else {
            // 如果不知道总大小，只打印已读取的字节数
            fmt.Printf("\rDownloading: %d bytes...", pr.total)
        }
    }
    return n, err
}

func main() {
    // 替换为实际的文件下载URL，例如一个测试文件
    downloadURL := "http://ipv4.download.thinkbroadband.com/5MB.zip" // 示例：一个5MB的测试文件

    fmt.Printf("开始下载文件: %s\n", downloadURL)

    resp, err := http.Get(downloadURL)
    if err != nil {
        fmt.Printf("下载请求失败: %v\n", err)
        os.Exit(1)
    }
    defer resp.Body.Close() // 确保关闭响应体

    if resp.StatusCode != http.StatusOK {
        fmt.Printf("服务器返回错误状态码: %d %s\n", resp.StatusCode, resp.Status)
        os.Exit(1)
    }

    // 尝试从HTTP响应头中获取文件总大小 (Content-Length)
    var totalSize int64
    if resp.ContentLength > 0 {
        totalSize = resp.ContentLength
        fmt.Printf("文件总大小: %d bytes\n", totalSize)
    } else {
        fmt.Println("无法获取文件总大小，将只显示已下载字节数。")
    }

    // 创建一个本地文件用于保存下载内容
    outputFile, err := os.Create("downloaded_file.zip")
    if err != nil {
        fmt.Printf("创建输出文件失败: %v\n", err)
        os.Exit(1)
    }
    defer outputFile.Close() // 确保关闭输出文件

    // 使用 ProgressReader 包装 resp.Body，实现进度跟踪
    progressReader := NewProgressReader(resp.Body, totalSize)

    // 使用 io.Copy 将带有进度跟踪的 Reader 内容写入本地文件
    // io.Copy 会持续调用 progressReader 的 Read 方法
    bytesWritten, err := io.Copy(outputFile, progressReader)
    if err != nil {
        fmt.Printf("\n文件写入失败: %v\n", err)
        os.Exit(1)
    }

    fmt.Printf("\n下载完成！共写入 %d 字节到 %s\n", bytesWritten, outputFile.Name())
}

开始下载文件: http://ipv4.download.thinkbroadband.com/5MB.zip
文件总大小: 5242880 bytes
Downloading: 5242880/5242880 bytes (100.00%)
下载完成！共写入 5242880 字节到 downloaded_file.zip