首頁 >後端開發 >Golang >如何追蹤 Go HTTP POST 請求中的上傳進度:使用 ProgressReader 的實用指南?

如何追蹤 Go HTTP POST 請求中的上傳進度:使用 ProgressReader 的實用指南?

Patricia Arquette
Patricia Arquette原創
2024-10-29 17:39:02946瀏覽

How to Track Upload Progress in Go HTTP POST Requests: A Practical Guide Using ProgressReader?

如何在Go HTTP POST 請求中追蹤上傳進度

在使用Go 開發適用於Linux 的ShareX 複製時,需要監控透過透過HTTP POST 請求發送的大檔案上傳進度。雖然手動開啟 TCP 連線並以區塊的形式寫入 HTTP 請求是一種選擇,但這並不是最高效、最可靠的解決方案。

ProgressReader:自訂 io.Reader

更有效的方法是建立一個自訂 io.Reader 來包圍實際的閱讀器。這個自訂閱讀器名為 ProgressReader,有一個 Reporter 函數,每次呼叫 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>

然後使用此自訂讀取器包裝正在上傳的檔案:

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

最後,使用io.Copy 將資料複製到ioutil.Discard,後者充當資料的接收器同時仍允許列印進度輸出。

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

該方法提供了一種簡單可靠的方法來追蹤 Go 中 HTTP POST 請求的上傳進度,使用戶能夠即時監控大檔案上傳的進度-時間。

以上是如何追蹤 Go HTTP POST 請求中的上傳進度:使用 ProgressReader 的實用指南?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn