迹忆客 专注技术分享

在本文中，我们将研究如何有效地使用 io.Reader(s)。 特别是我们将看到如何从 io.Reader 中多次读取。 例如，当读取或部分读取 HTTP 请求的响应主体时，或者只是需要多次从网络连接读取文件或数据时。 凭借 Go 标准库的高度可组合性，我们可以通过使用以下任一方法轻松实现这一点

在 Go (Golang) 中使用 io.TeeReader

io.TeeReader 包的灵感来自 tee unix 命令）。 tee unix 命令从标准输入读取并写入标准输出和一个或多个文件，有效地复制输入流。 这就是 TeeReader 在一定程度上所做的，让我们看看 Go 官方文档

func TeeReader(r Reader, w Writer) Reader

TeeReader 返回一个 Reader，它将从 r 读取的内容写入 w。 通过它执行的所有从 r 的读取都与对 w 的相应写入相匹配。 没有内部缓冲——写入必须在读取完成之前完成。 写入时遇到的任何错误都将报告为读取错误。

io.TeeReader 如何工作的例子

package main

import (
    "fmt"
    "io/ioutil"
    "io"
    "bytes"
    "strings"
)

func main() {
    reader := strings.NewReader("the quick brown fox jumps over the lazy dog")
    buf := &bytes.Buffer{}
    tee := io.TeeReader(reader, buf)
    reader1, _ := ioutil.ReadAll(tee)
    reader2, _ := ioutil.ReadAll(buf)
    fmt.Println(string(reader1))
    fmt.Println(string(reader2))
}

输出结果如下

the quick brown fox jumps over the lazy dog
the quick brown fox jumps over the lazy dog

注意 使用 io.TeeReader 时，请记住从输入阅读器读取所有数据，因为只要读取数据就会复制数据。 如果我们只读取输入流的一部分，TeeReader 将只将该部分复制到缓冲区中！

在 Go (Golang) 中使用 io.ReadSeeker

Go 中的 io.ReadSeeker 是 io.Reader 接口和 io.Seeker 接口的组合。 这允许我们通过传递相对于流的开始、当前位置或结束的偏移量（可以是文件流、字符串、网络连接或任何实现 一个 io.Reader）。 让我们先看看 io.Seeker Seek 方法的签名

type Seeker interface {
    Seek(offset int64, whence int) (int64, error)
}

Seek 将下一次Read或Write的偏移量设置为offset，根据whence解释：SeekStart表示相对于文件开头，SeekCurrent表示相对于当前偏移量，SeekEnd表示相对于结尾。 Seek 返回相对于文件开头的新偏移量和错误（如果有）。

让我们看一个示例，了解如何使用 io.Seeker 接口从同一个 io.Reader 读取多次

package main

import (
    "fmt"
    "io/ioutil"
    "io"
    "strings"
)

func main() {
    reader := strings.NewReader("the quick brown fox jumps over the lazy dog")
    reader1, _ := ioutil.ReadAll(reader)
    reader.Seek(0, io.SeekStart)
    reader2, _ := ioutil.ReadAll(reader)
    fmt.Println(string(reader1))
    fmt.Println(string(reader2))
}

注意 ：我们要读取和倒带的输入流或阅读器必须实现 io.Seeker 接口，例如 os.File 和 strings.Reader 确实实现了 io.Seeker 接口，但我们可能需要检查我们的特定案例

大家也可以在此处尝试 运行示例