17370845950

Go推荐用os.CreateTemp和os.MkdirTemp创建临时文件与目录，支持自动命名、权限与路径安全；测试中优先使用t.TempDir()和t.Cleanup()实现自动清理。

Go 语言通过 os 和 io/ioutil（Go 1.16+ 推荐用 os）标准库提供了简洁可靠的临时文件与目录创建方式，特别适合单元测试、集成测试中避免污染真实文件系统。

使用 os.CreateTemp 创建临时文件

这是 Go 1.16 引入的推荐方式，替代已弃用的 ioutil.TempFile。它自动处理命名冲突、权限设置和路径安全问题。

• 第一个参数是父目录路径（传 "" 表示默认系统临时目录，如 /tmp 或 C:\Users\...\AppData\Local\Temp）
• 第二个参数是模板字符串，支持 *** 占位符（例如 "test-*.log"），会被随机字符替换
• 返回文件句柄和实际路径，出错时需检查 err

示例：

f, err := os.CreateTemp("", "example-*.txt")
if err != nil {
    log.Fatal(err)
}
defer os.Remove(f.Name()) // 测试后清理
defer f.Close()

fmt.Fprintf(f, "hello test")

使用 os.MkdirTemp 创建临时目录

与 CreateTemp 类似，用于生成唯一命名的空目录，常用于模拟配置目录、缓存路径或测试多文件场景。

• 同样支持空字符串作为根目录，使用系统默认临时位置
• 模板规则一致："myapp-**" 会生成类似 myapp-a1b2c3 的目录名
• 创建后务必在测试结束时调用 os.RemoveAll 清理整棵树

示例：

dir, err := os.MkdirTemp("", "testdata-**")
if err != nil {
    log.Fatal(err)
}
defer os.RemoveAll(dir) // 注意：不是 os.Remove！

// 可继续在 dir 内创建子文件或子目录
subfile := filepath.Join(dir, "config.json")
os.WriteFile(subfile, []byte(`{"mode":"test"}`), 0644)

在测试中封装临时资源生命周期

为避免每个测试都重复写清理逻辑，可借助 *testing.T 的 TempDir() 和 T.Cleanup() 方法，让 Go 自动管理。

• t.TempDir() 返回一个测试专属临时目录，测试结束时自动删除（包括所有内容）
• t.Cleanup() 注册回调函数，在测试退出前（无论成功失败）执行，适合关闭文件、释放锁等
• 临时文件不提供内置自动清理，但可结合 t.TempDir() + os.Create 实现统一管理

示例：

func TestProcessConfig(t *testing.T) {
    tmpDir := t.TempDir() // 自动清理整个目录

    cfgPath := filepath.Join(tmpDir, "config.yaml")
    os.WriteFile(cfgPath, []byte("port: 8080"), 0644)

    // 启动被测逻辑（读取 cfgPath）
    result := LoadConfig(cfgPath)
    if result.Port != 8080 {
        t.Error("expected port 8080")
    }
}

注意事项与最佳实践

临时资源虽方便，但忽略细节可能导致测试不稳定或磁盘泄漏。

• 不要硬编码 /tmp 或 C:\temp —— 始终用 os.TempDir() 或空字符串依赖系统策略
• 文件/目录创建后立即检查 err，尤其在 CI 环境中临时目录可能受限
• 避免在 init() 或包级变量中创建临时资源 —— 生命周期不可控
• 跨平台注意路径分隔符：用 filepath.Join 而非字符串拼接
• 若需固定名称（如模拟特定结构），应手动创建并自行清理，不依赖临时 API

不复杂但容易忽略。