如何安全地使用临时文件实现原子写入并正确设置文件权限-Golang-PHP中文网

本文讲解在 Go 中如何通过临时文件实现原子写入，避免程序崩溃导致文件损坏，并解决 ioutil.TempFile 默认权限（0600）与期望目标权限（如 0644）不匹配的问题。

本文讲解在 go 中如何通过临时文件实现原子写入，避免程序崩溃导致文件损坏，并解决 `ioutil.tempfile` 默认权限（0600）与期望目标权限（0644）不匹配的问题。

在 Go 应用中，为保障数据一致性，推荐采用“先写临时文件、再原子重命名”的策略替代直接写入目标文件——这能有效防止程序崩溃或中断时留下不完整/损坏的文件。然而，ioutil.TempFile（Go 1.16+ 已弃用，应使用 os.CreateTemp）默认以 0600 权限创建临时文件，该权限不受当前进程 umask 影响（因其内部调用 os.OpenFile(..., 0600)），导致最终重命名后文件仍为 0600，而非预期的 0644 或 0660。

这不是设计缺陷，而是安全必需：临时文件需严格限制访问权限，防止敏感中间数据被其他用户或进程读取。因此，不应强行让临时文件继承宽松权限，而应在重命名后、对外可见前，显式设置目标权限。

✅ 推荐做法（简洁、安全、符合 POSIX 习惯）：

package main

import (
    "os"
    "os/exec"
    "path/filepath"
)

// atomicWrite 将数据写入目标路径，保证原子性与正确权限
func atomicWrite(dst string, data []byte, perm os.FileMode) error {
    // 1. 创建同目录下的临时文件（使用唯一名称，避免冲突）
    tmpDir := filepath.Dir(dst)
    tmpFile, err := os.CreateTemp(tmpDir, filepath.Base(dst)+".tmp.*")
    if err != nil {
        return err
    }
    defer os.Remove(tmpFile.Name()) // 清理残留（仅当写入失败时生效）

    // 2. 写入数据
    if _, err := tmpFile.Write(data); err != nil {
        tmpFile.Close()
        return err
    }
    if err := tmpFile.Close(); err != nil {
        return err
    }

    // 3. 显式设置期望权限（关键步骤！）
    if err := os.Chmod(tmpFile.Name(), perm); err != nil {
        return err
    }

    // 4. 原子重命名（覆盖目标文件）
    return os.Rename(tmpFile.Name(), dst)
}

// 使用示例
func main() {
    err := atomicWrite("/var/data/config.json", []byte(`{"mode":"prod"}`), 0644)
    if err != nil {
        panic(err)
    }
}

⚠️ 注意事项：