详解Go语言中如何高效遍历目录

这篇文章将为大家详细讲解有关详解Go语言中如何高效遍历目录，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

Go语言中高效遍历目录

在Go语言中，遍历目录是一个常见的任务。为了高效地完成此任务，有几种方法可以采用，具体取决于具体的用例。

使用 io/fs 包

io/fs 包提供了对文件系统的基本访问，包括遍历目录。函数 ReadDir 和 WalkDir 可用于遍历目录，它们返回包含目录中条目信息的 FileInfo 或 DirEntry 类型的切片。

package main

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

func main() {
    root := "." // 目录的路径
    err := filepath.WalkDir(root, func(path string, info fs.DirEntry, err error) error {
        if err != nil {
            return err
        }
        if info.IsDir() {
            fmt.Printf("目录: %s
", path)
        } else {
            fmt.Printf("文件: %s
", path)
        }
        return nil
    })
    if err != nil {
        fmt.Printf("遍历目录失败: %v
", err)
    }
}

使用 os 包

os 包也提供了遍历目录的功能，但它的方法可能不像 io/fs 包那么高效。函数 ReadDir 返回一个包含目录中文件名（不包含路径）的切片。

package main

import (
    "fmt"
    "os"
    "path/filepath"
)

func main() {
    root := "." // 目录的路径
    files, err := os.ReadDir(root)
    if err != nil {
        fmt.Printf("遍历目录失败: %v
", err)
    }
    for _, file := range files {
        if file.IsDir() {
            fmt.Printf("目录: %s
", filepath.Join(root, file.Name()))
        } else {
            fmt.Printf("文件: %s
", filepath.Join(root, file.Name()))
        }
    }
}

并行遍历

对于大型目录，并行遍历可以提高效率。Go语言的 sync/atomic 包提供了原子操作，可以安全地并发修改共享变量。此外，可以利用 goroutine 并发执行遍历任务。

package main

import (
    "fmt"
    "os"
    "path/filepath"
    "sync"
    "sync/atomic"
)

func main() {
    var count uint64 // 原子计数器，用于统计遍历的文件数
    var wg sync.WaitGroup // 等待组，用于等待所有 goroutine 完成

    root := "." // 目录的路径
    err := filepath.WalkDir(root, func(path string, info fs.DirEntry, err error) error {
        if err != nil {
            return err
        }
        if info.IsDir() {
            wg.Add(1)
            go func() {
                defer wg.Done()
                err := filepath.WalkDir(path, func(p string, i fs.DirEntry, err error) error {
                    if err != nil {
                        return err
                    }
                    if !i.IsDir() {
                        atomic.AddUint64(&count, 1)
                    }
                    return nil
                })
                if err != nil {
                    fmt.Printf("遍历目录失败: %v
", err)
                }
            }()
        } else {
            atomic.AddUint64(&count, 1)
        }
        return nil
    })
    if err != nil {
        fmt.Printf("遍历目录失败: %v
", err)
    }

    wg.Wait() // 等待所有 goroutine 完成
    fmt.Printf("文件总数: %d
", count)
}

以上就是详解Go语言中如何高效遍历目录的详细内容，更多请关注编程学习网其它相关文章！