1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-29 01:12:24 +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) { var du int64
if !stat.IsDir() { err := fp.Walk(f.FileName(), func(p string, fi os.FileInfo, err error) error {
return stat.Size(), nil if fi != nil && fi.Mode()&(os.ModeSymlink|os.ModeNamedPipe) == 0 {
du += fi.Size()
} }
return nil
file, err := os.Open(filename) })
if err != nil { return du, err
return 0, err
}
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 output, nil
} }