Golang如何使用bufio高效读写文件_Golang bufio缓冲读写操作方法

使用bufio包可提升Go语言文件读写性能。通过bufio.Reader减少系统调用,实现高效按行或字节读取;利用bufio.Writer缓冲数据,降低磁盘I/O次数,需调用Flush确保数据写入;组合Reader和Writer可高效完成文件复制,控制内存使用且提升速度。

在Go语言中,bufio 包为文件的读写操作提供了带缓冲的功能,能显著提升I/O性能。相比直接使用 os.File 的无缓冲读写,bufio.Readerbufio.Writer 通过减少系统调用次数来提高效率,特别适合处理大文件或频繁的小数据块操作。

使用 bufio.Reader 缓冲读取文件

当从文件中读取数据时,使用 bufio.Reader 可以避免每次读取都触发系统调用。它内部维护一个缓冲区,一次性从磁盘读取较多数据,后续的读操作优先从缓冲区获取。

常见用法包括按行读取、按字节读取等:

file, err := os.Open("example.txt")
if err != nil {
    log.Fatal(err)
}
defer file.Close()

reader := bufio.NewReader(file)
for {
    line, err := reader.ReadString('\n')
    if err != nil && err != io.EOF {
        log.Fatal(err)
    }
    // 去除换行符并处理
    line = strings.TrimSpace(line)
    fmt.Println(line)
    if err == io.EOF {
        break
    }
}

说明:使用 ReadString 按分隔符读取内容,适合处理日志、配置文件等文本数据。也可以使用 ReadBytesScanner(基于 bufio)实现更灵活的控制。

使用 bufio.Writer 缓冲写入文件

bufio.Writer 将写入的数据先存入内存缓冲区,当缓冲区满或显式调用 Flush 时才真正写入磁盘,极大减少了磁盘I/O次数。

file, err := os.Create("output.txt")
if err != nil {
    log.Fatal(err)
}
defer file.Close()

writer := bufio.NewWriter(file)
data := []string{"第一行", "第二行", "第三行"}
for _, line := range data {
    _, err := writer.WriteString(line + "\n")
    if err != nil {
        log.Fatal(err)
    }
}
// 必须调用 Flush,确保数据写入磁盘
err = writer.Flush()
if err != nil {
    log.Fatal(err)
}

注意:Flush 是关键步骤,遗漏会导致部分或全部数据未写入。适用于日志写入、批量导出等场景。

组合使用 Reader 和 Writer 实现高效文件复制

结合两者可以高效完成文件复制任务,减少系统调用和内存占用:

input, err := os.Open("source.txt")
if err != nil {
    log.Fatal(err)
}
defer input.Close()

output, err := os.Create("dest.txt")
if err != nil {
    log.Fatal(err)
}
defer output.Close()

reader := bufio.NewReader(input)
writer := bufio.NewWriter(output)
defer writer.Flush()

buffer := make([]byte, 4096)
for {
    n, err := reader.Read(buffer)
    if err != nil && err != io.EOF {
        log.Fatal(err)
    }
    if n == 0 {
        break
    }
    _, err = writer.Write(buffer[:n])
    if err != nil {
        log.Fatal(err)
    }
}

这种方式比逐字节读写快得多,且内存使用可控。

基本上就这些。合理使用 bufio 能让文件操作更高效,关键是理解缓冲机制并记得刷新写入缓冲区。不复杂但容易忽略细节。