Go 编程实例【文件目录】

小鱼儿 2023-10-08 23:06 120阅读 0赞

#### 阅读目录 ####

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

[root@bogon test]# go run main.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
    [root@bogon test]#