From 014906ec797c7075afed2b33705e0db63bcd2dbd Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Tue, 16 Apr 2019 01:37:11 -0700
Subject: [PATCH] weed shell: add command fs.meta.load to restore meta data

---
 weed/shell/command_fs_meta_load.go | 103 +++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 weed/shell/command_fs_meta_load.go

diff --git a/weed/shell/command_fs_meta_load.go b/weed/shell/command_fs_meta_load.go
new file mode 100644
index 000000000..ef3c39f96
--- /dev/null
+++ b/weed/shell/command_fs_meta_load.go
@@ -0,0 +1,103 @@
+package shell
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"os"
+
+	"github.com/chrislusf/seaweedfs/weed/filer2"
+	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+	"github.com/chrislusf/seaweedfs/weed/util"
+	"github.com/golang/protobuf/proto"
+)
+
+func init() {
+	commands = append(commands, &commandFsMetaLoad{})
+}
+
+type commandFsMetaLoad struct {
+}
+
+func (c *commandFsMetaLoad) Name() string {
+	return "fs.meta.load"
+}
+
+func (c *commandFsMetaLoad) Help() string {
+	return `load saved filer meta data to restore the directory and file structure
+
+	fs.meta.load <filer_host>_<port>.meta
+
+`
+}
+
+func (c *commandFsMetaLoad) Do(args []string, commandEnv *commandEnv, writer io.Writer) (err error) {
+
+	filerServer, filerPort, path, err := commandEnv.parseUrl(findInputDirectory(nil))
+	if err != nil {
+		return err
+	}
+
+	fileName := args[len(args)-1]
+
+	dst, err := os.OpenFile(fileName, os.O_RDONLY, 0644)
+	if err != nil {
+		return nil
+	}
+	defer dst.Close()
+
+	var dirCount, fileCount uint64
+
+	ctx := context.Background()
+
+	err = commandEnv.withFilerClient(ctx, filerServer, filerPort, func(client filer_pb.SeaweedFilerClient) error {
+
+		sizeBuf := make([]byte, 4)
+
+		for {
+			if n, err := dst.Read(sizeBuf); n != 4 {
+				if err == io.EOF{
+					return nil
+				}
+				return err
+			}
+
+			size := util.BytesToUint32(sizeBuf)
+
+			data := make([]byte, int(size))
+
+			if n, err := dst.Read(data); n != len(data) {
+				return err
+			}
+
+			fullEntry := &filer_pb.FullEntry{}
+			if err = proto.Unmarshal(data, fullEntry); err != nil {
+				return err
+			}
+
+			if _, err = client.CreateEntry(ctx, &filer_pb.CreateEntryRequest{
+				Directory: fullEntry.Dir,
+				Entry:     fullEntry.Entry,
+			}); err != nil {
+				return err
+			}
+
+			fmt.Fprintf(writer, "load %s\n", filer2.FullPath(fullEntry.Dir).Child(fullEntry.Entry.Name))
+
+			if fullEntry.Entry.IsDirectory {
+				dirCount++
+			} else {
+				fileCount++
+			}
+
+		}
+
+	})
+
+	if err == nil {
+		fmt.Fprintf(writer, "\ntotal %d directories, %d files", dirCount, fileCount)
+		fmt.Fprintf(writer, "\n%s is loaded to http://%s:%d%s\n", fileName, filerServer, filerPort, path)
+	}
+
+	return err
+}
\ No newline at end of file