go - WriteFile 中途取消

Question

我目前正在将一些标识符写入文件 last_imported.txt。每次导入标识符时，我都希望文件只包含该标识符。

它看起来像 ioutil.WriteFile("last_import.txt", []byte(id), 0644) 可以完成这项工作，但是它会随机将文件留空而不写入新标识符我在运行时取消作业，模拟突然崩溃。

我怀疑每当它留空时，那是因为从未达到 f.write(data)。

func WriteFile(filename string, data []byte, perm os.FileMode) error {
    f, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
    if err != nil {
        return err
    }
    n, err := f.Write(data)
    if err == nil && n < len(data) {
        err = io.ErrShortWrite
    }
    if err1 := f.Close(); err == nil {
        err = err1
    }
    return err
}

如何确保文件不会留空？

Answer 1

您应该使用deferClose() 文件。这会将缓冲区刷新到文件系统。

参见 https://blog.golang.org/defer-panic-and-recover

func WriteFile(filename string, data []byte, perm os.FileMode) error {
    f, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
    if err != nil {
        return err
    }

    defer f.Close() // <-- ADD THIS to close file *************************

    n, err := f.Write(data)
    if err == nil && n < len(data) {
        err = io.ErrShortWrite
    }
    if err1 := f.Close(); err == nil {
        err = err1
    }
    return err
}