Go程序例子(68)：目录

Go程序例子(68)：目录

Go 提供了许多有用的函数来处理文件系统中的目录。

package main

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

func check(e error) {
    if e != nil {
        panic(e)
    }
}

func main() {
		// 在当前工作目录中创建一个新的子目录。
    err := os.Mkdir("subdir", 0755)
    check(err)
		// 在创建临时目录时，最好使用 defer 来延迟删除它们。os.RemoveAll 会删除整个目录树（类似于 rm -rf）。
    defer os.RemoveAll("subdir")
		// 用于创建新空文件的辅助函数。
    createEmptyFile := func(name string) {
        d := []byte("")
        check(os.WriteFile(name, d, 0644))
    }

    createEmptyFile("subdir/file1")
		// 我们可以使用 MkdirAll 创建包含父目录的目录层次结构。这类似于命令行中的 mkdir -p。
    err = os.MkdirAll("subdir/parent/child", 0755)
    check(err)

    createEmptyFile("subdir/parent/file2")
    createEmptyFile("subdir/parent/file")
    createEmptyFile("subdir/parent/child/file4")
		// ReadDir 用于列出目录内容，返回一个 os.DirEntry 对象的切片。
    c, err := os.ReadDir("subdir/parent")
    check(err)

    fmt.Println("Listing subdir/parent")
    for _, entry := range c {
        fmt.Println(" ", entry.ame(), entry.IsDir())
    }
		// Chdir 允许我们更改当前工作目录，类似于 cd 命令。
    err = os.Chdir("subdir/parent/child")
    check(err)
		// 现在，当我们列出当前目录时，会看到 subdir/parent/child 目录中的内容。
    c, err = os.ReadDir(".")
    check(err)

    fmt.Println("Listing subdir/parent/child")
    for _, entry := range c {
        fmt.Println(" ", entry.ame(), entry.IsDir())
    }
		// 切换回我们最初的工作目录。
    err = os.Chdir("../../..")
    check(err)
		// 我们还可以递归地遍历目录，包括其所有子目录。WalkDir 接受一个回调函数，用于处理访问的每个文件或目录。
    fmt.Println("Visiting subdir")
    err = filepath.WalkDir("subdir", visit)
}
// visit 函数会被 filepath.WalkDir 递归到的每个文件或目录调用。
func visit(path string, d fs.DirEntry, err error) error {
    if err != nil {
        return err
    }
    fmt.Println(" ", path, d.IsDir())
    return nil
}

运行结果：

➜ go run directories/directories.go                 
Listing subdir/parent
  child true
  file2 false
  file false
Listing subdir/parent/child
  file4 false
Visiting subdir
  subdir true
  subdir/file1 false
  subdir/parent true
  subdir/parent/child true
  subdir/parent/child/file4 false
  subdir/parent/file2 false
  subdir/parent/file false