首頁  >  文章  >  後端開發  >  如何在 Go 中追蹤 POST 請求的上傳進度?

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

Mary-Kate Olsen
Mary-Kate Olsen原創
2024-10-30 22:11:03514瀏覽

 How to Track Upload Progress for POST Requests in Go?

在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中文網其他相關文章!

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