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

package main

import (
    "fmt"
    "io"
    "os"
    "net/http"
    "time" // 用于模拟实际下载，增加延迟
)

// PassThru 结构体包装了一个 io.Reader，并记录已传输的总字节数。
type PassThru struct {
    io.Reader
    total int64 // 已传输的总字节数
    // 可选：添加一个名称或ID，用于区分不同的传输任务
    // name string 
}

// Read 方法是 io.Reader 接口的实现。
// 它会调用底层 Reader 的 Read 方法，并更新已传输的字节数。
func (pt *PassThru) Read(p []byte) (int, error) {
    n, err := pt.Reader.Read(p) // 调用底层 Reader 的 Read 方法
    pt.total += int64(n)        // 更新总字节数

    if err == nil {
        // 实时打印进度，可以根据需要进行格式化
        // 例如，如果知道总大小，可以计算百分比
        fmt.Printf("\r已读取 %d 字节，总计 %d 字节...", n, pt.total)
        // 为了不频繁刷新，可以考虑只在达到特定阈值或时间间隔时打印
    }
    return n, err
}

// ... (PassThru 结构体定义) ...

import (
    "bytes"
    "strings"
)

func main() {
    // 模拟源数据
    srcData := strings.Repeat("这是一段模拟的输入数据。", 500) // 约 15KB
    src := bytes.NewBufferString(srcData)

    // 包装源数据 Reader
    passThruReader := &PassThru{Reader: src}

    var dst bytes.Buffer // 目标缓冲区

    fmt.Println("开始模拟数据传输...")
    count, err := io.Copy(&dst, passThruReader)
    if err != nil {
        fmt.Println("\n传输错误:", err)
        os.Exit(1)
    }

    fmt.Printf("\n数据传输完成。总计传输 %d 字节。\n", count)
    // fmt.Println("目标数据内容长度:", dst.Len()) // 验证传输完整性
}

开始模拟数据传输...
已读取 512 字节，总计 512 字节...
已读取 1024 字节，总计 1536 字节...
...
已读取 6128 字节，总计 22000 字节...
数据传输完成。总计传输 22000 字节。

// ... (PassThru 结构体定义) ...

func main() {
    fileURL := "https://speed.hetzner.de/100MB.bin" // 示例下载URL，请替换为实际可用的URL
    outputPath := "downloaded_file.bin"

    // 1. 创建输出文件
    outFile, err := os.Create(outputPath)
    if err != nil {
        fmt.Printf("创建文件失败: %v\n", err)
        return
    }
    defer outFile.Close()

    // 2. 发起HTTP GET请求
    fmt.Printf("开始下载文件: %s 到 %s\n", fileURL, outputPath)
    resp, err := http.Get(fileURL)
    if err != nil {
        fmt.Printf("HTTP GET请求失败: %v\n", err)
        return
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        fmt.Printf("服务器返回非成功状态码: %s\n", resp.Status)
        return
    }

    // 获取文件总大小 (如果服务器提供 Content-Length 头)
    totalSize := resp.ContentLength
    if totalSize > 0 {
        fmt.Printf("文件总大小: %d 字节 (%.2f MB)\n", totalSize, float64(totalSize)/(1024*1024))
    } else {
        fmt.Println("无法获取文件总大小，进度将只显示已下载量。")
    }

    // 3. 包装响应体 (resp.Body 是一个 io.Reader)
    passThruReader := &PassThru{Reader: resp.Body}

    // 4. 使用 io.Copy 将数据从包装器复制到文件
    // io.Copy 会不断调用 passThruReader.Read()
    bytesCopied, err := io.Copy(outFile, passThruReader)
    if err != nil {
        fmt.Printf("\n文件下载失败: %v\n", err)
        return
    }

    fmt.Printf("\n文件下载完成。总计传输 %d 字节。\n", bytesCopied)
    if totalSize > 0 && bytesCopied != totalSize {
        fmt.Println("警告：下载的字节数与Content-Length不匹配！")
    }
}