GO写入文件

223 阅读2分钟

go写入文件有4种写法

……但是在开始之前,我先提供一个判断文件是否存在,存在则打开不存在则创建的封装,相信你会用到的

// OpenFile 判断文件是否存在  存在则OpenFile 不存在则Create
func OpenFile(filename string) (*os.File, error) {
	if _, err := os.Stat(filename); os.IsNotExist(err) {
		fmt.Println("文件不存在")
		return os.Create(filename) //创建文件
	}
	fmt.Println("文件存在")
	return os.OpenFile(filename, os.O_APPEND, 0666) //打开文件
}

方法一:io.WriteString

/******************* 使用 io.WriteString 写入文件 **********************/
f1, err1 := OpenFile("./output1.txt")
if err1 != nil {
    log.Fatal(err1.Error())
}
defer f1.Close()
n, err1 := io.WriteString(f1, "测试文件1") //写入文件(字符串)
if err1 != nil {
    log.Fatal(err1.Error())
}
fmt.Printf("写入 %d 个字节\n", n)

方法二:ioutil.WriteFile

需要注意此种写法将直接新建文件进行写入,如果文件已经存在则会覆盖文件中原本的内容

/*******************  使用 ioutil.WriteFile 写入文件 *****************/
err2 := ioutil.WriteFile("./output2.txt", []byte("测试文件2"), 0666) //写入文件(字节数组)
if err2 != nil {
    log.Fatal(err1.Error())
}

方法三:(f *File) Write / WriteString

/*******************  使用 File(Write,WriteString) 写入文件 ********/
f3, err3 := OpenFile("./output3.txt")
if err3 != nil {
    log.Fatal(err1.Error())
}
defer f3.Close()
n31, err3 := f3.Write([]byte("测试文件3字节流")) //写入文件(字节数组)
if err3 != nil {
    log.Fatal(err3.Error())
}
fmt.Printf("写入 %d 个字节\n", n31)
n32, err3 := f3.WriteString("测试文件3字符串") //写入文件(字节数组)
fmt.Printf("写入 %d 个字节\n", n32)
f3.Sync()

方法四:bufio.NewWriter + (f *File) Write / WriteString

/******************* 使用 bufio.NewWriter 写入文件 *****************/
f4, err4 := OpenFile("./output4.txt")
if err4 != nil {
    log.Fatal(err4.Error())
}
w := bufio.NewWriter(f4) //创建新的 Writer 对象
n41, err4 := w.Write([]byte("测试文件4字节流"))
fmt.Printf("写入 %d 个字节\n", n41)
n42, err4 := w.WriteString("测试文件4字符串")
fmt.Printf("写入 %d 个字节\n", n42)
w.Flush()
f4.Close()

总结

其实查看四种方法的实现以后我们会发现,其实方法三和方法四都是实现了Writer interface,方法三是File结构体的实现,方法四是Writer结构体的实现 ​

方法二的本质是os.WriteFile,跟我们直接调用os.WriteFile是完全相同的效果 ​

而方法一则是直接调用了Writer interface的方法,根据调用的结构体来决定取用哪一个实现,当然上面的代码中取用的是File结构体的实现,所以方法一和方法三,在上述代码中实际调用的方法是同一个。

参考文档