這是我執行 GET 請求并將響應流式傳輸到cmd.Stdin.resp, err = httpClient.Get(url)if err != nil { err = errors.Wrap(err, "HTTP request failed") return}reader = bufio.NewReader(resp.Body)args = append(args, "-") // Keep listening on stdin for file datacmd := exec.Command("exiftool", args...)stdout, err := cmd.StdoutPipe()if err != nil { return}cmd.Stdin = readererr = cmd.Start()if err != nil { return}我想知道在它完成執行時流式傳輸了多少數據。所以我需要的是捕獲流式傳輸時正在讀取的內容，或者至少捕獲正在讀取的內容的大小。
1 回答
拉莫斯之舞
TA貢獻1820條經驗 獲得超10個贊
包裹讀者。算在包裝紙里。
type wrapper struct {
io.Reader
n int
}
func (w *wrapper) Read(p []byte) (int, error) {
n, err := w.Reader.Read(p)
w.n += n
return n, err
}
像這樣將它插入您的應用程序：
args = append(args, "-")
cmd := exec.Command("exiftool", args...)
stdout, err := cmd.StdoutPipe()
if err != nil {
return
}
reader := &wrapper{Reader: resp.Body}
cmd.Stdin = reader
err = cmd.Run()
if err != nil {
return
}
fmt.Println(reader.n) // prints number of bytes read.
因為 exec 包在從響應復制到標準輸入時使用緩沖區，所以 bufio.Reader 不太可能提供好處。如果有一些好處，請使用這些選項之一；
reader := &wrapper{Reader: bufio.NewReader(resp.Body)} // Option 1
cmd.Stdin = bufio.NewReader(reader) // Option 2
- 1 回答
- 0 關注
- 89 瀏覽
添加回答
舉報
0/150
提交
取消