GO 的 Web 开发系列（六）—— 遍历路径下的文件

文件 IO 处理是程序的基础功能，WEB 程序中通过文件 IO 实现附件的上传与下载。在 GO 中，有多种方式可以遍历文件目录，获取文件路径，本文从使用层面上论述这些函数。

预先准备一个包含子目录的目录，用于遍历测试：

一、filepath.Walk 函数

使用 filepath.Walk 函数可以很方便的遍历目录下的多级子文件和目录。

代码示例：

func main() {
	rootPath := "./docs"
	_ = filepath.Walk(rootPath, func(path string, info os.FileInfo, err error) error {
		fmt.Printf("name: %s, isDir: %t, path: %s\n", info.Name(), info.IsDir(), path)
		return nil
	})
}

输出结果：

name: docs, isDir: true, path: ./docs
name: admin, isDir: true, path: docs\admin
name: admin_docs.go, isDir: false, path: docs\admin\admin_docs.go
name: admin_swagger.json, isDir: false, path: docs\admin\admin_swagger.json      
name: admin_swagger.yaml, isDir: false, path: docs\admin\admin_swagger.yaml      
name: application.yaml, isDir: false, path: docs\application.yaml
name: content, isDir: true, path: docs\content
name: content_docs.go, isDir: false, path: docs\content\content_docs.go
name: content_swagger.json, isDir: false, path: docs\content\content_swagger.json
name: content_swagger.yaml, isDir: false, path: docs\content\content_swagger.yaml

二、os.ReadDir 函数

ReadDir 函数仅遍历当前目录下的子文件和目录，不会递归遍历，如果需要多级遍历则需要自己实现。

该函数有两个实现，分别为 os.ReadDir 和 ioutil.ReadDir ，两个函数结果相同。在文件数量多的情况下， os.ReadDir 的性能会更好。新版本 Go 中 ioutil.ReadDir 已经被标记废弃了，不推荐使用。

代码示例：

func main() {
	rootPath := "./docs"
	files, err := os.ReadDir(rootPath)
	if err != nil {
		fmt.Println(err)
	}
	for _, file := range files {
		fmt.Printf("name: %s, isDir: %t, path: %s\n", file.Name(), file.IsDir(), rootPath+"/"+file.Name())
	}
}

输出结果：

name: admin, isDir: true, path: ./docs/admin
name: application.yaml, isDir: false, path: ./docs/application.yaml
name: content, isDir: true, path: ./docs/content

三、filepath.Glob 函数

filepath.Glob 函数仅遍历当前目录下的子文件和目录，不会递归遍历，但可以使用通配符对遍历的文件进行过滤。

代码示例：

func main() {
	rootPath := "./docs"
	filePaths, err := filepath.Glob(filepath.Join(rootPath, "*"))
	if err != nil {
		fmt.Println(err)
	}
	for _, filePath := range filePaths {
		fmt.Println(filePath)
	}
	fmt.Println("## 查找 yaml 文件")
	filePaths, err = filepath.Glob(filepath.Join(rootPath, "*.yaml"))
	if err != nil {
		fmt.Println(err)
	}
	for _, filePath := range filePaths {
		fmt.Println(filePath)
	}
}

输出结果：

docs\admin
docs\application.yaml
docs\content
## 查找 yaml 文件    
docs\application.yaml