计算从 io.Reader 读取流式传输的字节数量

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《计算从 io.Reader 读取流式传输的字节数量》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

这是我的代码片段，它执行 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
}

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

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

正确答案

包裹阅​​读器。在包装器中计数。

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 包在从响应复制到 stdin 时使用缓冲区，因此 bufio.reader 不太可能提供任何好处。如果有一些好处，请使用这些选项中的一个；

reader := &wrapper{Reader: bufio.NewReader(resp.Body)}  // Option 1

cmd.Stdin = bufio.NewReader(reader)                     // Option 2

本篇关于《计算从 io.Reader 读取流式传输的字节数量》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！