Golang文件上传进度显示教程

本文详解了如何在 Golang 中通过自定义 `io.Reader` 包装器（`ProgressReader`）实现文件上传过程中的实时进度监控，巧妙绕过 `net/http` 标准库不支持上传回调的限制——不是在网络层监听，而是从数据源头介入，在每次读取文件流时累计已传输字节数并触发回调，结合 `multipart` 表单编码无缝集成，既保持代码简洁、兼容标准用法，又支持终端实时百分比显示、UI 层解耦、超时控制与后续扩展，是构建用户友好、生产就绪文件上传功能的实用方案。

在使用 Golang 编写 HTTP 客户端上传文件时，原生的 net/http 包并不直接提供上传进度的回调机制。但通过自定义 io.Reader 包装器，我们可以实时监控数据写入过程，从而实现上传进度显示。

1. 核心思路：用 io.Reader 包装器监听写入过程

HTTP 请求体在发送时会从 Request.Body 中逐块读取数据。我们可以在真正发送数据前，将原始文件流包装成一个带有计数功能的 Reader，在每次读取时更新已发送字节数，进而计算进度。

关键点： 不是监听网络层，而是控制数据源的读取过程。

2. 实现带进度的 Reader

定义一个结构体，包裹原始文件，并在 Read 方法中增加进度回调：

package main

import (
    "fmt"
    "io"
    "mime/multipart"
    "net/http"
    "os"
    "sync"
)

// ProgressReader 记录已读字节数并触发回调
type ProgressReader struct {
    file     io.Reader
    total    int64
    read     int64
    callback func(read, total int64)
    mu       sync.Mutex
}

func (pr *ProgressReader) Read(p []byte) (n int, err error) {
    n, err = pr.file.Read(p)
    pr.mu.Lock()
    pr.read += int64(n)
    pr.callback(pr.read, pr.total)
    pr.mu.Unlock()
    return n, err
}

// NewProgressReader 创建带进度回调的 Reader
func NewProgressReader(file io.Reader, total int64, callback func(read, total int64)) *ProgressReader {
    return &ProgressReader{
        file:     file,
        total:    total,
        read:     0,
        callback: callback,
    }
}

3. 构建 multipart 请求并上传

使用 multipart 编码构造请求体，将 ProgressReader 插入到编码流程中：

func uploadFile(filepath, url string) error {
    file, err := os.Open(filepath)
    if err != nil {
        return err
    }
    defer file.Close()

    fileInfo, _ := file.Stat()
    totalSize := fileInfo.Size()

    body := &bytes.Buffer{}
    writer := multipart.NewWriter(body)

    // 添加文件字段
    part, err := writer.CreateFormFile("upload", filepath)
    if err != nil {
        return err
    }

    // 包装原始文件为带进度的 Reader
    progressReader := NewProgressReader(file, totalSize, func(read, total int64) {
        fmt.Printf("\r上传进度: %d / %d (%.2f%%)", read, total, float64(read)/float64(total)*100)
    })

    _, err = io.Copy(part, progressReader)
    if err != nil {
        return err
    }

    err = writer.Close()
    if err != nil {
        return err
    }

    // 发起请求
    req, err := http.NewRequest("POST", url, body)
    if err != nil {
        return err
    }
    req.Header.Set("Content-Type", writer.FormDataContentType())

    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        return err
    }
    defer resp.Body.Close()

    fmt.Printf("\n上传完成，服务器返回状态: %s\n", resp.Status)
    return nil
}

注意： 需要导入 bytes 包："bytes"

4. 使用示例

调用上传函数：

func main() {
    err := uploadFile("./test.pdf", "https://httpbin.org/post")
    if err != nil {
        fmt.Printf("上传失败: %v\n", err)
    }
}

运行后你会看到类似输出：

上传进度: 102400 / 512000 (20.00%)

5. 注意事项与优化建议

• 进度回调中避免执行耗时操作，防止阻塞上传
• 大文件上传可结合分块上传（chunked upload）提升稳定性
• 可通过 channel 将进度传递给 UI 层或前端
• 生产环境建议添加超时控制和断点续传逻辑

基本上就这些。通过封装 Reader 实现透明的进度追踪，既不影响标准库使用方式，又能灵活扩展功能。

今天关于《Golang文件上传进度显示教程》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！