Golang临时文件创建 os.CreateTemp使用

os.CreateTemp

os

os.CreateTemp

os.CreateTemp

pattern

*os.File

package main

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

func main() {
    // 1. 创建一个临时文件
    // 第一个参数是目录，如果为空字符串，则使用系统默认临时目录 (os.TempDir())
    // 第二个参数是文件名模式，"my-app-*.txt" 表示文件名前缀是 "my-app-"，
    // 后缀是 ".txt"，中间的 "*" 会被替换成一个随机字符串。
    tmpFile, err := os.CreateTemp("", "my-app-*.txt")
    if err != nil {
        fmt.Printf("创建临时文件失败: %v\n", err)
        return
    }
    // 确保文件最终会被关闭
    defer func() {
        if err := tmpFile.Close(); err != nil {
            fmt.Printf("关闭临时文件失败: %v\n", err)
        }
    }()
    // 确保文件最终会被删除
    // 这一步非常关键，它保证了即使程序异常退出，临时文件也能被清理。
    defer func() {
        if err := os.Remove(tmpFile.Name()); err != nil {
            fmt.Printf("删除临时文件失败: %v\n", err)
        }
    }()

    fmt.Printf("临时文件已创建: %s\n", tmpFile.Name())

    // 2. 向临时文件写入数据
    content := []byte("这是我写入到临时文件的一些内容。\n")
    if _, err := tmpFile.Write(content); err != nil {
        fmt.Printf("写入临时文件失败: %v\n", err)
        return
    }
    fmt.Println("数据已写入临时文件。")

    // 3. 从临时文件读取数据（可选，这里只是为了演示）
    // 需要先将文件指针重置到文件开头
    if _, err := tmpFile.Seek(0, 0); err != nil {
        fmt.Printf("重置文件指针失败: %v\n", err)
        return
    }
    readContent, err := ioutil.ReadAll(tmpFile)
    if err != nil {
        fmt.Printf("读取临时文件失败: %v\n", err)
        return
    }
    fmt.Printf("从临时文件读取到内容:\n%s", string(readContent))

    // 4. 创建一个临时目录 (os.MkdirTemp)
    // 逻辑与os.CreateTemp类似，但创建的是目录
    tmpDir, err := os.MkdirTemp("", "my-temp-dir-*")
    if err != nil {
        fmt.Printf("创建临时目录失败: %v\n", err)
        return
    }
    defer func() {
        // 删除临时目录及其所有内容
        if err := os.RemoveAll(tmpDir); err != nil {
            fmt.Printf("删除临时目录失败: %v\n", err)
        }
    }()
    fmt.Printf("临时目录已创建: %s\n", tmpDir)

    // 在临时目录中创建另一个文件
    nestedFile, err := os.CreateTemp(tmpDir, "nested-*.log")
    if err != nil {
        fmt.Printf("在临时目录中创建文件失败: %v\n", err)
        return
    }
    defer func() {
        if err := nestedFile.Close(); err != nil {
            fmt.Printf("关闭嵌套文件失败: %v\n", err)
        }
        // 这个文件会随着父目录一起被RemoveAll删除，所以这里不需要单独Remove
        // 但为了严谨性，也可以加上 os.Remove(nestedFile.Name())
    }()
    fmt.Printf("在临时目录中创建了文件: %s\n", nestedFile.Name())

    // 程序结束时，tmpFile 和 tmpDir 都会被自动清理。
}

os.Create

ioutil.TempFile

ioutil

os

io

os.CreateTemp

os.MkdirTemp

os.CreateTemp

os.CreateTemp

pattern

*

"my-data-*.json"

os.CreateTemp

my-data-abcdefg.json

pattern

pattern

defer file.Close()

defer os.Remove(file.Name())

os.CreateTemp