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函数，可以非常简单地创建临时文件和目录。

本文参考链接：https://blog.csdn.net/neweastsun/article/details/108326692