ale64bit/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Issues Pull requests Projects Releases Packages Wiki Activity

[#414] cmd/cli: Support IR health-check

Browse source 
Add `--ir` flag to `control healthcheck` to communicate with IR node.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2021-06-15 10:54:09 +03:00 committed by Alex Vanin
parent 455fd952dd
commit 7cead1bc3a
1 changed files with 54 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
cmd/neofs-cli/modules/control.go
View file

@ -1,11 +1,15 @@
package cmd
import (
"crypto/ecdsa"
"fmt"
"github.com/nspcc-dev/neofs-api-go/pkg/client"
"github.com/nspcc-dev/neofs-api-go/pkg/object"
"github.com/nspcc-dev/neofs-api-go/util/signature"
"github.com/nspcc-dev/neofs-node/pkg/services/control"
ircontrol "github.com/nspcc-dev/neofs-node/pkg/services/control/ir"
ircontrolsrv "github.com/nspcc-dev/neofs-node/pkg/services/control/ir/server"
controlSvc "github.com/nspcc-dev/neofs-node/pkg/services/control/server"
"github.com/spf13/cobra"
)
@ -37,6 +41,16 @@ const (
netmapStatusOffline = "offline"
)
// control healthcheck flags
const (
healthcheckIRFlag = "ir"
)
// control healthcheck vars
var (
healthCheckIRVar bool
)
var netmapStatus string
func init() {
@ -61,6 +75,8 @@ func init() {
"List of object addresses to be removed in string format")
_ = dropObjectsCmd.MarkFlagRequired(dropObjectsFlag)
healthCheckCmd.Flags().BoolVar(&healthCheckIRVar, healthcheckIRFlag, false, "Communicate with IR node")
}
func healthCheck(cmd *cobra.Command, _ []string) error {
@ -69,6 +85,16 @@ func healthCheck(cmd *cobra.Command, _ []string) error {
return err
}
cli, err := getSDKClient(key)
if err != nil {
return err
}
if healthCheckIRVar {
healthCheckIR(cmd, key, cli)
return nil
}
req := new(control.HealthCheckRequest)
req.SetBody(new(control.HealthCheckRequest_Body))
@ -77,11 +103,6 @@ func healthCheck(cmd *cobra.Command, _ []string) error {
return err
}
cli, err := getSDKClient(key)
if err != nil {
return err
}
resp, err := control.HealthCheck(cli.Raw(), req)
if err != nil {
return err
@ -101,6 +122,34 @@ func healthCheck(cmd *cobra.Command, _ []string) error {
return nil
}
func healthCheckIR(cmd *cobra.Command, key *ecdsa.PrivateKey, c client.Client) {
req := new(ircontrol.HealthCheckRequest)
req.SetBody(new(ircontrol.HealthCheckRequest_Body))
if err := ircontrolsrv.SignMessage(key, req); err != nil {
cmd.PrintErrln(fmt.Errorf("could not sign request: %w", err))
return
}
resp, err := ircontrol.HealthCheck(c.Raw(), req)
if err != nil {
cmd.PrintErrln(fmt.Errorf("rpc failure: %w", err))
return
}
sign := resp.GetSignature()
if err := signature.VerifyDataWithSource(resp, func() ([]byte, []byte) {
return sign.GetKey(), sign.GetSign()
}); err != nil {
cmd.PrintErrln(fmt.Errorf("invalid response signature: %w", err))
return
}
cmd.Printf("Health status: %s\n", resp.GetBody().GetHealthStatus())
}
func setNetmapStatus(cmd *cobra.Command, _ []string) error {
key, err := getKey()
if err != nil {