从 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语言免费学习笔记（深入）”；

百宝箱

百宝箱是支付宝推出的一站式AI原生应用开发平台，无需任何代码基础，只需三步即可完成AI应用的创建与发布。

313

查看详情

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 是一个非常有用的工具。