62. 文件目录
对于操作文件系统中的 目录,Go 提供了几个非常有用的函数。
directories.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
推荐文章: