Go 프로그래밍 예제 [임시 파일 및 디렉토리]

프로그램이 실행 중일 때 런타임에 사용되고 프로그램이 종료된 후에는 더 이상 사용되지 않는 일부 데이터를 생성하는 경우가 많습니다.

임시 디렉토리 및 파일은 시간이 지나도 파일 시스템을 오염시키지 않으므로 위의 경우에 유용합니다.

// 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]#