scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-09-07 17:15:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Todd Gruben 2021-05-20 11:44:28 -05:00
parent 35f4c33de9
commit 3d577d762a
1 changed files with 9 additions and 56 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
ctl/restore.go
View file

@ -47,44 +47,11 @@ type RestoreCommand struct {
// NewRestoreCommand returns a new instance of RestoreCommand.
func NewRestoreCommand(stdin io.Reader, stdout, stderr io.Writer) *RestoreCommand {
/*
h := &gohttp.Client{}
host := "SOMETHING"
c, err := http.NewInternalClient(host, h)
if err != nil {
panic(err)
}
*/
return &RestoreCommand{
CmdIO: pilosa.NewCmdIO(stdin, stdout, stderr),
// client: c,
}
}
/* helper to allow for both gz and just plan tar
f, err := os.Open(cmd.Path)
if err != nil {
return (err)
}
defer f.Close()
var tarReader *tar.Reader
if strings.HasSuffix(cmd.Path, "gz") {
gzf, err := gzip.NewReader(f)
if err != nil {
return err
}
tarReader = tar.NewReader(gzf)
} else {
tarReader = tar.NewReader(f)
}
return
}
*/
func readSchema(path string) string {
return "{}"
}
// Run executes the restore.
func (cmd *RestoreCommand) Run(ctx context.Context) error {
// Create a client to the server.
@ -93,10 +60,15 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
return fmt.Errorf("creating client: %w", err)
}
cmd.client = client
f, err := os.Open(cmd.Path)
if err != nil {
return (err)
var f *os.File
// read from Stdin if path specified as -
if cmd.Path == "-" {
f = os.Stdin
} else {
f, err = os.Open(cmd.Path)
if err != nil {
return (err)
}
}
defer f.Close()
var tarReader *tar.Reader
@ -109,24 +81,6 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
} else {
tarReader = tar.NewReader(f)
}
//maybe begin transaction?
/*
schemaJson := readSchema(cmd.Path)
//Push the schema from the archive into the cluster belonging to the host.
client := &gohttp.Client{}
url := "FIXME"
_, err = client.Post(url+"/schema", "application/json", bytes.NewBufferString(schemaJson))
if err != nil {
return err
}
*/
//TODO (twg) load schema
//TODO (twg) load rbf shard
//TODO (twg) load row keys
//TODO (twg) load column keys
//TODO (twg) load row attributes keys
//TODO (twg) load col attributes keys
//TODO (twg) load idalloc
nodes, err := cmd.client.Nodes(ctx)
if err != nil {
return err
@ -154,7 +108,6 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
case "schema":
vprint.VV("Load Schema")
url := primary.URI.Path("/schema")
vprint.VV("SCHEMA %v", url)
_, err = c.Post(url, "application/json", tarReader)
if err != nil {
return err