使用Go获取目录中的文件数

Question

我如何获得io/ioutil.ReadDir()返回的项目数？

我有这个代码,它有效,但我不得不认为不是Go中的RightWay(tm).

package main

import "io/ioutil"
import "fmt"

func main() {
    files,_ := ioutil.ReadDir("/Users/dgolliher/Dropbox/INBOX")
    var count int
    for _, f := range files {
        fmt.Println(f.Name())
        count++
    }
    fmt.Println(count)
}

第8-12行似乎太过分了,无法计算ReadDir的结果,但是我找不到正确的语法来获取计数而不迭代范围.救命？

Answer 1

当我发布我的问题后,就把它弄好了.

在http://blog.golang.org/go-slices-usage-and-internals中找到答案

package main

import "io/ioutil"
import "fmt"

func main() {
    files,_ := ioutil.ReadDir("/Users/dgolliher/Dropbox/INBOX")
    fmt.Println(len(files))
}

Answer 2

ReadDir 返回按文件名排序的目录条目列表，因此它不仅仅是文件。对于那些只想获取文件计数（而不是目录）的人来说，这是一个小功能：

func fileCount(path string) (int, error){
    i := 0
    files, err := ioutil.ReadDir(path)
    if err != nil {
        return 0, err
    }
    for _, file := range files {
        if !file.IsDir() { 
            i++
        }
    }
    return i, nil
}

Answer 3

从 Go 1.16（2021 年 2 月）开始，更好的选择是os.ReadDir：

package main
import "os"

func main() {
   d, e := os.ReadDir(".")
   if e != nil {
      panic(e)
   }
   println(len(d))
}

os.ReadDir返回fs.DirEntry而不是fs.FileInfo，这意味着 省略了Size和ModTime方法，如果您只需要条目计数，则可以使过程更加高效。

https://golang.org/pkg/os#ReadDir