Golang 操作临时文件和目录

Golang 操作临时文件和目录

本文学习Go1.11 版本带来的新功能，ioutil包中提供的TempFile 和 TempDir函数用于创建临时文件和目录。其创建的临时文件和目录名称是全局唯一，且方式非常简单。

1. 临时文件

首先使用 TempFile 函数创建临时文件。假设我们正在创建对象识别系统，需要很多图片作为训练集。对于每个汽车图片，我们创建一个临时文件，无需担心文件名称重复。

package main

import (
    "fmt"
    "ioutil"
    "os"
)

func main() {
    // ioutil.TempFile 返回文件和错误
    // 这里指定目录car-images 和 文件名称格式 car-*.png
    file, err := ioutil.TempFile("car-images", "car-*.png")
    if err != nil {
        fmt.Println(err)
    }
    // 确保程序结束时删除临时文件
    defer os.Remove(file.Name())

    // 打印创建的临时文件名称
    fmt.Println(file.Name())
}

运行程序，输出如下：

car-images\car-345523891.png

我们看到自动替换了文件名称中的 *号，使用了一个随机数字。

2. 写临时文件

如果需要写数据至临时文件，使用 write函数：

package main

import (
    "fmt"
    "ioutil"
    "os"
)

func main() {
    // ioutil.TempFile 返回文件和错误
    // 这里指定目录car-images 和 文件名称格式 car-*.png
    file, err := ioutil.TempFile("car-images", "car-*.png")
    if err != nil {
        fmt.Println(err)
    }
    // 确保程序结束时删除临时文件
    defer os.Remove(file.Name())

    // 写 hello world\n 
    if _, err := file.Write([]byte("hello world\n")); err != nil {
        fmt.Println(err)
    }

    data, err := ioutil.ReadFile(file.Name())
    // 如果程序不能读文件打印错误原因
    if err != nil {
        fmt.Println(err)
    }

    // 如果读取成功打印内容
    fmt.Print(string(data))
}

上面往临时文件中写入hello world，然后读取并打印验证。

运行程序输出：

hello world

3. 创建临时目录

上面创建了临时文件，现在我们学习如何创建临时目录。对于我们上面提及的对象识别系统，可能需要创建临时目录存放非汽车的其他物品。这里扩展上面的示例，创建临时目录，然后在目录中创建临时文件：

package main

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

func main() {

	tempDir, err := ioutil.TempDir("", "cars-")
	if err != nil {
		fmt.Println(err)
	}
	defer os.RemoveAll(tempDir)

	file, err := ioutil.TempFile(tempDir, "car-*.png")
	if err != nil {
		fmt.Println(err)
	}
	defer os.Remove(file.Name())

	// 打印文件的完整名称
	fmt.Println(file.Name())

	if _, err := file.Write([]byte("hello world\n")); err != nil {
		fmt.Println(err)
	}

	data, err := ioutil.ReadFile(file.Name())
    // 如果程序不能读文件打印错误原因
	if err != nil {
		fmt.Println(err)
	}

    // 如果读取成功打印内容
	fmt.Print(string(data))
}

运行程序，我们能看到创建了全局唯一目录和临时文件名称，运行结果：

C:\Users\ADMINI~1\AppData\Local\Temp\cars-604917127\car-281252154.png
hello world

4. 总结

本文学习了ioutil包中提供的TempFile 和 TempDir函数，可以非常简单地创建临时文件和目录。