【go语言】3.4.2 使用 io 和 bufio 包进行流操作

101 阅读2分钟

在 Go 中,流操作指的是对数据流的读写,这对于大文件或网络数据传输来说特别重要,因为这些场景下数据不能一次性全部加载到内存中。io 和 bufio 包提供了一系列工具来高效地处理这些场景。

io.Reader 和 io.Writer

在 Go 中,所有的流操作都可以抽象为 io.Reader 和 io.Writer 接口。io.Reader 接口表示数据的来源,可以从中读取数据;io.Writer 接口表示数据的目的地,可以向其中写入数据。

下面是一个 io.Reader 的例子,它从一个字符串读取数据:

package main

import (
	"fmt"
	"io"
	"strings"
)

func main() {
	reader := strings.NewReader("Hello, World!")
	data := make([]byte, 5)
	n, err := reader.Read(data)
	if err != nil {
		fmt.Println("Error:", err)
		return
	}
	fmt.Printf("Read %d bytes: %s\n", n, string(data))
}

在这个例子中,strings.NewReader 返回一个 io.Reader,我们可以从中读取数据。Read 方法接收一个字节切片,将数据读取到这个切片中,然后返回读取的字节数和可能的错误。

bufio.Scanner 和 bufio.Writer

bufio 包提供了缓冲读写功能,这可以提高读写效率,特别是对于小数据块的频繁读写操作。bufio.Scanner 可以方便地按行读取数据,而 bufio.Writer 可以缓冲多次的写入操作,然后在必要时一次性写入到目标中。

下面是一个 bufio.Writer 的例子,它将多次的写入操作缓冲到一个字节切片中:

package main

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

func main() {
	writer := bufio.NewWriter(os.Stdout)
	writer.WriteString("Hello, ")
	writer.WriteString("World!")
	writer.Flush()
}

在这个例子中,bufio.NewWriter 返回一个 bufio.Writer,我们可以向其中写入数据。WriteString 方法将一个字符串写入到缓冲区中,然后 Flush 方法将缓冲区中的数据写入到目标中。

这就是 Go 中的流操作。虽然这些操作在开始时可能会让人感到困惑,但一旦你理解了这些基本概念,你就可以处理各种复杂的数据读写问题了。 推荐阅读:

mp.weixin.qq.com/s/dV2JzXfgj…

mp.weixin.qq.com/s/an83QZOWX…

file