从 Golang 的缓冲读取器中读取指定数量的字节

从 Golang 的缓冲读取器中读取指定数量的字节

正如摘要所述，bufio.Reader 提供了 Peek 方法，允许查看接下来的 n 个字节，但不会移动读取指针。然而，在某些场景下，我们需要读取指定数量的字节并推进读取器。本文将介绍如何使用 io.ReadFull 函数来实现这一目标，即使底层读取器是带缓冲的。

io.ReadFull 函数的作用是从 io.Reader 中读取指定长度的数据。如果读取的字节数小于指定的长度，且没有遇到 EOF，则 io.ReadFull 会返回一个错误。这确保了要么成功读取指定长度的数据，要么返回错误，避免了只读取部分数据的情况。

以下是使用 io.ReadFull 从 bufio.Reader 读取指定数量字节的示例代码：

package main
import (
"bufio"
"fmt"
"io"
"strings"
)
func main() {
// 创建一个字符串读取器
reader := strings.NewReader("Hello, world!")
// 创建一个带缓冲的读取器
bufReader := bufio.NewReader(reader)
// 要读取的字节数
numBytes := 5
// 创建一个字节切片来存储读取的数据
p := make([]byte, numBytes)
// 使用 io.ReadFull 读取指定数量的字节
n, err := io.ReadFull(bufReader, p)
// 处理错误
if err != nil {
fmt.Println("Error reading:", err)
return
}
// 打印读取的字节数和数据
fmt.Println("Read", n, "bytes:", string(p))
// 验证读取器是否已前进
remaining, _ := bufReader.Peek(5)
fmt.Println("Remaining:", string(remaining))
}

代码解释：

立即学习“go语言免费学习笔记（深入）”；

1. 创建读取器: 首先，我们使用 strings.NewReader 创建一个字符串读取器，并将其包装在 bufio.NewReader 中，创建一个带缓冲的读取器。
2. 指定读取字节数: numBytes 变量定义了我们想要读取的字节数。
3. 创建缓冲区: p 是一个字节切片，用于存储从读取器读取的数据。
4. 使用 io.ReadFull: io.ReadFull(bufReader, p) 尝试从 bufReader 读取 len(p) 个字节到 p 中。
5. 错误处理: 检查 err 是否为 nil。如果不是，则表示读取过程中发生了错误，例如 io.EOF 或其他 I/O 错误。
6. 输出结果: 如果读取成功，打印读取的字节数和读取到的字符串。
7. 验证读取器前进: 使用 bufReader.Peek 方法查看剩余的数据，验证读取器是否已经前进。

注意事项：

• io.ReadFull 会阻塞，直到读取了 len(p) 个字节或遇到错误。
• 如果底层 io.Reader 返回的字节数小于 len(p) 并且没有遇到错误，io.ReadFull 会返回 io.ErrUnexpectedEOF 错误。
• 确保提供的缓冲区 p 的大小足够存储预期读取的字节数。

总结：

io.ReadFull 提供了一种可靠的方式，从带缓冲的读取器中读取指定数量的字节并推进读取器。通过使用 io.ReadFull，我们可以确保要么成功读取指定长度的数据，要么得到一个错误，从而避免了只读取部分数据的情况。在需要精确控制读取字节数的场景下，io.ReadFull 是一个非常有用的工具。