扣丁书屋

golang 递归遍历一个文件夹

1年以前  |  阅读数:125 次  |    

示例只是递归的一个写法,实际生产品,可以封装成一个方法。

package main

import (
    "os"
    "fmt"
)

func main(){
    err :=  tree("src",0)
    if err != nil{
        fmt.Println(err)
    }

}


func tree(dstPath string,level int)error{
    dstF,err := os.Open(dstPath)
    if err != nil{
        return err
    }
    defer dstF.Close()
    fileInfo,err := dstF.Stat()
    if err != nil{
        return err
    }
    if !fileInfo.IsDir() {  //如果dstF是文件
        for i:=0;i < level;i++ {
            fmt.Print("--")
        }
        fmt.Println(dstPath)
        return nil
    }else {  //如果dstF是文件夹
        for i:=0;i < level;i++ {
            fmt.Print("--")
        }
        fmt.Println(dstF.Name())
        dir,err := dstF.Readdir(0)     //获取文件夹下各个文件或文件夹的fileInfo
        if err != nil{
            return err
        }
        for _,fileInfo = range dir{
            err = tree(dstPath+"/"+fileInfo.Name(),level+1)
            if err != nil{
                return err
            }
        }
        return nil
    }

}

相关文章: