From f3d316a846473fe8f33a90172291608d764a264d Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@uber.com>
Date: Tue, 19 Mar 2019 21:58:00 -0700
Subject: [PATCH] weed shell: adding file system disk usage fs.du

fix https://github.com/chrislusf/seaweedfs/issues/889

echo "fs.du http://localhost:8888/some/path" | weed shell
---
 weed/shell/command_fs_du.go | 137 ++++++++++++++++++++++++++++++++++++
 weed/shell/commands.go      |   1 +
 weed/shell/shell_liner.go   |   1 +
 3 files changed, 139 insertions(+)
 create mode 100644 weed/shell/command_fs_du.go

diff --git a/weed/shell/command_fs_du.go b/weed/shell/command_fs_du.go
new file mode 100644
index 000000000..cdc9d98ef
--- /dev/null
+++ b/weed/shell/command_fs_du.go
@@ -0,0 +1,137 @@
+package shell
+
+import (
+	"context"
+	"fmt"
+	"github.com/chrislusf/seaweedfs/weed/filer2"
+	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+	"github.com/chrislusf/seaweedfs/weed/util"
+	"google.golang.org/grpc"
+	"io"
+	"net/url"
+	"strconv"
+	"strings"
+)
+
+func init() {
+	commands = append(commands, &commandFsDu{})
+}
+
+type commandFsDu struct {
+}
+
+func (c *commandFsDu) Name() string {
+	return "fs.du"
+}
+
+func (c *commandFsDu) Help() string {
+	return "http://<filer_server>:<port>/dir[/file] # show disk usage"
+}
+
+func (c *commandFsDu) Do(args []string, commandEnv *commandEnv, writer io.Writer) (err error) {
+
+	filerServer, filerPort, path, err := parseFilerUrl(args[0])
+	if err != nil {
+		return err
+	}
+
+	dir, name := filer2.FullPath(path).DirAndName()
+	if strings.HasSuffix(path, "/") {
+		if path == "/"{
+			dir, name = "/", ""
+		}else{
+			dir, name = path[0 : len(path)-1], ""
+		}
+	}
+
+	ctx := context.Background()
+
+	return commandEnv.withFilerClient(ctx, filerServer, filerPort, func(client filer_pb.SeaweedFilerClient) error {
+
+		_, _, err = paginateDirectory(ctx, writer, client, dir, name, 1000)
+
+		return err
+
+	})
+
+}
+
+func paginateDirectory(ctx context.Context, writer io.Writer, client filer_pb.SeaweedFilerClient, dir, name string, paginateSize int) (blockCount uint64, byteCount uint64, err error) {
+
+	paginatedCount := -1
+	startFromFileName := ""
+
+	for paginatedCount == -1 || paginatedCount == paginateSize {
+		resp, listErr := client.ListEntries(ctx, &filer_pb.ListEntriesRequest{
+			Directory:          dir,
+			Prefix:             name,
+			StartFromFileName:  startFromFileName,
+			InclusiveStartFrom: false,
+			Limit:              uint32(paginateSize),
+		})
+		if listErr != nil {
+			err = listErr
+			return
+		}
+
+		paginatedCount = len(resp.Entries)
+
+		for _, entry := range resp.Entries {
+			if entry.IsDirectory {
+				subDir := fmt.Sprintf("%s/%s", dir, entry.Name)
+				if dir == "/" {
+					subDir = "/" + entry.Name
+				}
+				numBlock, numByte, err := paginateDirectory(ctx, writer, client, subDir, "", paginateSize)
+				if err == nil {
+					blockCount += numBlock
+					byteCount += numByte
+				}
+			} else {
+				blockCount += uint64(len(entry.Chunks))
+				byteCount += filer2.TotalSize(entry.Chunks)
+			}
+			startFromFileName = entry.Name
+
+			if name != "" && !entry.IsDirectory {
+				fmt.Fprintf(writer, "block:%4d\tbyte:%10d\t%s/%s\n", blockCount, byteCount, dir, name)
+			}
+		}
+	}
+
+	if name == "" {
+		fmt.Fprintf(writer, "block:%4d\tbyte:%10d\t%s\n", blockCount, byteCount, dir)
+	}
+
+	return
+
+}
+
+func parseFilerUrl(entryPath string) (filerServer string, filerPort int64, path string, err error) {
+	if strings.HasPrefix(entryPath, "http") {
+		var u *url.URL
+		u, err = url.Parse(entryPath)
+		if err != nil {
+			return
+		}
+		filerServer = u.Hostname()
+		portString := u.Port()
+		if portString != "" {
+			filerPort, err = strconv.ParseInt(portString, 10, 32)
+		}
+		path = u.Path
+	} else {
+		err = fmt.Errorf("path should have full url http://<filer_server>:<port>/path/to/dirOrFile : %s", entryPath)
+	}
+	return
+}
+
+func (env *commandEnv) withFilerClient(ctx context.Context, filerServer string, filerPort int64, fn func(filer_pb.SeaweedFilerClient) error) error {
+
+	filerGrpcAddress := fmt.Sprintf("%s:%d", filerServer, filerPort+10000)
+	return util.WithCachedGrpcClient(ctx, func(grpcConnection *grpc.ClientConn) error {
+		client := filer_pb.NewSeaweedFilerClient(grpcConnection)
+		return fn(client)
+	}, filerGrpcAddress, env.option.GrpcDialOption)
+
+}
diff --git a/weed/shell/commands.go b/weed/shell/commands.go
index 4df70ff55..280900c80 100644
--- a/weed/shell/commands.go
+++ b/weed/shell/commands.go
@@ -14,6 +14,7 @@ type ShellOptions struct {
 type commandEnv struct {
 	env          map[string]string
 	masterClient *wdclient.MasterClient
+	option       ShellOptions
 }
 
 type command interface {
diff --git a/weed/shell/shell_liner.go b/weed/shell/shell_liner.go
index cd015fe85..096532fdf 100644
--- a/weed/shell/shell_liner.go
+++ b/weed/shell/shell_liner.go
@@ -36,6 +36,7 @@ func RunShell(options ShellOptions) {
 		env: make(map[string]string),
 		masterClient: wdclient.NewMasterClient(context.Background(),
 			options.GrpcDialOption, "shell", strings.Split(*options.Masters, ",")),
+		option: options,
 	}
 
 	go commandEnv.masterClient.KeepConnectedToMaster()