package control import ( "github.com/nspcc-dev/neofs-api-go/v2/rpc/client" "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common" "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/commonflags" "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/key" "github.com/nspcc-dev/neofs-node/pkg/services/control" "github.com/spf13/cobra" ) var evacuateShardCmd = &cobra.Command{ Use: "evacuate", Short: "Evacuate objects from shard", Long: "Evacuate objects from shard to other shards", Run: evacuateShard, } func evacuateShard(cmd *cobra.Command, _ []string) { pk := key.Get(cmd) req := &control.EvacuateShardRequest{Body: new(control.EvacuateShardRequest_Body)} req.Body.Shard_ID = getShardIDList(cmd) req.Body.IgnoreErrors, _ = cmd.Flags().GetBool(dumpIgnoreErrorsFlag) signRequest(cmd, pk, req) cli := getClient(cmd, pk) var resp *control.EvacuateShardResponse var err error err = cli.ExecRaw(func(client *client.Client) error { resp, err = control.EvacuateShard(client, req) return err }) common.ExitOnErr(cmd, "rpc error: %w", err) cmd.Printf("Objects moved: %d\n", resp.GetBody().GetCount()) verifyResponse(cmd, resp.GetSignature(), resp.GetBody()) cmd.Println("Shard has successfully been evacuated.") } func initControlEvacuateShardCmd() { commonflags.InitWithoutRPC(evacuateShardCmd) flags := evacuateShardCmd.Flags() flags.String(controlRPC, controlRPCDefault, controlRPCUsage) flags.StringSlice(shardIDFlag, nil, "List of shard IDs in base58 encoding") flags.Bool(dumpIgnoreErrorsFlag, false, "Skip invalid/unreadable objects") _ = evacuateShardCmd.MarkFlagRequired(shardIDFlag) }