获取从 io.Reader Read 流式传输的字节数

Question

这是我执行 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 data

cmd := exec.Command("exiftool", args...)

stdout, err := cmd.StdoutPipe()

if err != nil {

    return

}

cmd.Stdin = reader

err = cmd.Start()

if err != nil {

    return

}

我想知道在它完成执行时流式传输了多少数据。

所以我需要的是捕获流式传输时正在读取的内容，或者至少捕获正在读取的内容的大小。

Answer 1

包裹读者。算在包装纸里。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 = readererr = 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 1cmd.Stdin = bufio.NewReader(reader)                     // Option 2