如何在 Go 中跟踪 POST 请求的上传进度？

在 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 操作并向回调报告进度。

实践中的实现

要将这个概念付诸实践，您可以采用以下步骤：

1. 创建 ProgressReader 实例，包装文件读取器：

   file, _ := os.Open("/tmp/blah.go")
   pr := &ProgressReader{file, func(r int64) { ... }}

2. 定义要处理的 Reporter 函数进度更新：

   pr.Reporter = func(r int64) {
       total += r
       if r > 0 {
           fmt.Println("progress", r)
       } else {
           fmt.Println("done", r)
       }
   }

3. 使用自定义阅读器执行文件复制：

   io.Copy(ioutil.Discard, pr)

通过使用这种方法，您可以毫不费力地监控 Go 中 POST 请求的上传进度，为用户提供文件传输过程的实时洞察。

以上是如何在 Go 中跟踪 POST 请求的上传进度？的详细内容。更多信息请关注PHP中文网其他相关文章！