Go 编程实例【临时文件和目录】

我就是我 2023-10-08 23:07 112阅读 0赞

#### 阅读目录 ####

在程序运行时，我们经常创建一些运行时用到，程序结束后就不再使用的数据。

临时目录和文件对于上面的情况很有用，因为它不会随着时间的推移而污染文件系统。

// temporary-files-and-directories.go
    package main
    
    import (
    	"fmt"
    	"io/ioutil"
    	"os"
    	"path/filepath"
    )
    
    func check(e error) {
        
    	if e != nil {
        
    		panic(e)
    	}
    }
    
    func main() {
        
    
    	// 创建临时文件最简单的方法是调用 `ioutil.TempFile` 函数。
    	// 它会创建并打开文件，我们可以对文件进行读写。
    	// 函数的第一个参数传 `""`，`ioutil.TempFile` 会在操作系统的默认位置下创建该文件。
    	f, err := ioutil.TempFile("", "sample")
    	check(err)
    
    	// 打印临时文件的名称。
    	// 文件名以 `ioutil.TempFile` 函数的第二个参数作为前缀，
    	// 剩余的部分会自动生成，以确保并发调用时，生成不重复的文件名。
    	// 在类 Unix 操作系统下，临时目录一般是 `/tmp`。
    	fmt.Println("Temp file name:", f.Name())
    
    	// defer 删除该文件。
    	// 尽管操作系统会自动在某个时间清理临时文件，但手动清理是一个好习惯。
    	defer os.Remove(f.Name())
    
    	// 我们可以向文件写入一些数据。
    	_, err = f.Write([]byte{
        1, 2, 3, 4})
    	check(err)
    
    	// 如果需要写入多个临时文件，最好是为其创建一个临时 *目录*。
    	// `ioutil.TempDir` 的参数与 `TempFile` 相同，
    	// 但是它返回的是一个 *目录名*，而不是一个打开的文件。
    	dname, err := ioutil.TempDir("", "sampledir")
    	fmt.Println("Temp dir name:", dname)
    
    	defer os.RemoveAll(dname)
    
    	// 现在，我们可以通过拼接临时目录和临时文件合成完整的临时文件路径，并写入数据。
    	fname := filepath.Join(dname, "file1")
    	err = ioutil.WriteFile(fname, []byte{
        1, 2}, 0666)
    	check(err)
    }

[root@bogon test]# go run main.go 
    Temp file name: /tmp/sample348430782
    Temp dir name: /tmp/sampledir2156752374
    [root@bogon test]#