耐心和持久胜过激烈和狂热。
哈喽大家好,我是陈明勇,本文分享的知识是 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 结构体里的写入方法,可以不用频繁操作磁盘,提高写入效率。