在go语言中，有时我们需要获取一个特定目录下所有文件和文件夹的列表，但又不想递归地遍历其子目录。尽管filepath.walk函数是一个强大的工具，用于遍历文件系统树，但它默认会递归地进入所有子目录，这在只需要获取单层目录内容时显得过于复杂。针对这种场景，go标准库提供了一个更简洁、高效的解决方案：os.readdir函数。
使用 os.ReadDir 列出目录内容
os.ReadDir函数位于os包中，专门用于读取指定目录的内容。它会返回该目录下的所有文件和子目录的条目列表，并按照文件名进行排序。其核心优势在于它不会自动递归到子目录中，完美契合了单层目录列表的需求。
函数签名与返回值
os.ReadDir函数的签名为：
func ReadDir(name string) ([]DirEntry, error)
- name：表示要读取的目录路径。
- []DirEntry：返回一个os.DirEntry类型的切片，其中包含了目录中的所有条目。
- error：如果读取目录时发生错误（例如目录不存在或权限不足），则返回一个错误。
os.DirEntry是一个接口，提供了访问目录条目信息的方法，例如：
- Name() string：返回条目的文件名（不包含路径）。
- IsDir() bool：如果条目是一个目录，则返回true；否则返回false。
- Type() fs.FileMode：返回条目的文件模式和类型。
- Info() (FileInfo, error)：返回条目的fs.FileInfo，其中包含更详细的文件信息。
示例代码
以下是一个基本的Go程序，演示了如何使用os.ReadDir列出当前目录下的所有文件和文件夹的名称，并区分出目录类型。
立即学习“go语言免费学习笔记（深入）”；
package main
import (
"fmt"
"log"
"os"
)
func main() {
// 指定要读取的目录路径，这里以当前目录为例
dirPath := "./"
// 调用 os.ReadDir 读取目录内容
entries, err := os.ReadDir(dirPath)
if err != nil {
// 错误处理，如果目录不存在或权限不足等，程序将终止并打印错误
log.Fatalf("无法读取目录 %s: %v", dirPath, err)
}
fmt.Printf("目录 '%s' 中的内容：\n", dirPath)
// 遍历所有目录条目
for _, entry := range entries {
// 使用 IsDir() 方法判断条目是文件还是目录
if entry.IsDir() {
fmt.Printf(" [目录] %s\n", entry.Name())
} else {
fmt.Printf(" [文件] %s\n", entry.Name())
}
}
}代码解释：
- import 语句：引入了fmt用于格式化输出，log用于错误日志记录，以及os包来使用ReadDir函数。
- dirPath := "./"：定义了要读取的目录路径。"./"表示当前执行程序的目录。你可以将其更改为任何其他绝对或相对路径，例如"/tmp"或"../mydata"。
- entries, err := os.ReadDir(dirPath)：这是核心调用。它尝试读取dirPath指定的目录。如果成功，entries将是一个os.DirEntry切片，err为nil；否则，err将包含错误信息。
- 错误处理：if err != nil { log.Fatalf(...) } 是标准的Go错误处理模式。如果ReadDir返回错误，程序将通过log.Fatalf打印错误并退出。
- 遍历条目：for _, entry := range entries 循环遍历ReadDir返回的所有os.DirEntry。
- 判断类型：entry.IsDir() 方法是判断当前条目是目录还是文件的关键。根据其返回值，我们可以输出不同的标识。
- entry.Name()：获取并打印条目的名称。
注意事项与最佳实践
- 错误处理：始终检查os.ReadDir返回的错误。目录不存在、权限不足等都可能导致错误。
- 路径：确保提供的目录路径是有效的。相对路径是相对于程序执行时的当前工作目录。
- 隐藏文件：os.ReadDir会列出包括隐藏文件和目录在内的所有条目（例如在Unix-like系统中以.开头的文件）。如果需要过滤，可以在遍历时根据entry.Name()进行判断。
- 性能：对于非常大的目录，os.ReadDir会一次性将所有条目读入内存。如果目录包含数百万个条目，这可能会消耗大量内存。但在大多数常见场景下，这并非问题。
- fs.FileInfo：如果除了名称和类型之外，还需要文件的修改时间、大小等更详细的信息，可以使用entry.Info()方法获取fs.FileInfo接口，然后通过该接口访问这些属性。但请注意，entry.Info()可能会导致额外的系统调用，因此仅在需要时使用。
总结
os.ReadDir函数是Go语言中用于非递归列出目录内容的理想选择。它简洁、高效，并提供了os.DirEntry接口来方便地获取每个条目的名称和类型。通过掌握这个函数，开发者可以轻松地实现各种文件系统操作中常见的单层目录内容管理需求，避免了filepath.Walk在不需要递归时的复杂性。
