d519890f03
* feat: set logging info for cloud- or oss-only commands * fix: add cloud-only to BucketSchemasApi * fix: api-only flagging and %w for return error wrapping * fix: keep the model assignment
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/influxdata/influx-cli/v2/clients/restore"
|
|
"github.com/influxdata/influx-cli/v2/pkg/cli/middleware"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
func newRestoreCmd() *cli.Command {
|
|
var params restore.Params
|
|
|
|
return &cli.Command{
|
|
Name: "restore",
|
|
Usage: "Restores a backup directory to InfluxDB",
|
|
Description: `Restore influxdb.
|
|
|
|
Examples:
|
|
# backup all data
|
|
influx restore /path/to/restore
|
|
`,
|
|
ArgsUsage: "path",
|
|
Before: middleware.WithBeforeFns(withCli(), withApi(true)),
|
|
Flags: append(
|
|
commonFlagsNoPrint(),
|
|
&cli.BoolFlag{
|
|
Name: "full",
|
|
Usage: "Fully restore and replace all data on server",
|
|
Destination: ¶ms.Full,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "org-id",
|
|
Usage: "The original ID of the organization to restore",
|
|
EnvVars: []string{"INFLUX_ORG_ID"},
|
|
Destination: ¶ms.OrgID,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "org",
|
|
Usage: "The original name of the organization to restore",
|
|
Aliases: []string{"o"},
|
|
EnvVars: []string{"INFLUX_ORG"},
|
|
Destination: ¶ms.Org,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "bucket-id",
|
|
Usage: "The original ID of the bucket to restore",
|
|
Destination: ¶ms.BucketID,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "bucket",
|
|
Usage: "The original name of the bucket to restore",
|
|
Aliases: []string{"b"},
|
|
Destination: ¶ms.Bucket,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "new-bucket",
|
|
Usage: "New name to use for the restored bucket",
|
|
Destination: ¶ms.NewBucketName,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "new-org",
|
|
Usage: "New name to use for the restored organization",
|
|
Destination: ¶ms.NewOrgName,
|
|
},
|
|
),
|
|
Action: func(ctx *cli.Context) error {
|
|
if ctx.NArg() != 1 {
|
|
return errors.New("restore path must be specified as a single positional argument")
|
|
}
|
|
params.Path = ctx.Args().Get(0)
|
|
|
|
if params.Full && (params.Org != "" ||
|
|
params.OrgID != "" ||
|
|
params.Bucket != "" ||
|
|
params.BucketID != "" ||
|
|
params.NewOrgName != "" ||
|
|
params.NewBucketName != "") {
|
|
return errors.New("--full restore cannot be limited to a single org or bucket")
|
|
}
|
|
|
|
if params.NewOrgName != "" && params.OrgID == "" && params.Org == "" {
|
|
return errors.New("--org-id or --org must be set to use --new-org")
|
|
}
|
|
if params.NewBucketName != "" && params.BucketID == "" && params.Bucket == "" {
|
|
return errors.New("--bucket-id or --bucket must be set to use --new-bucket")
|
|
}
|
|
|
|
api := getAPI(ctx)
|
|
client := restore.Client{
|
|
CLI: getCLI(ctx),
|
|
RestoreApi: api.RestoreApi.OnlyOSS(),
|
|
OrganizationsApi: api.OrganizationsApi,
|
|
}
|
|
return client.Restore(ctx.Context, ¶ms)
|
|
},
|
|
}
|
|
}
|