在 go 语言中，处理文件系统操作是常见的需求。当我们需要获取一个目录下所有文件和文件夹的列表，但不希望递归地遍历子目录时，os 包提供的 readdir 函数是理想的选择。与 filepath.walk 不同，readdir 专注于单层目录的读取，避免了不必要的递归操作。
使用 os.ReadDir 列出目录内容
os.ReadDir 函数用于读取指定目录的内容。它接收一个字符串参数，表示要读取的目录路径，并返回一个 os.DirEntry 类型的切片以及一个错误。如果目录不存在或没有读取权限，将返回相应的错误。
os.DirEntry 接口提供了访问目录条目信息的方法，其中最常用的是：
- Name() string: 返回目录条目的名称。
- IsDir() bool: 判断该条目是否为目录。
下面是一个基本的示例，展示如何列出当前目录下的所有文件和文件夹的名称：
package main
import (
"fmt"
"log"
"os"
)
func main() {
// 读取当前目录（"."）的内容
entries, err := os.ReadDir("./")
if err != nil {
// 如果发生错误，使用 log.Fatalf 打印错误并退出程序
log.Fatalf("无法读取目录: %v", err)
}
fmt.Println("当前目录内容:")
// 遍历所有目录条目并打印其名称
for _, entry := range entries {
fmt.Println(entry.Name())
}
}运行上述代码，你将看到当前目录下所有文件和文件夹的名称列表。请注意，此示例中并未区分文件和目录，它们都被统一列出。
区分文件与目录
在很多场景下，我们需要明确区分文件和目录。os.DirEntry 接口的 IsDir() 方法为此提供了便利。我们可以利用这个方法，在遍历时判断每个条目的类型。
以下是修改后的示例代码，它会明确标识出每个条目是文件还是目录：
package main
import (
"fmt"
"log"
"os"
)
func main() {
// 读取指定目录的内容，这里以当前目录为例
dirPath := "./"
entries, err := os.ReadDir(dirPath)
if err != nil {
log.Fatalf("无法读取目录 '%s': %v", dirPath, err)
}
fmt.Printf("目录 '%s' 的内容:\n", dirPath)
for _, entry := range entries {
if entry.IsDir() {
fmt.Printf("DIR: %s\n", entry.Name())
} else {
fmt.Printf("FILE: %s\n", entry.Name())
}
}
}通过这个示例，你可以清晰地看到每个条目是文件还是目录，这对于需要进行特定操作（例如只处理文件或只遍历子目录）的场景非常有用。
注意事项
在使用 os.ReadDir 函数时，有几个关键点需要注意：
- 错误处理：始终检查 os.ReadDir 返回的错误。文件系统操作可能会因各种原因失败，例如目录不存在、权限不足等。
- 非递归性：os.ReadDir 只会读取指定目录的第一层内容，不会自动进入子目录。如果你需要递归遍历整个文件树，应考虑使用 filepath.Walk 或自行实现递归逻辑。
- 结果排序：os.ReadDir 返回的 os.DirEntry 切片是按文件名（字典序）排序的。
- 性能考虑：对于包含大量文件和子目录的目录，os.ReadDir 会一次性将所有条目加载到内存中。在极端情况下，这可能会对内存造成一定压力。
- 权限：确保 Go 程序运行的用户对目标目录具有足够的读取权限。
总结
os.ReadDir 是 Go 语言中一个强大且直观的函数，用于非递归地列出指定目录的内容。通过结合 os.DirEntry 接口提供的方法，我们可以轻松地获取条目名称并区分文件与目录。掌握这一功能，能够帮助开发者高效地进行文件系统操作，构建健壮的 Go 应用程序。在处理单层目录内容列表时，os.ReadDir 应当是你的首选工具。
