123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- // +build linux freebsd solaris
- package directory
- import (
- "os"
- "path/filepath"
- "syscall"
- )
- // Size walks a directory tree and returns its total size in bytes.
- func Size(dir string) (size int64, err error) {
- data := make(map[uint64]struct{})
- err = filepath.Walk(dir, func(d string, fileInfo os.FileInfo, err error) error {
- if err != nil {
- // if dir does not exist, Size() returns the error.
- // if dir/x disappeared while walking, Size() ignores dir/x.
- if os.IsNotExist(err) && d != dir {
- return nil
- }
- return err
- }
- // Ignore directory sizes
- if fileInfo == nil {
- return nil
- }
- s := fileInfo.Size()
- if fileInfo.IsDir() || s == 0 {
- return nil
- }
- // Check inode to handle hard links correctly
- inode := fileInfo.Sys().(*syscall.Stat_t).Ino
- // inode is not a uint64 on all platforms. Cast it to avoid issues.
- if _, exists := data[uint64(inode)]; exists {
- return nil
- }
- // inode is not a uint64 on all platforms. Cast it to avoid issues.
- data[uint64(inode)] = struct{}{}
- size += s
- return nil
- })
- return
- }
|