首页 >后端开发 >Golang >如何在 Go 中跟踪大文件的 POST 请求进度?

如何在 Go 中跟踪大文件的 POST 请求进度?

Mary-Kate Olsen
Mary-Kate Olsen原创
2024-10-28 14:57:301100浏览

How to Track POST Request Progress for Large Files in Go?

在 Go 中跟踪 POST 请求进度

问题:

旨在使用 POST 在 Go 中构建文件共享应用程序的开发人员请求在监控需要延长上传时间的大文件的上传进度方面面临挑战。

解决方案:

带有进度跟踪的包装 io.Reader:

另一种方法是围绕实际文件读取器创建自定义 io.Reader 包装器,而不是利用中间 TCP 连接。这样可以在每次调用 Read 函数时跟踪进度。

代码实现:

这是一个示例实现:

<code class="go">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
}</code>

使用这个包装器,将实际的文件读取器包装在 ProgressReader 中:

<code class="go">file, _ := os.Open("/tmp/blah.go")
total := int64(0)
pr := &ProgressReader{file, func(r int64) {
    total += r
    if r > 0 {
        fmt.Println("progress", r)
    } else {
        fmt.Println("done", r)
    }
}}</code>

随后,可以通过将内容复制到丢弃位置来跟踪进度:

<code class="go">io.Copy(ioutil.Discard, pr)</code>

此方法可以实现无缝使用 POST 请求监控大文件的上传进度。

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

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn