写文件

上一节课介绍了 Go 读文件的常用操作，本章节将介绍 Go 写文件的相关操作，包括：

  • 创建文件
  • 在文件指定位置写入内容
  • 通过 Buffered Writter 写文件

创建文件

  • 使用 os.Create(name string) 方法创建文件
  • 使用 os.Stat(name string) 方法获取文件信息

简单示例：

  1. package main
  3. import (
  4.     "fmt"
  5.     "io/ioutil"
  6.     "os"
  7. )
  9. func checkErr(err error) {
  10.     if err != nil {
  11.         panic(err)
  12.     }
  13. }
  15. func main() {
  16.     path := "test.txt"
  18.     newFile, err := os.Create(path)
  19.     checkErr(err)
  20.     defer newFile.Close()
  22.     fileInfo, err := os.Stat(path)
  23.     if err != nil {
  24.         if os.IsNotExist(err) {
  25.             fmt.Println("file doesn't exist!")
  26.             return
  27.         }
  28.     }
  29.     fmt.Println("file does exist, file name : ", fileInfo.Name())
  30. }

除了上述方法外，还可以通过 ioutil.WriteFile 一步完成文件创建和写入操作。假如该文件之前已经存在，那么将会覆盖掉原来的内容，写入新的内容。

示例如下： 

  1. package main
  3. import (
  4.     "fmt"
  5.     "io/ioutil"
  6. )
  8. func checkErr(err error) {
  9.     if err != nil {
  10.         panic(err)
  11.     }
  12. }
  14. func readFile(path string) {
  15.     data, err := ioutil.ReadFile(path)
  16.     checkErr(err)
  17.     fmt.Println("file content: ", string(data))
  18. }
  20. func main() {
  21.     path := "test.txt"
  22.     str := "hello"
  24.     err := ioutil.WriteFile(path, []byte(str), 0644)
  25.     checkErr(err)
  26.     readFile(path)
  27. }

在文件指定位置写入内容

使用 writeAt 可以在文件指定位置写入内容

  1. package main
  3. import (
  4.     "fmt"
  5.     "io/ioutil"
  6.     "os"
  7. )
  9. func checkErr(err error) {
  10.     if err != nil {
  11.         panic(err)
  12.     }
  13. }
  15. func readFile(path string) {
  16.     data, err := ioutil.ReadFile(path)
  17.     checkErr(err)
  18.     fmt.Println("file content: ", string(data))
  19. }
  21. func main() {
  22.     path := "test.txt"
  23.     str := "hello"
  24.     newStr := "world"
  26.     newFile, err := os.Create(path)
  27.     checkErr(err)
  29.     n1, err := newFile.WriteString(str)
  30.     checkErr(err)
  31.     fmt.Println("n1: ", n1)
  32.     readFile(path)
  34.     n2, err := newFile.WriteAt([]byte(newStr), 6)
  35.     checkErr(err)
  36.     fmt.Println("n2: ", n2)
  37.     readFile(path) // the file content should be "helloworld"
  39.     n3, err := newFile.WriteAt([]byte(newStr), 0)
  40.     checkErr(err)
  41.     fmt.Println("n3: ", n3)
  42.     readFile(path) // the file content should be "worldworld"
  43. }

通过 Buffered Writer 写文件

使用 Buffered Writer 可以避免太多次的磁盘 IO 操作。写入的内容首先是存在内存中，当调用 Flush() 方法后才会写入磁盘。

  1. package main
  3. import (
  4.     "bufio"
  5.     "fmt"
  6.     "io/ioutil"
  7.     "os"
  8. )
  10. func checkErr(err error) {
  11.     if err != nil {
  12.         panic(err)
  13.     }
  14. }
  16. func readFile(path string) {
  17.     data, err := ioutil.ReadFile(path)
  18.     checkErr(err)
  19.     fmt.Println("file content: ", string(data))
  20. }
  22. func main() {
  23.     path := "test.txt"
  24.     str := "hello"
  26.     newFile, err := os.Create(path)
  27.     checkErr(err)
  28.     defer newFile.Close()
  30.     bufferWriter := bufio.NewWriter(newFile)
  32.     for _, v := range str {
  33.         written, err := bufferWriter.WriteString(string(v))
  34.         checkErr(err)
  35.         fmt.Println("written: ", written)
  36.     }
  38.     readFile(path) // NOTE: you'll read nothing here because without Flush() operation
  40.     // let's check how much is stored in buffer
  41.     unflushSize := bufferWriter.Buffered()
  42.     fmt.Println("unflushSize: ", unflushSize)
  44.     // write memory buffer to disk
  45.     bufferWriter.Flush()
  47.     readFile(path) // now you can get content from file
  48. }