From 0690a54299c08adfdc0589799e632e7f567f951c Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Fri, 15 Jan 2021 14:55:06 +0300 Subject: [PATCH] [#315] cmd/cli: Add set-status command to control section Add set-status command that uses SetNetmapStatus rpc of Control API. Signed-off-by: Leonard Lyubich --- cmd/neofs-cli/modules/control.go | 83 +++++++++++++++++++++++++++++++- 1 file changed, 81 insertions(+), 2 deletions(-) diff --git a/cmd/neofs-cli/modules/control.go b/cmd/neofs-cli/modules/control.go index 94232c37..f72c5679 100644 --- a/cmd/neofs-cli/modules/control.go +++ b/cmd/neofs-cli/modules/control.go @@ -2,11 +2,13 @@ package cmd import ( "context" + "fmt" "github.com/nspcc-dev/neofs-api-go/util/signature" "github.com/nspcc-dev/neofs-api-go/v2/client" "github.com/nspcc-dev/neofs-node/pkg/services/control" controlSvc "github.com/nspcc-dev/neofs-node/pkg/services/control/server" + "github.com/pkg/errors" "github.com/spf13/cobra" ) @@ -23,12 +25,38 @@ var healthCheckCmd = &cobra.Command{ RunE: healthCheck, } +var setNetmapStatusCmd = &cobra.Command{ + Use: "set-status", + Short: "Set status of the storage node in NeoFS network map", + Long: "Set status of the storage node in NeoFS network map", + RunE: setNetmapStatus, +} + +const ( + netmapStatusFlag = "status" + + netmapStatusOnline = "online" + netmapStatusOffline = "offline" +) + +var netmapStatus string + func init() { rootCmd.AddCommand(controlCmd) - controlCmd.AddCommand(healthCheckCmd) + controlCmd.AddCommand( + healthCheckCmd, + setNetmapStatusCmd, + ) - controlCmd.AddCommand(snapshotCmd) + setNetmapStatusCmd.Flags().StringVarP(&netmapStatus, netmapStatusFlag, "", "", + fmt.Sprintf("new netmap status keyword ('%s', '%s')", + netmapStatusOnline, + netmapStatusOffline, + ), + ) + + _ = setNetmapStatusCmd.MarkFlagRequired(netmapStatusFlag) } func getControlServiceClient() (control.ControlServiceClient, error) { @@ -89,3 +117,54 @@ func healthCheck(cmd *cobra.Command, _ []string) error { return nil } + +func setNetmapStatus(cmd *cobra.Command, _ []string) error { + key, err := getKey() + if err != nil { + return err + } + + var status control.NetmapStatus + + switch netmapStatus { + default: + return errors.Errorf("unsupported status %s", netmapStatus) + case netmapStatusOnline: + status = control.NetmapStatus_ONLINE + case netmapStatusOffline: + status = control.NetmapStatus_OFFLINE + } + + req := new(control.SetNetmapStatusRequest) + + body := new(control.SetNetmapStatusRequest_Body) + req.SetBody(body) + + body.SetStatus(status) + + if err := controlSvc.SignMessage(key, req); err != nil { + return err + } + + cli, err := getControlServiceClient() + if err != nil { + return err + } + + resp, err := cli.SetNetmapStatus(context.Background(), req) + if err != nil { + return err + } + + sign := resp.GetSignature() + + if err := signature.VerifyDataWithSource(resp, func() ([]byte, []byte) { + return sign.GetKey(), sign.GetSign() + }); err != nil { + return err + } + + cmd.Println("Network status update request successfully sent.") + + return nil +}