Go by Example 中文版: 目录

对于操作文件系统中的 目录 ,Go 提供了几个非常有用的函数。

package main
import (
    "fmt"
    "io/ioutil"
    "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(ioutil.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.FileInfo 类型的切片对象。

    c, err := ioutil.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 = ioutil.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

下一个例子: 临时文件和目录