aiweek-reconstruction/tools/file.go

25 lines
361 B
Go

package tools
import (
"os"
"path/filepath"
)
//计算文件夹中的文件数目
func CountFilesInDir(dirPath string) (int, error) {
fileCount := 0
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if !info.IsDir() {
fileCount++
}
return nil
})
return fileCount, err
}