Skip to content

目录

对于操作文件系统中的 目录 ,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)

    // 当我们列出 当前 目录,就可以看到 subdir/parent/child 的内容了。
    c, err = os.ReadDir(".")
    check(err)

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

    // cd 回到最开始的地方。
    err = os.Chdir("../../..")
    check(err)

    // 当然,我们也可以遍历一个目录及其所有子目录。 Walk 接受一个路径和回调函数,用于处理访问到的每个目录和文件。
    fmt.Println("Visiting subdir")
    err = filepath.Walk("subdir", visit)
}

// filepath.Walk 遍历访问到每一个目录和文件后,都会调用 visit。
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