working commit

cmd/mbasectl/dump.go

@@ -0,0 +1,82 @@
+package main
+
+import (
+	"bytes"
+	"context"
+	"io"
+	"os"
+
+	"mbase/pkg/client"
+	"mbase/pkg/mbctl"
+)
+
+func (util *Util) GetDump(ctx context.Context) (*mbctl.GetDumpResult, error) {
+	var err error
+	res := &mbctl.GetDumpResult{}
+	grpcConn, cli, err := client.NewClient(&util.access)
+	if err != nil {
+		return res, err
+	}
+	defer grpcConn.Close()
+
+	file := os.Stdout
+	if util.filename == "-" {
+		util.filename = ""
+	}
+	if util.filename != "" {
+		file, err = os.OpenFile(util.filename, os.O_CREATE|os.O_WRONLY, 0640)
+		if err != nil {
+			return res, err
+		}
+		defer file.Close()
+	}
+
+	params := &mbctl.GetDumpParams{}
+	res, err = cli.GetDump(ctx, params)
+	if err != nil {
+		return res, err
+	}
+
+	_, err = file.Write([]byte(res.Dump))
+	if err != nil {
+		return res, err
+	}
+
+	res.Dump = ""
+	return res, err
+}
+
+func (util *Util) RestoreDump(ctx context.Context) (*mbctl.RestoreDumpResult, error) {
+	var err error
+	res := &mbctl.RestoreDumpResult{}
+	grpcConn, cli, err := client.NewClient(&util.access)
+	if err != nil {
+		return res, err
+	}
+	defer grpcConn.Close()
+
+	file := os.Stdin
+	if util.filename != "" {
+		file, err = os.Open(util.filename)
+		if err != nil {
+			return res, err
+		}
+		defer file.Close()
+	}
+
+	buffer := bytes.NewBuffer(nil)
+	_, err = io.Copy(buffer, file)
+	if err != nil {
+		return res, err
+	}
+
+	params := &mbctl.RestoreDumpParams{
+		Dump: string(buffer.Bytes()),
+	}
+	res, err = cli.RestoreDump(ctx, params)
+	if err != nil {
+		return res, err
+	}
+
+	return res, err
+}