获取从 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 datacmd := exec.Command("exiftool", args...)stdout, err := cmd.StdoutPipe()if err != nil {    return}cmd.Stdin = readererr = 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 = 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