golang遍历返回全部目录不返回具体的文件名

  1. 云栖社区>
  2. 博客>
  3. 正文

golang遍历返回全部目录不返回具体的文件名

青铜剑 2019-08-22 11:41:39 浏览121
展开阅读全文

使用参考:

d := dir.NewDir("/")
dirs, err := d.LoopLevelDir(0)

// 实现遍历目录的功能
// 也可以指定层级遍历,遍历几层目录
package dir

import (

"fmt"
"io/ioutil"
"strings"
"time"

)

type Dir struct {

rootDir   string   // 需要遍历的根目录
curDirs   []string // 当前级别的目录
dirs      []string // 全部的目录
nDirLevel int      // 当前目录级别
bInit     bool     // 是否初始化

}

// 创建一个目录
func NewDir(rootDir string) *Dir {

var dir Dir
dir.rootDir = rootDir
dir.bInit = true
dir.nDirLevel = 0
dir.curDirs = []string{rootDir}
return &dir

}

// 只遍历单层目录
func (d *Dir) loopOneDir(dirPath string) ([]string, error) {

var ds []string
// 如果是读取目录失败的,就是没有权限,直接返回空就好了
dir, err := ioutil.ReadDir(dirPath)
if err != nil {
    return nil, nil
    //return nil, fmt.Errorf("ioutil.ReadDir(%s)错误,%s", dirPath, err.Error())
}

for _, fi := range dir {
    if fi.IsDir() {
        onePath := strings.Replace(dirPath+"/"+fi.Name(), "//", "/", -1)
        ds = append(ds, onePath)
    }
}

return ds, nil

}

// 遍历目录,如果nLevel为0表示一直遍历下去直到为空截止
func (d *Dir) LoopLevelDir(nLevel int) ([]string, error) {

if !d.bInit {
    return nil, fmt.Errorf("未初始化")
}

var err error
for {
    d.nDirLevel++
    var tmpAllDirs, tmpOneDirs []string
    for i := 0; i < len(d.curDirs); i++ {
        if tmpOneDirs, err = d.loopOneDir(d.curDirs[i]); err != nil {
            return nil, fmt.Errorf("遍历根目录[%s]失败,%s", d.rootDir, err.Error())
        }
        tmpAllDirs = append(tmpAllDirs, tmpOneDirs...)
    }
    d.curDirs = tmpAllDirs
    if len(tmpAllDirs) == 0 {
        break
    }
    fmt.Printf("[%s] 第[%d]层目录遍历完成,共计[%d]文件夹\n", FormatTime(), d.nDirLevel, len(tmpAllDirs))
    d.dirs = append(d.dirs, tmpAllDirs...)
    if d.nDirLevel >= nLevel && nLevel != 0 {
        break
    }
}
return d.dirs, nil

}

// 获得当前格式化的数据,形如2017-03-15 16:07:32.236
func FormatTime() string {

return fmt.Sprintf("%04d-%02d-%02d %02d:%02d:%02d.%03d", time.Now().Year(),
    time.Now().Month(), time.Now().Day(), time.Now().Hour(), time.Now().Minute(),
    time.Now().Second(), time.Now().Nanosecond()/1e6)

}

网友评论

登录后评论
0/500
评论
青铜剑
+ 关注