83 lines
1.4 KiB
Go
83 lines
1.4 KiB
Go
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
|
|
}
|