golang 扫描目录

扫描目录是Golang中一个常见的操作，尤其在需要处理大量文件的应用中，它能帮助我们快速定位到指定目录下的所有文件，并进行遍历、查找、统计等操作。本文将介绍如何使用Golang扫描目录，并提供一些实际应用场景。

一、基本用法

Golang提供了标准库中的filepath.Walk函数来扫描目录。这个函数接受三个参数：根目录路径、一个回调函数和处理错误的函数。回调函数会在遍历到每个文件或目录的时候被调用，并传递相应的路径信息。

下面是一个简单的示例，展示了如何使用filepath.Walk来扫描目录：

```go func scanDir(rootDir string) error { return filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error { fmt.Println(path) return nil }) } func main() { rootDir := "/path/to/directory" err := scanDir(rootDir) if err != nil { log.Fatal(err) } } ```

二、筛选特定类型文件

有时候我们只想获取某种类型的文件，可以在回调函数中进行判断，并进行相应的处理。例如，我们只想获取所有的文本文件：

```go func scanTextFiles(rootDir string) error { return filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error { if !info.IsDir() && strings.HasSuffix(info.Name(), ".txt") { fmt.Println(path) } return nil }) } ```

这里我们使用strings.HasSuffix函数判断文件名是否以“.txt”结尾，如果是的话，则打印出路径。

三、并发扫描

当需要处理大量文件时，使用并发扫描可以显著提高程序的执行速度。Golang提供了协程（goroutine）和通道（channel）来实现并发编程。我们可以使用多个协程同时扫描目录，并将结果发送到一个通道中来实现。

```go func scanDirConcurrently(rootDir string) ([]string, error) { result := make(chan string) var wg sync.WaitGroup errc := make(chan error, 1) wg.Add(1) go func() { defer wg.Done() err := filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error { if err != nil { select { case errc <- err: default: } return nil } if !info.IsDir() && strings.HasSuffix(info.Name(), ".txt") { result <- path } return nil }) if err != nil { select { case errc <- err: default: } } }() go func() { wg.Wait() close(result) }() paths := []string{} for path := range result { paths = append(paths, path) } select { case err := <-errc: return nil, err default: return paths, nil } } ```

通过将任务划分为多个协程并发执行，我们可以充分利用多核处理器的性能。上述代码使用了一个通道result来收集扫描结果，以及一个通道errc来处理错误。函数返回的结果是扫描到的所有文件路径。

这只是扫描目录的一些基本用法和应用场景，Golang标准库提供了丰富的功能，可以通过组合和调整参数来实现更多复杂的需求。希望这篇文章能够帮助你在Golang中高效地进行目录扫描操作。