seaweedfs/weed/shell/command_fs_du.go

85 lines
1.9 KiB
Go
Raw Normal View History

package shell
import (
"fmt"
2019-12-13 08:22:37 +00:00
"io"
2020-09-01 07:21:19 +00:00
"github.com/chrislusf/seaweedfs/weed/filer"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
2020-03-23 07:01:34 +00:00
"github.com/chrislusf/seaweedfs/weed/util"
)
func init() {
Commands = append(Commands, &commandFsDu{})
}
type commandFsDu struct {
}
func (c *commandFsDu) Name() string {
return "fs.du"
}
func (c *commandFsDu) Help() string {
2019-03-23 18:54:26 +00:00
return `show disk usage
fs.du /dir
fs.du /dir/file_name
fs.du /dir/file_prefix
2019-03-23 18:54:26 +00:00
`
}
func (c *commandFsDu) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
2020-03-24 04:26:15 +00:00
path, err := commandEnv.parseUrl(findInputDirectory(args))
if err != nil {
return err
}
2020-03-24 04:26:15 +00:00
if commandEnv.isDirectory(path) {
path = path + "/"
}
2019-12-13 08:22:37 +00:00
var blockCount, byteCount uint64
2020-03-23 07:01:34 +00:00
dir, name := util.FullPath(path).DirAndName()
2020-03-24 04:26:15 +00:00
blockCount, byteCount, err = duTraverseDirectory(writer, commandEnv, dir, name)
2019-12-13 08:22:37 +00:00
if name == "" && err == nil {
fmt.Fprintf(writer, "block:%4d\tbyte:%10d\t%s\n", blockCount, byteCount, dir)
}
2019-12-13 08:22:37 +00:00
return
}
2020-03-23 07:01:34 +00:00
func duTraverseDirectory(writer io.Writer, filerClient filer_pb.FilerClient, dir, name string) (blockCount, byteCount uint64, err error) {
err = filer_pb.ReadDirAllEntries(filerClient, util.FullPath(dir), name, func(entry *filer_pb.Entry, isLast bool) error {
2020-03-24 04:37:04 +00:00
var fileBlockCount, fileByteCount uint64
2019-12-13 08:22:37 +00:00
if entry.IsDirectory {
subDir := fmt.Sprintf("%s/%s", dir, entry.Name)
if dir == "/" {
subDir = "/" + entry.Name
}
numBlock, numByte, err := duTraverseDirectory(writer, filerClient, subDir, "")
2019-12-13 08:22:37 +00:00
if err == nil {
blockCount += numBlock
byteCount += numByte
}
2019-12-13 08:22:37 +00:00
} else {
2020-03-24 04:37:04 +00:00
fileBlockCount = uint64(len(entry.Chunks))
2020-09-01 07:21:19 +00:00
fileByteCount = filer.FileSize(entry)
blockCount += fileBlockCount
byteCount += fileByteCount
}
2019-12-13 08:22:37 +00:00
if name != "" && !entry.IsDirectory {
2020-03-24 04:37:04 +00:00
fmt.Fprintf(writer, "block:%4d\tbyte:%10d\t%s/%s\n", fileBlockCount, fileByteCount, dir, entry.Name)
2019-12-13 08:22:37 +00:00
}
return nil
2019-12-13 08:22:37 +00:00
})
return
}