forked from TrueCloudLab/frostfs-node
[#1731] neofs-cli: Add control shards evacuate
command
Signed-off-by: Evgenii Stratonikov <evgeniy@morphbits.ru>
This commit is contained in:
parent
091d7d30f6
commit
8fc88487db
2 changed files with 57 additions and 0 deletions
55
cmd/neofs-cli/modules/control/evacuate_shard.go
Normal file
55
cmd/neofs-cli/modules/control/evacuate_shard.go
Normal file
|
@ -0,0 +1,55 @@
|
||||||
|
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 = getShardID(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.String(shardIDFlag, "", "Shard ID in base58 encoding")
|
||||||
|
flags.Bool(dumpIgnoreErrorsFlag, false, "Skip invalid/unreadable objects")
|
||||||
|
|
||||||
|
_ = evacuateShardCmd.MarkFlagRequired(shardIDFlag)
|
||||||
|
_ = evacuateShardCmd.MarkFlagRequired(controlRPC)
|
||||||
|
}
|
|
@ -15,9 +15,11 @@ func initControlShardsCmd() {
|
||||||
shardsCmd.AddCommand(setShardModeCmd)
|
shardsCmd.AddCommand(setShardModeCmd)
|
||||||
shardsCmd.AddCommand(dumpShardCmd)
|
shardsCmd.AddCommand(dumpShardCmd)
|
||||||
shardsCmd.AddCommand(restoreShardCmd)
|
shardsCmd.AddCommand(restoreShardCmd)
|
||||||
|
shardsCmd.AddCommand(evacuateShardCmd)
|
||||||
|
|
||||||
initControlShardsListCmd()
|
initControlShardsListCmd()
|
||||||
initControlSetShardModeCmd()
|
initControlSetShardModeCmd()
|
||||||
initControlDumpShardCmd()
|
initControlDumpShardCmd()
|
||||||
initControlRestoreShardCmd()
|
initControlRestoreShardCmd()
|
||||||
|
initControlEvacuateShardCmd()
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue