Golang io.TeeReader() 函数的示例

在 Golang 中，我们可以使用 io.TeeReader() 函数将一个 io.Reader 对象复制到两个不同的输出流中，这个函数非常适用于需要将一组数据同时写到两个地方的场景。本文将详细讲解 TeeReader() 的使用方法，并提供一些示例代码。

io.TeeReader() 函数介绍

io.TeeReader() 函数的定义如下：

func TeeReader(r io.Reader, w io.Writer) io.Reader

TeeReader() 函数接收两个参数：一个 io.Reader 对象和一个 io.Writer 对象。在将数据从 io.Reader 对象中读出的同时，TeeReader() 会将这些数据写入到给定的 io.Writer 对象中。这个函数的返回值是一个新的 io.Reader 对象，它会将数据直接传递给调用者。需要注意的是，TeeReader() 函数不会关闭给定的 io.Reader 对象和 io.Writer 对象。

TeeReader() 函数示例

下面我们来看一个例子，假设我们需要从一个文本文件中读取 100 字节的数据，并将它们同时写入到标准输出和一个文件中。

package main

import (
    "io"
    "os"
)

func main() {
    file, _ := os.Create("output.txt")
    defer file.Close()

    // 创建一个输入流，从当前目录下的 example.txt 文件中读取 100 字节数据
    input, _ := os.Open("example.txt")
    defer input.Close()

    // 创建两个输出流：标准输出和文件输出
    outputA := io.MultiWriter(os.Stdout, file)

    // 创建一个 io.TeeReader 对象，将数据从输入流中读出，并同时写入 outputA 中
    reader := io.TeeReader(input, outputA)

    // 从 reader 中读取数据（此时，已经将数据写入到了 outputA 中）
    data := make([]byte, 100)
    reader.Read(data)

    return
}

在上述代码中，我们首先创建了两个输出流：标准输出和文件输出。然后，我们使用 TeeReader() 函数将数据从输入流 input 中读出，并同时写入 outputA 中。最后，我们从 reader 中读取数据（此时，已经将数据写入到了 outputA 中），并将其写入到一个 100 字节的缓冲区中。