如何在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 := &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中文網其他相關文章!