通过自定义io.Reader实现上传进度监控,使用ProgressReader包装文件流并在Read方法中回调进度,结合multipart上传文件,实时显示传输百分比,不影响标准库用法且易于扩展。

在使用 Golang 编写 HTTP 客户端上传文件时,原生的 net/http 包并不直接提供上传进度的回调机制。但通过自定义 io.Reader 包装器,我们可以实时监控数据写入过程,从而实现上传进度显示。
1. 核心思路:用 io.Reader 包装器监听写入过程
HTTP 请求体在发送时会从 Request.Body 中逐块读取数据。我们可以在真正发送数据前,将原始文件流包装成一个带有计数功能的 Reader,在每次读取时更新已发送字节数,进而计算进度。
关键点: 不是监听网络层,而是控制数据源的读取过程。
2. 实现带进度的 Reader
定义一个结构体,包裹原始文件,并在 Read 方法中增加进度回调:
立即学习“go语言免费学习笔记(深入)”;
package mainimport ( "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 创建带进度回调的 Readerfunc 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上传完成,服务器返回状态: %sn", resp.Status) return nil}
注意: 需要导入 bytes 包:"bytes"
4. 使用示例
调用上传函数:
func main() { err := uploadFile("./test.pdf", "https://httpbin.org/post") if err != nil { fmt.Printf("上传失败: %vn", err) }}
运行后你会看到类似输出:
上传进度: 102400 / 512000 (20.00%)
5. 注意事项与优化建议
进度回调中避免执行耗时操作,防止阻塞上传大文件上传可结合分块上传(chunked upload)提升稳定性可通过 channel 将进度传递给 UI 层或前端生产环境建议添加超时控制和断点续传逻辑
基本上就这些。通过封装 Reader 实现透明的进度追踪,既不影响标准库使用方式,又能灵活扩展功能。
以上就是如何用 Golang 实现文件上传进度显示_Golang HTTP Client 文件传输示例的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1424013.html
微信扫一扫
支付宝扫一扫