写文件

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

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

创建文件

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

简单示例：

package main

import (
    "fmt"
    "io/ioutil"
    "os"
)

func checkErr(err error) {
    if err != nil {
        panic(err)
    }
}

func main() {
    path := "test.txt"

    newFile, err := os.Create(path)
    checkErr(err)
    defer newFile.Close()

    fileInfo, err := os.Stat(path)
    if err != nil {
        if os.IsNotExist(err) {
            fmt.Println("file doesn't exist!")
            return
        }
    }
    fmt.Println("file does exist, file name : ", fileInfo.Name())
}

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

示例如下：

package main

import (
    "fmt"
    "io/ioutil"
)

func checkErr(err error) {
    if err != nil {
        panic(err)
    }
}

func readFile(path string) {
    data, err := ioutil.ReadFile(path)
    checkErr(err)
    fmt.Println("file content: ", string(data))
}

func main() {
    path := "test.txt"
    str := "hello"

    err := ioutil.WriteFile(path, []byte(str), 0644)
    checkErr(err)
    readFile(path)
}

在文件指定位置写入内容

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

package main

import (
    "fmt"
    "io/ioutil"
    "os"
)

func checkErr(err error) {
    if err != nil {
        panic(err)
    }
}

func readFile(path string) {
    data, err := ioutil.ReadFile(path)
    checkErr(err)
    fmt.Println("file content: ", string(data))
}

func main() {
    path := "test.txt"
    str := "hello"
    newStr := "world"

    newFile, err := os.Create(path)
    checkErr(err)

    n1, err := newFile.WriteString(str)
    checkErr(err)
    fmt.Println("n1: ", n1)
    readFile(path)

    n2, err := newFile.WriteAt([]byte(newStr), 6)
    checkErr(err)
    fmt.Println("n2: ", n2)
    readFile(path) // the file content should be "helloworld"

    n3, err := newFile.WriteAt([]byte(newStr), 0)
    checkErr(err)
    fmt.Println("n3: ", n3)
    readFile(path) // the file content should be "worldworld"
}

通过 Buffered Writer 写文件

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

package main

import (
    "bufio"
    "fmt"
    "io/ioutil"
    "os"
)

func checkErr(err error) {
    if err != nil {
        panic(err)
    }
}

func readFile(path string) {
    data, err := ioutil.ReadFile(path)
    checkErr(err)
    fmt.Println("file content: ", string(data))
}

func main() {
    path := "test.txt"
    str := "hello"

    newFile, err := os.Create(path)
    checkErr(err)
    defer newFile.Close()

    bufferWriter := bufio.NewWriter(newFile)

    for _, v := range str {
        written, err := bufferWriter.WriteString(string(v))
        checkErr(err)
        fmt.Println("written: ", written)
    }

    readFile(path) // NOTE: you'll read nothing here because without Flush() operation

    // let's check how much is stored in buffer
    unflushSize := bufferWriter.Buffered()
    fmt.Println("unflushSize: ", unflushSize)

    // write memory buffer to disk
    bufferWriter.Flush()

    readFile(path) // now you can get content from file
}