Golang 中的 Bufio 包详解之 Bufio.Reader

开发运维 2023-09-07 共饮一杯手机阅读

bufio.Reader

bufio.Reader 是一个带有缓冲区的 io.Reader 接口的实现，提供了一系列方法来帮助读取数据。使用 bufio.Reader 可以减少 I/O 操作，降低读取数据的时间和资源开销。主要特征是它会在内存中存储从底层 io.Reader 中读取到的数据，然后先从内存缓冲区中读取数据，这样可以减少访问底层 io.Reader 对象的次数以及减轻操作系统的压力。结构体定义和对应的方法如下：

type Reader struct {
    buf          []byte
    rd           io.Reader // reader provided by the client
    r, w         int       // buf read and write positions
    err          error
    lastByte     int // last byte read for UnreadByte; -1 means invalid
    lastRuneSize int // size of last rune read for UnreadRune; -1 means invalid
}

下面是 bufio.Reader 的一些主要方法:

func (b *Reader) Read(p []byte) (n int, err error)：从缓冲区中读取数据到 p 中，返回读取的字节数和可能的读取错误。如果 p 的长度大于缓冲区的大小，则会触发缓冲区的扩容操作。
func (b *Reader) ReadByte() (byte, error)：从缓冲区中读取一个字节，并返回该字节和可能发生的错误信息。
func (b *Reader) ReadRune() (r rune, size int, err error)：从缓冲区中读取一个 UTF-8 编码的字符，返回该字符和可能发生的错误。如果缓冲区中没有足够的字节来表示一个完整的 UTF-8 字符，则返回一个错误。
func (b *Reader) ReadLine() (line []byte, isPrefix bool, err error)：从缓冲区中读取一行，并返回该行内容和可能发生的错误。

其他方法就不一一说明了，最好自己去看去使用去体会。

优势

bufio.Reader 提供了带缓冲的读取操作，先在内存中存储通过系统调用读取到的数据，然后从内存缓冲区中读取数据，大大减少了系统调用次数，减轻了操作系统的压力，加快了数据读取速度。

bufio.Reader 提供了很多类型的读取方法，例如 ReadByte()、 ReadRune() 和 ReadLine() 等，使用起来非常方便。

使用示例

简单使用示例如下：

package main

import (
    "bufio"
    "fmt"
    "os"
)

func main() {
    file, err := os.Open("file.txt")
    if err != nil {
        panic(err)
    }
    defer file.Close()

    reader := bufio.NewReader(file)
    buffer := make([]byte, 4)

    for {
        n, err := reader.Read(buffer)
        if err != nil {
            break
        }
        fmt.Print(string(buffer[:n]))
    }
}

使用 NewReader() 方法创建一个 bufio.Reader 实例，然后创建了一个缓冲区 buffer，并在一个循环中使用 Read() 方法从缓冲区中读取数据。