golang 读取目录

在现代软件开发中，读取目录是一个非常常见的需求。无论是查找文件、检索文件列表还是统计文件数量，我们都需要通过读取目录来实现。而对于Golang开发者来说，如何高效地读取目录是一个非常重要的技能。本文将介绍Golang中如何读取目录，并提供一些实用的代码示例。

使用os包读取目录

在Golang中，我们可以使用os包提供的函数来读取目录。os包中有一个名为ReadDir的函数，它可以读取一个目录并返回该目录下的所有文件和子目录。下面是一个简单的示例代码：

package main

import (
	"fmt"
	"io/fs"
	"os"
)

func main() {
	dir := "./myfolder"
	files, err := os.ReadDir(dir)
	if err != nil {
		fmt.Println(err)
		return
	}

	for _, file := range files {
		fmt.Println(file.Name())
	}
}

上述代码首先使用os包的ReadDir函数读取指定目录，然后通过for循环遍历返回的文件和目录列表，并打印出它们的名称。这样，我们就可以轻松地获取目录下的所有文件和子目录的名称了。

递归读取目录

有时候，我们需要递归地读取一个目录及其子目录中的所有文件。Golang中的filepath包提供了一些功能来帮助我们实现这个需求。下面是一个递归读取目录的示例代码：

package main

import (
	"fmt"
	"io/fs"
	"path/filepath"
)

func visitDir(path string, d fs.DirEntry, err error) error {
	if err != nil {
		fmt.Println(err)
		return nil
	}

	if d.IsDir() {
		fmt.Println("Directory:", d.Name())
	} else {
		fmt.Println("File:", d.Name())
	}

	return nil
}

func main() {
	dir := "./myfolder"
	err := filepath.WalkDir(dir, visitDir)
	if err != nil {
		fmt.Println(err)
		return
	}
}

上述代码使用filepath包的WalkDir函数来遍历指定目录及其子目录中的所有文件和目录。通过定义visitDir函数作为回调函数，我们可以在遍历过程中处理每一个文件和目录。visitDir函数接收三个参数，分别代表当前文件或目录的路径、fs.DirEntry对象和任何可能出现的错误。我们可以根据需要在visitDir函数中进行相应的处理。

过滤读取目录

有时候，我们想要过滤掉一些特定类型的文件，比如只读取以.go为后缀的文件。Golang中的filepath包提供了一个名为Match函数，它可以用来判断一个文件路径是否与指定的模式匹配。下面是一个过滤读取目录的示例代码：

package main

import (
	"fmt"
	"io/fs"
	"path/filepath"
)

func visitDir(path string, d fs.DirEntry, err error) error {
	if err != nil {
		fmt.Println(err)
		return nil
	}

	if !d.IsDir() && filepath.Ext(path) == ".go" {
		fmt.Println("Go file:", d.Name())
	}

	return nil
}

func main() {
	dir := "./myfolder"
	err := filepath.WalkDir(dir, visitDir)
	if err != nil {
		fmt.Println(err)
		return
	}
}

上述代码通过在visitDir函数中判断文件路径的后缀是否为.go来过滤只读取后缀为.go的文件。这样，我们就可以只获取目录中的Go源代码文件了。

通过使用Golang的os包和filepath包，我们可以轻松地读取目录，并递归地遍历目录下的所有文件和子目录。我们还可以通过过滤操作，仅获取满足特定条件的文件。掌握这些技巧，将有助于提高我们处理目录的效率。