golang 遍历文件夹和文件

你的名字 2021-09-02 04:03 851阅读 0赞

代码示例：

package main
    
    import (
        "fmt"
        "io/ioutil"
        "os"
        "strings"
    )
    
    //获取指定目录下的所有文件和目录
    func GetFilesAndDirs(dirPth string) (files []string, dirs []string, err error) {
        dir, err := ioutil.ReadDir(dirPth)
        if err != nil {
            return nil, nil, err
        }
    
        PthSep := string(os.PathSeparator)
        //suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
    
        for _, fi := range dir {
            if fi.IsDir() { // 目录, 递归遍历
                dirs = append(dirs, dirPth+PthSep+fi.Name())
                GetFilesAndDirs(dirPth + PthSep + fi.Name())
            } else {
                files = append(files, dirPth+PthSep+fi.Name())
    
                // 过滤指定格式
                // ok := strings.HasSuffix(fi.Name(), ".go")
                // if ok {
                //     files = append(files, dirPth+PthSep+fi.Name())
                // }
            }
        }
    
        return files, dirs, nil
    }
    
    //获取指定目录下的所有文件,包含子目录下的文件
    func GetAllFiles(dirPth string) (files []string, err error) {
        var dirs []string
        dir, err := ioutil.ReadDir(dirPth)
        if err != nil {
            return nil, err
        }
    
        PthSep := string(os.PathSeparator)
        //suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
    
        for _, fi := range dir {
            if fi.IsDir() { // 目录, 递归遍历
                dirs = append(dirs, dirPth+PthSep+fi.Name())
                GetAllFiles(dirPth + PthSep + fi.Name())
            } else {
                // 过滤指定格式
                ok := strings.HasSuffix(fi.Name(), ".go")
                if ok {
                    files = append(files, dirPth+PthSep+fi.Name())
                }
            }
        }
    
        // 读取子目录下文件
        for _, table := range dirs {
            temp, _ := GetAllFiles(table)
            for _, temp1 := range temp {
                files = append(files, temp1)
            }
        }
    
        return files, nil
    }
    
    func main() {
        files, dirs, _ := GetFilesAndDirs("./simplemath")
    
        for _, dir := range dirs {
            fmt.Printf("获取的文件夹为[%s]\n", dir)
        }
    
        for _, table := range dirs {
            temp, _, _ := GetFilesAndDirs(table)
            for _, temp1 := range temp {
                files = append(files, temp1)
            }
        }
    
        for _, table1 := range files {
            fmt.Printf("获取的文件为[%s]\n", table1)
        }
    
        fmt.Printf("=======================================\n")
        xfiles, _ := GetAllFiles("./simplemath")
        for _, file := range xfiles {
            fmt.Printf("获取的文件为[%s]\n", file)
        }
    }

输出结果：

获取的文件夹为[./simplemath\mid]
    获取的文件夹为[./simplemath\sub]
    获取的文件为[./simplemath\add.go]
    获取的文件为[./simplemath\mid\mid.go]
    获取的文件为[./simplemath\sub\sub.go]
    
    获取的文件为[./simplemath\add.go]
    获取的文件为[./simplemath\mid\mid.go]
    获取的文件为[./simplemath\sub\sub.go]