在线文档  >   Golang练习   >   目录

Go语言提供了多个有用的函数来操作文件系统中的目录。

package main

import (
    "fmt"
    "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/file3")
    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.Name(), entry.IsDir())
    }

    // `Chdir`让我们改变当前的工作目录,类似于 `cd`。
    err = os.Chdir("subdir/parent/child")
    check(err)

    // 当我们在当前目录中列出子目录`/parent/child` 的内容时,我们将看到其内容。
    c, err = os.ReadDir(".")
    check(err)

    fmt.Println("Listing subdir/parent/child")
    for _, entry := range c {
        fmt.Println(" ", entry.Name(), entry.IsDir())
    }

    // 切换回起始目录。
    err = os.Chdir("../../..")
    check(err)

    // 我们也可以递归地访问目录及其所有子目录。
    // Walk 接受一个回调函数来处理访问的每个文件或目录。
    fmt.Println("Visiting subdir")
    err = filepath.Walk("subdir", visit)
}

// visit 函数被用于 filepath.Walk 递归访问每个文件或目录。
func visit(p string, info os.FileInfo, err error) error {
    if err != nil {
        return err
    }
    fmt.Println(" ", p, info.IsDir())
    return nil
}

运行结果如下:

$ go run directories.go
Listing subdir/parent
  child true
  file2 false
  file3 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/file3 false