耐心和持久胜过激烈和狂热。
哈喽大家好，我是陈明勇，今天分享的知识是 Go 文件的写入操作。如果本文对你有帮助，不妨点个赞，如果你是 Go 语言初学者，不妨点个关注，一起成长一起进步，如果本文有错误的地方，欢迎指出！
前言
上篇文章 一文掌握 Go 文件的读取操作
介绍了如何使用 Go os 包和 bufio 包里的几个函数和方法，通过案例展示如何读取文件里的内容。本文接着上篇文章的内容，介绍文件的写入操作。
File.Write、File.WriteString、File.WriteAt
-
File.Write(b []byte) (n int, err error)
直接操作磁盘往文件里写入数据，写入单位为字节。
-
b 参数：写入的数据，类型为字节切片。 - 返回值
n：写入的字节数。 - 返回值
err：写入数据的过程中产生的错误。
-
File.WriteString(s string) (n int, err error)
直接操作磁盘往指定文件里写入数据，写入单位为字符串。
-
s 参数：写入的字符串数据。 - 返回值
n：写入的字节数。 - 返回值
err：写入数据的过程中产生的错误。
-
File.WriteAt(b []byte, off int64) (n int, err error)
从指定位置 off 往文件里顺序写入数据，如果某个偏移量上有数据，则会覆盖。
-
b 参数：写入的数据，类型为字节切片。 -
off 参数：偏移量，从此位置开始写入数据。 - 返回值
n：写入的字节数。 - 返回值
err：写入数据的过程中产生的错误。
文件写入操作
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。
-
offset 参数：所要设置的偏移量。 -
whence：相对于哪个位置开始设置偏移量的标志，可选值为 0 → 开头位置，1 → 当前位置，2 → 末尾位置。
应用
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
}
}
- 打开
test.txt 文件，指定的模式为 os.O_CREATE，如果文件不存在则会自动创建； - 使用
WriteString 方法往文件里写入 G0lang 字符串； - 此时发现第二个字符错了，
0 应该改成 o；此时的偏移量是指向尾部的；使用 Seek 方法将偏移量移到第二个位置，然后写入字符 o，由于当前位置已有数据 0，因此 o 将会覆盖 0；
bufio.NewWriter、Writer.WriteString、Writer.Flush
如果需要多次执行写入文件的操作，推荐使用 bufio 里的 Writer 结构体去操作，它会开辟一个缓冲区，默认大小为 4096 字节。在数据没有被刷入磁盘之前，所写入的数据都会暂时保存到缓冲区里。
-
NewWriter(w io.Writer) *Writer
开辟一个默认值为 4096 字节的缓冲区，用于暂存写入文件的数据内容，返回一个 Writer 结构体的指针变量
-
w 参数：类型为 Writer 接口，实现这个接口的数据类型变量都可以作为参数，例如 File。 - 返回值
*Writer：一个 Writer 结构体的指针变量，通过该变量可以往缓冲区里写入数据。
-
Writer.WriteString(s string) (int, error)
往缓冲区写入内容的方法。
- 参数
s 为写入的字符串。 - 第一个返回值为写入的字节数。
- 第二个返回值为写入数据的过程中产生的错误。
-
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()
}
- 首先打开
test.txt 文件，指定的模式为 os.O_CREATE，如果文件不存在则会自动创建； - 然后使用
NewWriter 函数获取一个 Writer 结构体的指针变量 writer； - 接着通过
writer 的 WriteString 方法将内容保存到缓冲区里； - 最后调用
Flush 方法，将所有的缓存数据写入磁盘。
小结
本文先是对 File.Write、File.WriteString、File.WriteAt 进行介绍，通过例子演示它们的使用方式；然后介绍 File.Seek，说明了它的用法；最后引出 bufio.NewWriter、Writer.WriteString、Writer.Flush，使用它们代替 File 结构体里的写入方法，可以不用频繁操作磁盘，提高写入效率。