Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go语言文件写入

一文带你掌握Go语言中文件的写入操作

作者：陈明勇

这篇文章主要和大家分享一下Go语言中文件的写入操作，文中的示例代码讲解详细，对我们学习Go语言有一定的帮助，需要的小伙伴可以参考一下

前言

上篇文章 一文掌握 Go 文件的读取操作 介绍了如何使用 Go os 包和 bufio 包里的几个函数和方法，通过案例展示如何读取文件里的内容。本文接着上篇文章的内容，介绍文件的写入操作。

File.Write、File.WriteString、File.WriteAt

File.Write(b []byte) (n int, err error)

直接操作磁盘往文件里写入数据，写入单位为字节。

File.WriteString(s string) (n int, err error)

直接操作磁盘往指定文件里写入数据，写入单位为字符串。

File.WriteAt(b []byte, off int64) (n int, err error)

从指定位置 off 往文件里顺序写入数据，如果某个偏移量上有数据，则会覆盖。

文件写入操作

import (
    "fmt"
    "os"
)

func main() {
    file, err := os.OpenFile("test.txt", os.O_CREATE, 0)
    if err != nil {
            fmt.Println(err)
            return
    }
    defer file.Close()
    count, err := file.Write([]byte{'H', 'e', 'l', 'l', 'o', ' ', 'W', 'o', 'r', 'l', 'd', '\n'})
    if err != nil {
            return
    }
    fmt.Printf("写入了 %d 字节\n", count)
    count, err = file.WriteString("Hello Golang")
    if err != nil {
            return
    }
    fmt.Printf("写入了长度为 %d 的字符串\n", count)
    count, err = file.WriteAt([]byte{'x', 'x', 'x', 'x', 'x', 'x', 'x', 'x', 'x', 'x', 'x'}, 0)
    if err != nil {
            return
    }
    fmt.Printf("写入了 %d 字节\n", count)
}

首先打开 test.txt 文件，指定的模式为 os.O_CREATE，如果文件不存在则会自动创建；

然后通过 Write 方法以字符的形式往文件里写入 Hello World\n 的内容；

接着通过 WriteString 方法以字符串的形式往文件里写入 Hello Golang 内容；此时文件里的内容如下所示：

Hello World
Hello Golang

最后通过 WriteAt 方法，指定从偏移量为 0 的位置开始写入数据 xxxxxxxxxxx，由于 0 以及之后位置都有数据，因此原有数据被覆盖了。最后文件的内容为：

xxxxxxxxxxx
Hello Golang

File.Seek

File.Seek(offset int64, whence int)

相对于开头位置或当前位置或末尾位置，将设置当前读或写的偏移量设置为 offset。

应用

import (
    "fmt"
    "os"
)

func main() {
    file, err := os.OpenFile("test.txt", os.O_CREATE, 0)
    if err != nil {
            fmt.Println(err)
            return
    }
    defer file.Close()
    _, err = file.WriteString("G0lang")
    if err != nil {
            return
    }
    _, err = file.Seek(1, 0)
    if err != nil {
            fmt.Println(err)
            return
    }
    _, err = file.Write([]byte{'o'})
    if err != nil {
            fmt.Println(err)
            return
    }
}

bufio.NewWriter、Writer.WriteString、Writer.Flush

如果需要多次执行写入文件的操作，推荐使用 bufio 里的 Writer 结构体去操作，它会开辟一个缓冲区，默认大小为 4096 字节。在数据没有被刷入磁盘之前，所写入的数据都会暂时保存到缓冲区里。

NewWriter(w io.Writer) *Writer

开辟一个默认值为 4096 字节的缓冲区，用于暂存写入文件的数据内容，返回一个 Writer 结构体的指针变量

Writer.WriteString(s string) (int, error)

往缓冲区写入内容的方法。

Writer.Flush() error

将所有的缓存数据写入磁盘。

返回值为数据写入磁盘的过程中产生的错误。

文件写入操作

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

func main() {
    file, err := os.OpenFile("test.txt", os.O_CREATE, 0)
    if err != nil {
            fmt.Println(err)
            return
    }
    defer file.Close()
    writer := bufio.NewWriter(file)
    _, err = writer.WriteString("Hello World\n")
    if err != nil {
            fmt.Println(err)
            return
    }
    _, err = writer.WriteString("Hello Golang\n")
    if err != nil {
            fmt.Println(err)
            return
    }
    _, err = writer.WriteString("Hello Gopher\n")
    if err != nil {
            fmt.Println(err)
            return
    }
    writer.Flush()
}

小结

本文先是对 File.Write、File.WriteString、File.WriteAt 进行介绍，通过例子演示它们的使用方式；然后介绍 File.Seek，说明了它的用法；最后引出 bufio.NewWriter、Writer.WriteString、Writer.Flush，使用它们代替 File 结构体里的写入方法，可以不用频繁操作磁盘，提高写入效率。

到此这篇关于一文带你掌握Go语言中文件的写入操作的文章就介绍到这了,更多相关Go语言文件写入内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:
阅读全文