GO语言进阶教程之遍历文件夹

1,165 阅读1分钟

学习io之后,尤其是文件操作,我们就可以遍历给定的目录文件夹了。可以使用ioutil包下的readDir()方法,这个方法可以获取指定目录下的内容,返回文件和子目录。

因为文件夹下还有子文件夹,而ioutil包的ReadDir()只能获取一层目录,所以我们需要自己去设计算法来实现,最容易实现的思路就是使用递归。

示例代码:

package main​import (    "io/ioutil"    "fmt"    "log")​func main() {    /**    遍历文件夹:     */​    dirname := "/Users/ruby/Documents/pro"    listFiles(dirname, 0)​}​func listFiles(dirname string, level int) {    // level用来记录当前递归的层次    // 生成有层次感的空格    s := "|--"    for i := 0; i < level; i++ {        s = "|   " + s    }​    fileInfos, err := ioutil.ReadDir(dirname)    if err != nil{        log.Fatal(err)    }    for _, fi := range fileInfos {        filename := dirname + "/" + fi.Name()        fmt.Printf("%s%s\n", s, filename)        if fi.IsDir() {            //继续遍历fi这个目录            listFiles(filename, level+1)        }    }}​​

运行结果:

GO语言进阶教程之遍历文件夹