Go语言中实现文件下载进度实时监控：自定义io.Reader的实践

package main

import (
    "bytes"
    "fmt"
    "io"
    "os"
    "strings"
    "net/http" // 引入http包用于实际下载示例
)

// PassThru 包装了一个io.Reader，用于在读取数据时进行计数和打印进度。
type PassThru struct {
    io.Reader
    total int64 // 已传输的总字节数
}

// Read '覆盖'了底层io.Reader的Read方法。
// io.Copy会调用此方法。我们利用它来跟踪字节计数，然后转发调用。
func (pt *PassThru) Read(p []byte) (int, error) {
    n, err := pt.Reader.Read(p)
    pt.total += int64(n)

    // 仅在没有错误发生时打印进度，避免打印过多无关信息
    if err == nil {
        fmt.Printf("\r已读取 %d 字节，累计总数: %d 字节", n, pt.total)
    } else if err == io.EOF {
        fmt.Println("\n文件下载完成。") // 文件结束时打印最终状态
    }

    return n, err
}

func main() {
    // 1. 定义目标文件路径
    filePath := "downloaded_file.zip" // 假设下载一个zip文件

    // 2. 创建本地文件用于写入下载内容
    out, err := os.Create(filePath)
    if err != nil {
        fmt.Printf("创建文件失败: %v\n", err)
        return
    }
    defer out.Close() // 确保文件句柄在函数结束时关闭

    // 3. 发起HTTP GET请求下载文件
    // 替换为实际可下载的文件URL
    downloadURL := "http://example.com/some_large_file.zip" // 示例URL，请替换为真实可下载文件
    fmt.Printf("开始从 %s 下载文件...\n", downloadURL)
    resp, err := http.Get(downloadURL)
    if err != nil {
        fmt.Printf("发起HTTP请求失败: %v\n", err)
        return
    }
    defer resp.Body.Close() // 确保HTTP响应体在函数结束时关闭

    // 检查HTTP响应状态码
    if resp.StatusCode != http.StatusOK {
        fmt.Printf("下载失败，HTTP状态码: %d %s\n", resp.StatusCode, resp.Status)
        return
    }

    // 4. 包装resp.Body，实现进度监控
    // resp.Body本身是一个io.Reader，我们用PassThru来包装它
    readerWithProgress := &PassThru{Reader: resp.Body}

    // 5. 使用io.Copy将带进度的Reader内容复制到本地文件
    // io.Copy会调用readerWithProgress的Read方法，从而触发进度打印
    bytesTransferred, err := io.Copy(out, readerWithProgress)
    if err != nil {
        fmt.Printf("文件复制过程中发生错误: %v\n", err)
        return
    }

    fmt.Printf("下载完成！总共传输了 %d 字节到 %s\n", bytesTransferred, filePath)

    // -----------------------------------------------------------------------
    // 额外示例：使用内存中的数据流进行进度监控（与原始答案类似）
    fmt.Println("\n--- 内存数据流进度监控示例 ---")
    var src io.Reader    // 源数据
    var dst bytes.Buffer // 目标缓冲区

    // 创建一些随机输入数据作为源
    src = bytes.NewBufferString(strings.Repeat("Some random input data for demonstration. ", 100))

    // 包装它与我们的自定义io.Reader。
    src = &PassThru{Reader: src}

    count, err := io.Copy(&dst, src)
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }

    fmt.Printf("\n内存数据流传输完成，总共传输了 %d 字节\n", count)
}

开始从 http://example.com/some_large_file.zip 下载文件...
已读取 512 字节，累计总数: 512 字节
已读取 1024 字节，累计总数: 1536 字节
已读取 2048 字节，累计总数: 3584 字节
...
已读取 6128 字节，累计总数: 22000 字节
文件下载完成。
下载完成！总共传输了 22000 字节到 downloaded_file.zip

--- 内存数据流进度监控示例 ---
已读取 512 字节，累计总数: 512 字节
已读取 1024 字节，累计总数: 1536 字节
已读取 2048 字节，累计总数: 3584 字节
已读取 4096 字节，累计总数: 7680 字节
已读取 8192 字节，累计总数: 15872 字节
已读取 6128 字节，累计总数: 22000 字节
内存数据流传输完成，总共传输了 22000 字节