Go 中如何高效遍历目录？探索几种方法

在Go中，有几种方法可以高效地遍历目录。以下是一些主要的方法：

1. 使用os包的ReadDir函数

os包的ReadDir函数可以读取目录中的所有文件和子目录，并返回一个FileInfo列表。这是一个同步函数，会阻塞当前线程直到读取完毕。

package main
 
import (
    "fmt"
    "os"
)
 
func main() {
    files, err := os.ReadDir("/path/to/dir")
    if err != nil {
        log.Fatal(err)
    }
 
    for _, file := range files {
        fmt.Println(file.Name())
    }
}

2. 使用ioutil包的ReadDir函数

ioutil包的ReadDir函数与os包的ReadDir函数类似，只不过它读取目录的方式略有不同。它读取目录内容到内存中，然后返回文件信息的切片。

package main
 
import (
    "fmt"
    "io/ioutil"
    "log"
)
 
func main() {
    files, err := ioutil.ReadDir("/path/to/dir")
    if err != nil {
        log.Fatal(err)
    }
 
    for _, file := range files {
        fmt.Println(file.Name())
    }
}

3. 使用filepath包的Walk函数

filepath包的Walk函数是一个用来遍历目录和子目录的函数。它接受一个目录路径和一个处理文件的函数作为参数。

package main
 
import (
    "fmt"
    "os"
    "path/filepath"
)
 
func main() {
    err := filepath.Walk("/path/to/dir", func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }
        fmt.Println(path)
        return nil
    })
    if err != nil {
        log.Fatal(err)
    }
}

4. 使用os包的ReadDir和Stat函数

os包的Stat函数可以获取文件或目录的信息。如果是目录，可以配合ReadDir函数进行遍历。

package main
 
import (
    "fmt"
    "log"
    "os"
)
 
func main() {
    info, err := os.Stat("/path/to/dir")
    if err != nil {
        log.Fatal(err)
    }
 
    if info.IsDir() {
        files, err := os.ReadDir("/path/to/dir")
        if err != nil {
            log.Fatal(err)
        }
 
        for _, file := range files {
            fmt.Println(file.Name())
        }
    }
}

以上就是在Go中遍历目录的几种常见方法。你可以根据实际需求和场景选择合适的方法。