在 Go 中实现 POST 请求的上传进度跟踪
寻求为 Linux 创建 ShareX 克隆,通过 HTTP POST 无缝上传文件请求时,出现了一个关键问题:如何有效跟踪上传进度,特别是对于可能需要几分钟才能传输的较大文件?
虽然手动建立 TCP 连接并分块发送 HTTP 请求是一种可行的解决方案,但它携带处理 HTTPS 站点时存在局限性,并且可能不是最有效的方法。幸运的是,Go 提供了一种替代方案,可以轻松地将进度跟踪集成到请求流程中。
自定义 io.Reader 以进行进度监控
关键的见解在于创建自定义 io .Reader 包装了实际的阅读器。通过重写 Read() 方法,您可以捕获每次 Read 调用的进度。
考虑以下代码片段:
type ProgressReader struct { io.Reader Reporter func(r int64) } func (pr *ProgressReader) Read(p []byte) (n int, err error) { n, err = pr.Reader.Read(p) pr.Reporter(int64(n)) return }
在此结构中,Reporter是一个回调函数,接收读取的字节数作为参数。然后,ProgressReader 拦截 Read 操作并向回调报告进度。
实践中的实现
要将这个概念付诸实践,您可以采用以下步骤:
创建 ProgressReader 实例,包装文件读取器:
file, _ := os.Open("/tmp/blah.go") pr := &ProgressReader{file, func(r int64) { ... }}
定义要处理的 Reporter 函数进度更新:
pr.Reporter = func(r int64) { total += r if r > 0 { fmt.Println("progress", r) } else { fmt.Println("done", r) } }
使用自定义阅读器执行文件复制:
io.Copy(ioutil.Discard, pr)
通过使用这种方法,您可以毫不费力地监控 Go 中 POST 请求的上传进度,为用户提供文件传输过程的实时洞察。
以上是如何在 Go 中跟踪 POST 请求的上传进度?的详细内容。更多信息请关注PHP中文网其他相关文章!