1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-28 17:03:58 +08:00

Use filepath.Walk to compute serialfile total size

License: MIT
Signed-off-by: rht <rhtbot@gmail.com>
This commit is contained in:
rht
2015-08-31 09:53:04 +07:00
parent 8c652907de
commit b45e82418b

View File

@ -126,31 +126,16 @@ func (f *serialFile) Stat() os.FileInfo {
} }
func (f *serialFile) Size() (int64, error) { func (f *serialFile) Size() (int64, error) {
return size(f.stat, f.FileName()) if !f.stat.IsDir() {
} return f.stat.Size(), nil
func size(stat os.FileInfo, filename string) (int64, error) {
if !stat.IsDir() {
return stat.Size(), nil
} }
file, err := os.Open(filename) var du int64
if err != nil { err := fp.Walk(f.FileName(), func(p string, fi os.FileInfo, err error) error {
return 0, err if fi != nil && fi.Mode()&(os.ModeSymlink|os.ModeNamedPipe) == 0 {
} du += fi.Size()
files, err := file.Readdir(0)
if err != nil {
return 0, err
}
file.Close()
var output int64
for _, child := range files {
s, err := size(child, fp.Join(filename, child.Name()))
if err != nil {
return 0, err
} }
output += s return nil
} })
return output, nil return du, err
} }